From 3b9f09718227dd179f9ca58d95e2f01ff91c77a8 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 11 九月 2019 14:23:26 +0800
Subject: [PATCH] change num 5 to 25
---
ruleserver/personTrack.go | 38 ++++++++++++++++++++++++++++++++------
1 files changed, 32 insertions(+), 6 deletions(-)
diff --git a/ruleserver/personTrack.go b/ruleserver/personTrack.go
index bd1ed57..17be27f 100644
--- a/ruleserver/personTrack.go
+++ b/ruleserver/personTrack.go
@@ -4,6 +4,7 @@
"basic.com/pubsub/protomsg.git"
"github.com/golang/protobuf/proto"
"ruleprocess/logger"
+ "ruleprocess/structure"
)
var TrackPond = make(map[string]*PersonTrack)
@@ -21,11 +22,15 @@
// 妫�鏌ユ槸鍚﹀墠鍚庝袱娆＄殑鏁版嵁id鏄惁瀹屽叏鐩稿悓锛堜汉鑴革級
func FaceIsSame(msg *protomsg.SdkMessage) {
logger.Debug("+++++++++++++++++++++杩借釜寮�濮�+++++++++++++++鎽勫儚鏈篿d涓猴細", msg.Cid, "---缂撳瓨姹犱负锛�", TrackPond)
+ if msg.Tasklab == nil {
+ logger.Info("鏁版嵁涓虹┖锛屼笉蹇呰拷韪�")
+ return
+ }
for _, sdkinfo := range msg.Tasklab.Sdkinfos { // 閬嶅巻鍚勭畻娉曠殑sdkData
if sdkinfo.Sdktype == "FaceDetect" { // 浜鸿劯妫�娴�
logger.Info("鏁版嵁闀垮害涓猴細", len(sdkinfo.Sdkdata))
if len(sdkinfo.Sdkdata) > 1 {
- num = 5
+ num = 25
FaceIsInPond(msg.Cid, sdkinfo)
} else {
if num > 0 { // 杩炵画num娆℃病鏈夋暟鎹墠绠楁槸娌℃湁鏁版嵁锛屼笉鐒跺彧绠椾綔涓㈠抚
@@ -103,17 +108,33 @@
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) {
+ if res.TimeLabel == "10" {
+ return true
+ }
+ }
+ }
+ return false
+}
// 杩囨护鎺夐偅浜涘凡鍦ㄧ紦瀛樹腑涓斿垎鍊兼洿浣庣殑浜鸿劯锛屾洿鏂扮紦瀛橈紙娌℃湁鐨勫姞涓婏紝鍒嗗�兼洿鏂颁负鏇撮珮鐨勶紝澶氱殑鍒犻櫎锛�
func FaceIsInPond(cameraId string, sdkinfor *protomsg.SdkmsgWithTask) string {
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)
+ }
faceParam := protomsg.ParamFacePos{}
err := proto.Unmarshal(sdkinfor.Sdkdata, &faceParam)
if err != nil {
logger.Info("瑙ｆ瀽face sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err)
}
logger.Info("================杩借釜涔嬪墠浜鸿劯鐨勪釜鏁帮細", len(faceParam.Faces))
+ for _,face := range faceParam.Faces {
+ logger.Info("鏂版潵鐨勭殑face鏁版嵁锛�",face.Pos.FaceID,face.Pos.FAngle.Confidence)
+ }
var facesTemp = faceParam.Faces // 鍏堟妸鏁版嵁杞瓨涓�浠斤紝涓嶇劧涓�浼氬効鏁版嵁鍒犲噺涔嬪悗鎵句笉鍒板師濮嬫暟鎹紝涓嶈兘璁╃紦瀛樻暟鎹洿鏂颁簡
for i := 0; i < len(faceParam.Faces); {
faceFlag := false
@@ -132,18 +153,23 @@
}
}
if !faceFlag { // 姝や汉鑴镐笉鍦ㄦ睜瀛愪腑
- logger.Info("娣诲姞鏁版嵁鏇存柊缂撳瓨")
+ logger.Info("姝や汉鑴镐笉鍦ㄦ睜瀛愪腑锛屽線姹犲瓙涓坊鍔犳暟鎹洿鏂扮紦瀛�")
TrackPond[cameraId].Faces = append(TrackPond[cameraId].Faces, &Face{faceParam.Faces[i].Pos.FaceID, faceParam.Faces[i].Pos.FAngle.Confidence})
i++
}
}
logger.Info("鍙嶅簭鍒楀寲閲嶆柊瑁呴厤涔嬪墠浜鸿劯鐨勪釜鏁帮細",len(faceParam.Faces))
+
sdkinfor.Sdkdata, err = proto.Marshal(&faceParam)
if err != nil {
- logger.Error("yolo搴忓垪鍖栭敊璇�", err)
+ logger.Error("face搴忓垪鍖栭敊璇�", err)
}
- logger.Info("璺熶箣鍓嶇浉鍚岋紝娓呯┖yolo鏁版嵁")
- // 鍙嶅悜寰幆 锛岀湅閭ｄ簺缂撳瓨鏈夎�屾暟鎹病鏈夌殑灏卞垹闄ゆ暟鎹洿鏂扮紦瀛�
+ //logger.Info("璺熶箣鍓嶇浉鍚岋紝娓呯┖浜鸿劯鏁版嵁")
+ // 鍙嶅悜寰幆 锛岀湅閭ｄ簺缂撳瓨鏈夎�屾暟鎹病鏈夌殑灏卞垹闄ゆ睜瀛愪腑鐨勬暟鎹洿鏂扮紦瀛�
+ logger.Info("涓存椂瀛樺偍鐨勬柊鏉ユ暟鎹細")
+ for _, temp := range facesTemp {
+ logger.Info("涓存椂瀛樺偍鐨勬柊鏉ョ殑鐨刦ace鏁版嵁锛�",temp.Pos.FaceID,temp.Pos.FAngle.Confidence)
+ }
for i := 0; i < len(TrackPond[cameraId].Faces); {
flag := false
for _, temp := range facesTemp {
@@ -155,7 +181,7 @@
i++
} else {
// 姝ゆ暟鎹湪缂撳瓨涓絾涓嶅湪鏉ョ殑鏁版嵁甯т腑锛屽垹闄ゆ鏁版嵁鏇存柊缂撳瓨
- logger.Info("鍒犻櫎鏁版嵁鏇存柊缂撳瓨")
+ logger.Info("鍒犻櫎姹犲瓙涓殑鏁版嵁鏇存柊缂撳瓨")
TrackPond[cameraId].Faces = append(TrackPond[cameraId].Faces[:i], TrackPond[cameraId].Faces[i+1:]...)
}
}
--
Gitblit v1.8.0