From 4e7e3aa3cdfe45c0cb06509ed11330bf9e7019b8 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期一, 22 七月 2019 10:48:19 +0800 Subject: [PATCH] 更正追踪人脸数量错误 --- ruleserver/personTrack.go | 60 ++++++++++++++++++++++++++++++------------------------------ 1 files changed, 30 insertions(+), 30 deletions(-) diff --git a/ruleserver/personTrack.go b/ruleserver/personTrack.go index 8ffd0ad..449abcb 100644 --- a/ruleserver/personTrack.go +++ b/ruleserver/personTrack.go @@ -21,36 +21,36 @@ func IsSame(msg *protomsg.SdkMessage) { logger.Debug("++++++++++++++++++++++++++++杩借釜寮�濮�+++++++++++++++++++++鎽勫儚鏈篿d涓猴細",msg.Cid) for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 閬嶅巻鍚勭畻娉曠殑sdkData - //if sdkinfo.Sdktype == "FaceDetect" { // 浜鸿劯妫�娴� - // if len(sdkinfo.Sdkdata) > 1 { - // faceParam := protomsg.ParamFacePos{} - // err := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam) - // if err != nil { - // logger.Info("瑙f瀽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鐨剆dkData娓呯┖ - // faceParam.Faces = (faceParam.Faces)[0:0] - // sdkinfo.Sdkdata,err = proto.Marshal(&faceParam) - // if err != nil { - // logger.Error("浜鸿劯搴忓垪鍖栭敊璇�",err) - // } - // logger.Info("===================杩借釜涓�斾汉鑴哥殑涓暟",len(faceParam.Faces)) - // } else { - // TrackPond[msg.Cid] = &PersonTrack{Face:len(faceParam.Faces)} - // } - // //for i := 0; i < len(faceParam.Faces); { - // // if flag := IsInPond(msg.Cid,faceParam.Faces[i]);flag { // 宸茬粡瀛樺湪锛屼粠msg鐨勪汉鑴告暟缁勪腑鍒犻櫎杩欎釜鍏冪礌锛屼笉蹇呬紶鍒颁笅涓�涓幆鑺� - // // faceParam.Faces = append(faceParam.Faces[:i], faceParam.Faces[i+1:]...) - // // } else { - // // i++ - // // } - // //} - // } else { - // continue - // } - //} + if sdkinfo.Sdktype == "FaceDetect" { // 浜鸿劯妫�娴� + if len(sdkinfo.Sdkdata) > 1 { + faceParam := protomsg.ParamFacePos{} + err := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam) + if err != nil { + logger.Info("瑙f瀽face sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err) + continue + } + logger.Info("================杩借釜涔嬪墠浜鸿劯鐨勪釜鏁帮細",len(faceParam.Faces)) + if faceParam.Faces != nil && TrackPond[msg.Cid] != nil && len(faceParam.Faces) == TrackPond[msg.Cid].Face { // yolo鐨勫鏋滄暟閲忕浉鍚屽垯瑙嗕负涓嶅彉銆佹妸yolo鐨剆dkData娓呯┖ + faceParam.Faces = (faceParam.Faces)[0:0] + sdkinfo.Sdkdata,err = proto.Marshal(&faceParam) + if err != nil { + logger.Error("浜鸿劯搴忓垪鍖栭敊璇�",err) + } + logger.Info("===================杩借釜涓�斾汉鑴哥殑涓暟",len(faceParam.Faces)) + } else { + TrackPond[msg.Cid] = &PersonTrack{Face:len(faceParam.Faces)} + } + //for i := 0; i < len(faceParam.Faces); { + // if flag := IsInPond(msg.Cid,faceParam.Faces[i]);flag { // 宸茬粡瀛樺湪锛屼粠msg鐨勪汉鑴告暟缁勪腑鍒犻櫎杩欎釜鍏冪礌锛屼笉蹇呬紶鍒颁笅涓�涓幆鑺� + // faceParam.Faces = append(faceParam.Faces[:i], faceParam.Faces[i+1:]...) + // } else { + // i++ + // } + //} + } else { + continue + } + } if sdkinfo.Sdktype == "Yolo" { if len(sdkinfo.Sdkdata) > 1 { yoloParam := protomsg.ParamYoloObj{} -- Gitblit v1.8.0