追踪

panlei

2019-07-27 bb215e57c6a8ae527820697b59cec900d4d3fdfe

 ruleserver/personTrack.go

@@ -19,7 +19,7 @@
   logger.Debug("---------------------缓存池--------------------------：", TrackPond)
   for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 遍历各算法的sdkData
      if sdkinfo.Sdktype == "FaceDetect" { // 人脸检测
         //if len(sdkinfo.Sdkdata) > 1 {
         if len(sdkinfo.Sdkdata) > 1 {
         faceParam := protomsg.ParamFacePos{}
         err := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
         if err != nil {
@@ -47,12 +47,14 @@
               TrackPond[msg.Cid] = &PersonTrack{Face: len(faceParam.Faces)}
            }
         }
         //} else {
         //   continue
         //}
         } else {
            TrackPond[msg.Cid].Face = 0
            logger.Info("没有人脸，被重置为0")
            continue
         }
      }
      if sdkinfo.Sdktype == "Yolo" {
         //if len(sdkinfo.Sdkdata) > 1 {
         if len(sdkinfo.Sdkdata) > 1 {
         yoloParam := protomsg.ParamYoloObj{}
         err := proto.Unmarshal(sdkinfo.Sdkdata, &yoloParam)
         if err != nil {
@@ -86,9 +88,11 @@
               TrackPond[msg.Cid] = &PersonTrack{Yolo: yoloNum}
            }
         }
         //} else {
         //   continue
         //}
         } else {
            TrackPond[msg.Cid].Yolo = 0
            logger.Info("没有yolo，被重置为0")
            continue
         }
      }
   }
   logger.Debug("---------------------------------------追踪结束--------------------------------------")