From 2171427a314af4830f3452bc560b104b27efd970 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期四, 31 十月 2019 14:13:30 +0800 Subject: [PATCH] 人体追踪相同的把yolo标签也删掉 --- ruleserver/ruleToformula.go | 2 +- ruleserver/personTrack.go | 17 ++++++++++------- 2 files changed, 11 insertions(+), 8 deletions(-) diff --git a/ruleserver/personTrack.go b/ruleserver/personTrack.go index dc69af8..eb4fe23 100644 --- a/ruleserver/personTrack.go +++ b/ruleserver/personTrack.go @@ -50,7 +50,7 @@ } // 杩借釜浜轰綋锛屾鏌ユ暟閲忔槸鍚︿竴鑷� -func BodyIsSame(msg *protomsg.SdkMessage) bool{ +func BodyIsSame(args *structure.SdkDatas,msg *protomsg.SdkMessage) bool{ logger.Debug("+++++++++++++++++++++浜轰綋杩借釜寮�濮�+++++++++++++++鎽勫儚鏈篿d涓猴細", msg.Cid, "---缂撳瓨姹犱负锛�", TrackPond) for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 閬嶅巻鍚勭畻娉曠殑sdkData if sdkinfo.Sdktype == "Yolo" { @@ -73,12 +73,14 @@ logger.Info("================杩借釜涔嬪墠yolo鐨勪釜鏁帮細", yoloNum, "杩樻病鏈夎繖涓憚鍍忔満鐨勭紦瀛�") } if yoloParam.Infos != nil && TrackPond[msg.Cid] != nil && yoloNum == TrackPond[msg.Cid].Yolo { // yolo鐨勫鏋滄暟閲忕浉鍚屽垯瑙嗕负涓嶅彉銆佹妸yolo鐨剆dkData娓呯┖ - //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 { @@ -88,6 +90,7 @@ logger.Info("鏂板缓褰撳墠鎽勫儚鏈虹紦瀛樻睜涓殑yolo涓暟锛�", yoloNum) TrackPond[msg.Cid] = &PersonTrack{Yolo: yoloNum} } + logger.Debug("---------------------------------浜轰綋杩借釜缁撴潫--------------------------------------") return false } } else { diff --git a/ruleserver/ruleToformula.go b/ruleserver/ruleToformula.go index 67a0df8..562d284 100644 --- a/ruleserver/ruleToformula.go +++ b/ruleserver/ruleToformula.go @@ -71,7 +71,7 @@ // 濡傛灉鏍囩涓惈鏈夋寔缁椂闂撮娆℃姤璀︾殑timeLabel鐨勮瘽鍒欎笉闇�瑕佽繃浜轰綋杩借釜锛屼笉鐒跺氨娌$殑鎻掑叆浜� fk := TrackOrNot(args.RuleResult) if !fk { - BodyIsSame(message) + BodyIsSame(args,message) } } } -- Gitblit v1.8.0