~jinersong/ruleprocess.git

			@@ -9,17 +9,16 @@
			var TrackPond = make(map[string]*PersonTrack)

			type PersonTrack struct {
			Face int // 人脸的id数组改为数量
			Yolo int // yolo对象的id数组改为数量
			Face int // 人脸的id数组改为数量
			FaceN int // 添加容错，初始值为5，连续5次不对才算不对
			Yolo int // yolo对象的id数组改为数量
			YoloN int // 添加容错，初始值为5，连续5次不对才算不对
			}

			func (pt *PersonTrack) Set(msg protomsg.SdkMessage) {

			}

			// 检查是否前后两次的数据id是否完全相同（人脸和yolo）
			// 检查是否前后两次的数据id是否完全相同（人脸和yolo）改为直接检查数量是否一致
			func IsSame(msg *protomsg.SdkMessage) {
			logger.Debug("++++++++++++++++++++++++++++追踪开始+++++++++++++++++++++摄像机id为：",msg.Cid)
			logger.Debug("+++++++++++++++++++++追踪开始+++++++++++++++摄像机id为：", msg.Cid)
			logger.Debug("---------------------缓存池--------------------------：", TrackPond)
			for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 遍历各算法的sdkData
			if sdkinfo.Sdktype == "FaceDetect" { // 人脸检测
			if len(sdkinfo.Sdkdata) > 1 {
			@@ -29,16 +28,26 @@
			logger.Info("解析face sdk数据时出现错误", err)
			continue
			}
			logger.Info("================追踪之前人脸的个数：",len(faceParam.Faces))
			if faceParam.Faces != nil && TrackPond[msg.Cid] != nil && len(faceParam.Faces) == TrackPond[msg.Cid].Yolo { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			logger.Info("================追踪之前人脸的个数：", len(faceParam.Faces))
			if faceParam.Faces != nil && TrackPond[msg.Cid] != nil && len(faceParam.Faces) == TrackPond[msg.Cid].Face { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			faceParam.Faces = (faceParam.Faces)[0:0]
			sdkinfo.Sdkdata,err = proto.Marshal(&faceParam)
			sdkinfo.Sdkdata, err = proto.Marshal(&faceParam)
			if err != nil {
			logger.Error("人脸序列化错误",err)
			logger.Error("人脸序列化错误", err)
			}
			logger.Info("===================追踪中途人脸的个数",len(faceParam.Faces))
			logger.Info("===================追踪中途人脸的个数", len(faceParam.Faces))
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Face:len(faceParam.Faces)}
			if TrackPond[msg.Cid] != nil {
			if TrackPond[msg.Cid].FaceN > 0 {
			TrackPond[msg.Cid].FaceN--
			}
			if TrackPond[msg.Cid].FaceN == 0 { //此时连续五次不对
			TrackPond[msg.Cid].Face = len(faceParam.Faces)
			TrackPond[msg.Cid].FaceN = 5
			}
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Face: len(faceParam.Faces),FaceN:5}
			}
			}
			//for i := 0; i < len(faceParam.Faces); {
			// if flag := IsInPond(msg.Cid,faceParam.Faces[i]);flag { // 已经存在，从msg的人脸数组中删除这个元素，不必传到下一个环节
			@@ -59,15 +68,32 @@
			logger.Info("解析yolo数据时出现错误", err)
			continue
			}
			logger.Info("================追踪之前yolo的个数：",len(yoloParam.Infos))
			if yoloParam.Infos != nil && TrackPond[msg.Cid] != nil && len(yoloParam.Infos) == TrackPond[msg.Cid].Yolo { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			var yoloNum int = 0
			for i := 0; i < len(yoloParam.Infos); i++ {
			if yoloParam.Infos[i].Typ == 0 {
			yoloNum++
			}
			}
			logger.Info("================追踪之前yolo的个数：", yoloNum)
			if yoloParam.Infos != nil && TrackPond[msg.Cid] != nil && yoloNum == TrackPond[msg.Cid].Yolo { // yolo的如果数量相同则视为不变、把yolo的sdkData清空
			yoloParam.Infos = (yoloParam.Infos)[0:0]
			sdkinfo.Sdkdata,err = proto.Marshal(&yoloParam)
			sdkinfo.Sdkdata, err = proto.Marshal(&yoloParam)
			if err != nil {
			logger.Error("yolo序列化错误",err)
			logger.Error("yolo序列化错误", err)
			}
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Yolo:len(yoloParam.Infos)}
			if TrackPond[msg.Cid] != nil {
			TrackPond[msg.Cid].Yolo = yoloNum
			if TrackPond[msg.Cid].YoloN > 0 {
			TrackPond[msg.Cid].YoloN--
			}
			if TrackPond[msg.Cid].YoloN == 0 { //此时连续五次不对
			TrackPond[msg.Cid].Yolo = yoloNum
			TrackPond[msg.Cid].YoloN = 5
			}
			} else {
			TrackPond[msg.Cid] = &PersonTrack{Yolo: yoloNum,YoloN:5}
			}
			}
			} else {
			continue
			@@ -76,6 +102,7 @@
			}
			logger.Debug("---------------------------------------追踪结束--------------------------------------")
			}

			//// 判断一个人脸id在不在追踪池中
			//func FaceIsInPond(cameraId string,face *protomsg.ResultFaceDetect) bool{
			// if TrackPond[cameraId].Face > 0 {
			@@ -100,4 +127,4 @@
			// return false
			// }
			// return false
			//}
			//}