merge master

panlei

2019-11-07 6e8ebf781ac457b7ec77f7e4c9ade4c3d36002bc

 ruleserver/personTrack.go

@@ -2,8 +2,8 @@

import (
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/logger.git"
   "github.com/golang/protobuf/proto"
   "basic.com/valib/logger.git"
   "ruleprocess/structure"
   "sync"
)
@@ -57,7 +57,7 @@
}

//  追踪人体，检查数量是否一致
func BodyIsSame(msg *protomsg.SdkMessage) bool {
func BodyIsSame(args *structure.SdkDatas,msg *protomsg.SdkMessage) bool{
   logger.Debug("+++++++++++++++++++++人体追踪开始+++++++++++++++摄像机id为：", msg.Cid, "---缓存池为：", TrackPond)
   for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 遍历各算法的sdkData
      if sdkinfo.Sdktype == "Yolo" {
@@ -80,36 +80,30 @@
               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)
               //if err != nil {
               //   logger.Error("yolo序列化错误", err)
               //}
               logger.Info("跟之前相同，清空yolo数据")
               yoloParam.Infos = (yoloParam.Infos)[0:0]
               sdkinfo.Sdkdata, err = proto.Marshal(&yoloParam)
               if err != nil {
                  logger.Error("yolo序列化错误", err)
               }
               delete(args.RuleResult,"yolo")
               logger.Info("清除yolo标签，",args.RuleResult["yolo"])
               logger.Info("跟之前相同，清空yolo数据，人体追踪结束")
               return true
            } else {
               if TrackPond[msg.Cid] != nil {
                  logger.Info("更新当前摄像机缓存池中的yolo个数：", yoloNum)
                  lock.Lock()
                  TrackPond[msg.Cid].Yolo = yoloNum
                  lock.Unlock()
               } else {
                  logger.Info("新建当前摄像机缓存池中的yolo个数：", yoloNum)
                  lock.Lock()
                  TrackPond[msg.Cid] = &PersonTrack{Yolo: yoloNum}
                  lock.Unlock()
               }
               return false
            }
         } else {
            if TrackPond[msg.Cid] != nil {
               lock.Lock()
               TrackPond[msg.Cid].Yolo = 0
               lock.Unlock()
            } else {
               lock.Lock()
               TrackPond[msg.Cid] = &PersonTrack{Yolo: 0}
               lock.Unlock()
            }
            logger.Info("摄像机:" + msg.Cid + "-没有yolo，被重置为0")
            continue
@@ -119,6 +113,7 @@
   logger.Debug("---------------------------------人体追踪结束--------------------------------------")
   return false
}

func TrackOrNot(label map[string]interface{}) bool {
   if label["yolo"] != nil && len(label["yolo"].([]structure.Result)) > 0 {
      for _, res := range label["yolo"].([]structure.Result) {
@@ -150,7 +145,6 @@
      for _, face := range faceParam.Faces {
         facesTemp = append(facesTemp, *face) // 先把数据转存一份，不然一会儿数据删减之后找不到原始数据，不能让缓存数据更新了
      }

      for i := 0; i < len(faceParam.Faces); {
         faceFlag := false
         for _, val := range TrackPond[cameraId].Faces {
@@ -176,7 +170,6 @@
         }
      }
      logger.Info("反序列化重新装配之前人脸的个数：", len(faceParam.Faces))

      logger.Info("临时存储的新来数据：")
      for _, temp := range facesTemp {
         logger.Info("临时存储的新来的的face数据：", temp.Pos.FaceID, temp.Pos.FAngle.Confidence)