From c0d0a8df1bbe1cf0cb663d9a2732f5575572c4d7 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期三, 10 七月 2019 11:44:43 +0800 Subject: [PATCH] 重要更新,塞符合条件的人脸标签、 --- ruleserver/ruleToformula.go | 70 ++++++++++++++-------------------- 1 files changed, 29 insertions(+), 41 deletions(-) diff --git a/ruleserver/ruleToformula.go b/ruleserver/ruleToformula.go index 68f5380..8be94d4 100644 --- a/ruleserver/ruleToformula.go +++ b/ruleserver/ruleToformula.go @@ -185,8 +185,8 @@ // 鎶妔dk浠庢暟鎹抚涓婃彁鍙栫殑鎸夌収鍖哄煙鍒嗙被褰掔疆 func SdkDataFormat(cameraId string, arg *SdkData, cameraPolygons []protomsg.CameraPolygon) { logger.Info("==================================鏈瑂dkData涓В鍑烘潵鐨勭洰鏍囨暟鎹�=======================================") - for _,photo := range arg.Photo{ - logger.Info("--------鐪嬬湅瑙f瀽鍑烘潵鐨勬暟鎹�---", cameraId,arg.IpcId,photo.Rects,photo.Score) + for _, photo := range arg.Photo { + logger.Info("--------鐪嬬湅瑙f瀽鍑烘潵鐨勬暟鎹�---", cameraId, arg.IpcId, photo.Rects, photo.Score) } for _, polygon := range cameraPolygons { //logger.Println("++++++鍦ㄨ繖鍎跨湅涓�涓嬪尯鍩熷晩:", polygon.Polygon) @@ -258,7 +258,7 @@ } for j := 0; j < len(groupRule.Rules); j++ { // 鍙湁鏄汉鑴稿姣斿苟涓旈厤缃殑鍙傛暟鏄姣斿簳搴撴墠璧帮紙鑻ラ厤鐨勫姣旈槇鍊煎垯涓嶈蛋锛� - if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972"{ + if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 鎼滅储鍒颁汉鑴告娴嬬殑sdkdata 鐜板湪鍏充簬浜鸿劯鐨勫彧鏈変粬涓�涓猻dk锛屽叏鎵句粬 for _, areaMap := range sdkData.AreaMapList { @@ -273,9 +273,9 @@ logger.Error("getBaseInfo瑙e帇閿欒", err1) } for key, val := range m { - baseinfo,err1 := esutil.Dbpersoninfosbyid(key) + baseinfo, err1 := esutil.Dbpersoninfosbyid(key) if err1 != nil { - logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊",err1) + logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊", err1) } baseinfo.CompareScore = val arg.Liker = append(arg.Liker, &baseinfo) @@ -293,7 +293,7 @@ for key, val := range m { baseinfo, err1 := esutil.Dbpersoninfosbyid(key) if err1 != nil { - logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊",err1) + logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊", err1) } baseinfo.CompareScore = val arg.Liker = append(arg.Liker, &baseinfo) @@ -449,21 +449,21 @@ } else { // 鏈�鍚庢垚鍔熸姤璀︽墠鎶婄鍚堟潯浠剁殑浜鸿劯鏁版嵁濉炶繘缁撴灉鏍囩閲� faces := []Arg{} + faceFlag := false + for j := 0; j < len(groupRule.Rules); j++ { + if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { + faceFlag = true + } + } for _, sdkData := range args.Sdkdata { - for _, areaMap := range sdkData.AreaMapList { - for j := 0; j < len(groupRule.Rules); j++ { - faces = append(faces, putFaceToResult(groupRule.Rules[j], &areaMap)...) + if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" && faceFlag { // sdkData閲屾湁浜鸿劯鏁版嵁涓旈厤缃簡绠楁硶鎵嶆妸绗﹀悎鏉′欢鐨勬暟鎹杩涙爣绛鹃噷鍘� + for _, areaMap := range sdkData.AreaMapList { + faces = append(faces, putFaceToResult(&areaMap)...) } } } // 閰嶄簡浜鸿劯鐨勭畻娉曟墠鎶婁汉鑴哥殑鏁版嵁鐢╁嚭鏉ユ墦鏍囩 - flag11 := false - for j := 0; j < len(groupRule.Rules); j++ { - if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { - flag11 = true - } - } - if flag11 { + if faceFlag { args.RuleResult["face"] = faces } // 鍘荤湅姹犲瓙閲屾槸鍚︽湁涓庢湰甯ф暟鎹湁鍏崇殑瀹氭椂鍣紝濡傛灉鏈夛紝鐪嬫鏃舵槸鍚﹁蛋鍒�0锛屾病鏈夋瀹氭椂鍣ㄦ垨鏈夊畾鏃跺櫒璧板埌0鐨勮瘽杩斿洖鎴愬姛鎶ヨ @@ -481,22 +481,12 @@ // 鎶婁粬浠殑浣嶇疆鏁版嵁涔熶紶涓嬪幓 locations := []Rect{} for _, sdkData := range args.Sdkdata { - for _, areaMap := range sdkData.AreaMapList { - for j := 0; j < len(groupRule.Rules); j++ { - locations = append(locations, putYolosToResult(groupRule.Rules[j], &areaMap)...) + if sdkData.IpcId == "02D54B61-0F16-C604-8567-FC4BE493C523" && sdkNames != "" { // 鎶妝olo鏁版嵁鐨勫悇涓洰鏍囩殑鍧愭爣杈撳嚭鏂逛究鍚庨潰鐢绘 + for _, areaMap := range sdkData.AreaMapList { + locations = append(locations, putYolosToResult(&areaMap)...) } } } - //// 閰嶄簡浜鸿劯鐨勭畻娉曟墠鎶婁汉鑴哥殑鏁版嵁鐢╁嚭鏉ユ墦鏍囩 - //flag11 := false - //for j := 0; j < len(groupRule.Rules); j++ { - // if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a971" { - // flag11 = true - // } - //} - //if flag11 { - // args.RuleResult["face"] = faces - //} if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText, locations}) logger.Info("-------------------yolo缁撴灉鏍囩", args.RuleResult["yolo"].([]Result)) @@ -514,26 +504,23 @@ } } -func putFaceToResult(rule *protomsg.Rule, am *AreaMap) []Arg { +func putFaceToResult(am *AreaMap) []Arg { faces := []Arg{} - if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { - if len(am.filterData) > 0 { - for _, data := range am.filterData { - faces = append(faces, data) - } + if len(am.filterData) > 0 { + for _, data := range am.filterData { + faces = append(faces, data) } } + //logger.Println("-----------------------------------------------鍚浣犳槸绌虹殑锛�",faces) return faces } -func putYolosToResult(rule *protomsg.Rule, am *AreaMap) []Rect { +func putYolosToResult(am *AreaMap) []Rect { locations := []Rect{} - if rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a97e" && rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a972" { - if len(am.filterData) > 0 { - for _, data := range am.filterData { - locations = append(locations, data.Location) - } + if len(am.filterData) > 0 { + for _, data := range am.filterData { + locations = append(locations, data.Location) } } //logger.Println("-----------------------------------------------鍚浣犳槸绌虹殑锛�",faces) @@ -752,6 +739,7 @@ } // 鍏堟竻绌鸿繃婊ゅ悗鐨勬暟鎹紝鍐嶅線閲屽鏈杩囨护鍚庣殑鏁版嵁 am.filterData = am.filterData[0:0] + logger.Info("-----------------------浜鸿劯杩囨护鐨刟rgs锛�", args) for _, arg := range args { var formula string if rule.SdkArgAlias == "score" { -- Gitblit v1.8.0