From f2035d0fcbb1f09fdc00c468ed6a3d0842d0ad02 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期二, 22 十月 2019 10:30:44 +0800 Subject: [PATCH] 加一行空格测试goland ctrl+k --- ruleserver/personTrack.go | 132 ++++++++++++++++++++++--------------------- 1 files changed, 67 insertions(+), 65 deletions(-) diff --git a/ruleserver/personTrack.go b/ruleserver/personTrack.go index 9bf036e..b7639f5 100644 --- a/ruleserver/personTrack.go +++ b/ruleserver/personTrack.go @@ -21,6 +21,7 @@ } var num int = 25 + // 妫�鏌ユ槸鍚﹀墠鍚庝袱娆$殑鏁版嵁id鏄惁瀹屽叏鐩稿悓锛堜汉鑴革級 func FaceIsSame(msg *protomsg.SdkMessage) { logger.Debug("+++++++++++++++++++++浜鸿劯杩借釜寮�濮�+++++++++++++++鎽勫儚鏈篿d涓猴細", msg.Cid, "---缂撳瓨姹犱负锛�", TrackPond) @@ -32,7 +33,7 @@ FaceIsInPond(msg.Cid, sdkinfo) } else { if num > 0 { // 杩炵画num娆℃病鏈夋暟鎹墠绠楁槸娌℃湁鏁版嵁锛屼笉鐒跺彧绠椾綔涓㈠抚 - logger.Info("鎴戣涓轰綘鍙槸涓㈠抚浜�,姝ゆ椂鐨刵um鍊间负锛�",num) + logger.Info("鎴戣涓轰綘鍙槸涓㈠抚浜�,姝ゆ椂鐨刵um鍊间负锛�", num) num-- } else { if TrackPond[msg.Cid] != nil { @@ -56,71 +57,71 @@ } // 杩借釜浜轰綋锛屾鏌ユ暟閲忔槸鍚︿竴鑷� -func BodyIsSame(msg *protomsg.SdkMessage) bool{ +func BodyIsSame(msg *protomsg.SdkMessage) bool { logger.Debug("+++++++++++++++++++++浜轰綋杩借釜寮�濮�+++++++++++++++鎽勫儚鏈篿d涓猴細", msg.Cid, "---缂撳瓨姹犱负锛�", TrackPond) for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 閬嶅巻鍚勭畻娉曠殑sdkData if sdkinfo.Sdktype == "Yolo" { - if len(sdkinfo.Sdkdata) > 1 { - yoloParam := protomsg.ParamYoloObj{} - err := proto.Unmarshal(sdkinfo.Sdkdata, &yoloParam) - if err != nil { - logger.Info("瑙f瀽yolo鏁版嵁鏃跺嚭鐜伴敊璇�", err) - continue - } - var yoloNum int = 0 - for i := 0; i < len(yoloParam.Infos); i++ { - if yoloParam.Infos[i].Typ == 0 { - yoloNum++ - } - } - if TrackPond[msg.Cid] != nil { - logger.Info("================杩借釜涔嬪墠yolo鐨勪釜鏁帮細", yoloNum, "鐜板湪缂撳瓨姹犱腑璁板綍鐨勪釜鏁帮細", TrackPond[msg.Cid].Yolo) - } else { - 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鏁版嵁") - 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 - } + if len(sdkinfo.Sdkdata) > 1 { + yoloParam := protomsg.ParamYoloObj{} + err := proto.Unmarshal(sdkinfo.Sdkdata, &yoloParam) + if err != nil { + logger.Info("瑙f瀽yolo鏁版嵁鏃跺嚭鐜伴敊璇�", err) + continue + } + var yoloNum int = 0 + for i := 0; i < len(yoloParam.Infos); i++ { + if yoloParam.Infos[i].Typ == 0 { + yoloNum++ + } + } + if TrackPond[msg.Cid] != nil { + logger.Info("================杩借釜涔嬪墠yolo鐨勪釜鏁帮細", yoloNum, "鐜板湪缂撳瓨姹犱腑璁板綍鐨勪釜鏁帮細", TrackPond[msg.Cid].Yolo) + } else { + 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鏁版嵁") + return true + } else { + if TrackPond[msg.Cid] != nil { + logger.Info("鏇存柊褰撳墠鎽勫儚鏈虹紦瀛樻睜涓殑yolo涓暟锛�", yoloNum) + lock.Lock() + TrackPond[msg.Cid].Yolo = yoloNum + lock.Unlock() } 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 + 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 } } } logger.Debug("---------------------------------浜轰綋杩借釜缁撴潫--------------------------------------") return false } -func TrackOrNot(label map[string]interface{}) bool{ +func TrackOrNot(label map[string]interface{}) bool { if label["yolo"] != nil && len(label["yolo"].([]structure.Result)) > 0 { - for _,res := range label["yolo"].([]structure.Result) { + for _, res := range label["yolo"].([]structure.Result) { if res.TimeLabel == "10" { return true } @@ -133,8 +134,8 @@ func FaceIsInPond(cameraId string, sdkinfor *protomsg.SdkmsgWithTask) { if TrackPond[cameraId] != nil { logger.Info("----椹墠鐐細", TrackPond[cameraId], "=====", len(TrackPond[cameraId].Faces)) - for _,face := range TrackPond[cameraId].Faces { - logger.Info("缂撳瓨涓瓨鍌ㄧ殑face鏁版嵁锛�",face.Id,face.Score) + for _, face := range TrackPond[cameraId].Faces { + logger.Info("缂撳瓨涓瓨鍌ㄧ殑face鏁版嵁锛�", face.Id, face.Score) } faceParam := protomsg.ParamFacePos{} err := proto.Unmarshal(sdkinfor.Sdkdata, &faceParam) @@ -142,12 +143,12 @@ logger.Info("瑙f瀽face sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err) } logger.Info("================杩借釜涔嬪墠浜鸿劯鐨勪釜鏁帮細", len(faceParam.Faces)) - for _,face := range faceParam.Faces { - logger.Info("鏂版潵鐨勭殑face鏁版嵁锛�",face.Pos.FaceID,face.Pos.FAngle.Confidence) + for _, face := range faceParam.Faces { + logger.Info("鏂版潵鐨勭殑face鏁版嵁锛�", face.Pos.FaceID, face.Pos.FAngle.Confidence) } var facesTemp []protomsg.ResultFaceDetect - for _,face := range faceParam.Faces { - facesTemp = append(facesTemp,*face) // 鍏堟妸鏁版嵁杞瓨涓�浠斤紝涓嶇劧涓�浼氬効鏁版嵁鍒犲噺涔嬪悗鎵句笉鍒板師濮嬫暟鎹紝涓嶈兘璁╃紦瀛樻暟鎹洿鏂颁簡 + for _, face := range faceParam.Faces { + facesTemp = append(facesTemp, *face) // 鍏堟妸鏁版嵁杞瓨涓�浠斤紝涓嶇劧涓�浼氬効鏁版嵁鍒犲噺涔嬪悗鎵句笉鍒板師濮嬫暟鎹紝涓嶈兘璁╃紦瀛樻暟鎹洿鏂颁簡 } for i := 0; i < len(faceParam.Faces); { @@ -156,7 +157,7 @@ if faceParam.Faces[i].Pos.FaceID == val.Id && faceParam.Faces[i].Pos.FAngle.Confidence <= val.Score { // 鍦ㄦ睜瀛愰噷骞朵笖鍒嗗�兼洿浣庯紝鏄鎶涘純鐨勬暟鎹� faceFlag = true //return "true" - logger.Info("鍒嗗�间负锛�",faceParam.Faces[i].Pos.FAngle.Confidence,"--缂撳瓨鐨勫垎鍊间负锛�",val.Score,"姝ゆ暟鎹敱浜庡湪姹犲瓙涓笖鍒嗗�兼洿浣庯紝鏄琚姏寮冪殑鏁版嵁") + logger.Info("鍒嗗�间负锛�", faceParam.Faces[i].Pos.FAngle.Confidence, "--缂撳瓨鐨勫垎鍊间负锛�", val.Score, "姝ゆ暟鎹敱浜庡湪姹犲瓙涓笖鍒嗗�兼洿浣庯紝鏄琚姏寮冪殑鏁版嵁") faceParam.Faces = append(faceParam.Faces[:i], faceParam.Faces[i+1:]...) break } @@ -174,10 +175,11 @@ i++ } } - logger.Info("鍙嶅簭鍒楀寲閲嶆柊瑁呴厤涔嬪墠浜鸿劯鐨勪釜鏁帮細",len(faceParam.Faces)) + logger.Info("鍙嶅簭鍒楀寲閲嶆柊瑁呴厤涔嬪墠浜鸿劯鐨勪釜鏁帮細", len(faceParam.Faces)) + logger.Info("涓存椂瀛樺偍鐨勬柊鏉ユ暟鎹細") for _, temp := range facesTemp { - logger.Info("涓存椂瀛樺偍鐨勬柊鏉ョ殑鐨刦ace鏁版嵁锛�",temp.Pos.FaceID,temp.Pos.FAngle.Confidence) + logger.Info("涓存椂瀛樺偍鐨勬柊鏉ョ殑鐨刦ace鏁版嵁锛�", temp.Pos.FaceID, temp.Pos.FAngle.Confidence) } sdkinfor.Sdkdata, err = proto.Marshal(&faceParam) if err != nil { @@ -203,7 +205,7 @@ } } else { lock.Lock() - TrackPond[cameraId] = &PersonTrack{Faces: nil} + TrackPond[cameraId] = &PersonTrack{Faces: nil} lock.Unlock() } } -- Gitblit v1.8.0