派生自 ruleprocess
ruleserver/ruleToformula.go
@@ -174,7 +174,7 @@ func SdkDataFormat(cameraId string, arg *SdkData, cameraPolygons []protomsg.CameraPolygon) { logger.Info("==================================本sdkData中解出来的目标数据=======================================") for _, photo := range arg.Photo { logger.Info("--------看看解析出来的数据---", cameraId, arg.IpcId, photo.Rects, photo.Score) logger.Info("--------解析出来的数据---", cameraId, arg.IpcId, photo.Rects, photo.Score) } for _, polygon := range cameraPolygons { //logger.Println("++++++在这儿看一下区域啊:", polygon.Polygon) @@ -234,15 +234,23 @@ } logger.Info("----------------------------------------map是", m) for key, val := range m { logger.Info("_______________________________________对比库的返回值", key, val) baseinfo, err1 := esutil.Dbpersoninfosbyid(key) if err1 != nil { logger.Error("查询底库人员信息出错", err1) } baseinfo.CompareScore = val baseinfo.CompareScore = Retain(val) arg.Liker = append(arg.Liker, &baseinfo) } //logger.Info("------------------第一次看args:",*arg) } // 保留四位小数 func Retain(f float32)(float32) { s:= strconv.FormatFloat(float64(f), 'f', 4, 64) v, err := strconv.ParseFloat(s, 32) if err != nil { logger.Error("保留四位小数转换错误") } return float32(v) } func Compare(args *SdkDatas, groupRule *protomsg.GroupRule) { compareFlag := 0 @@ -260,7 +268,7 @@ } } // 为了测试作下弊，直接让其比对全部底库 compareFlag = 1 //compareFlag = 1 if compareFlag == 0 { logger.Info("没有配置对比底库参数") return @@ -269,16 +277,19 @@ if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { for _, sdkData := range args.Sdkdata { if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" { // 搜索到人脸检测的sdkdata 现在关于人脸的只有他一个sdk，全找他 logger.Info("============================================第一关") logger.Info("============================================进行人脸对比") for _, areaMap := range sdkData.AreaMapList { logger.Info("-------------人脸比对之前目标数量",len(areaMap.args)) // 拿区域中每个人脸特征值去对比，填充其liker if groupRule.Rules[j].PolygonId == areaMap.areaId { logger.Info("--------------看看compareFlag的值：",compareFlag) for _, arg := range areaMap.args { if compareFlag == 1 { arg.fillLiker() } if compareFlag == 2 { array := strings.Split(tableIds, ",") logger.Info("--------------------------看看对比底库的值;",array) for i := 0; i < len(array)-1; i++ { arg.fillLiker() } @@ -288,6 +299,7 @@ areaMap.filterData = areaMap.args //logger.Info("=======第一次看args：",(areaMap.filterData)) } logger.Info("-------------------------------人脸对比之后的目标数量",len(areaMap.args)) } } } @@ -319,6 +331,9 @@ if ipcId == sdkData.IpcId { logger.Info("当前走的规则的算法是--：", sdkName, "---") logger.Info("========================有几个区域：",len(sdkData.AreaMapList)) for _,areaMap := range sdkData.AreaMapList{ logger.Info("检查一下区域的具体数据：",areaMap.areaJson) } for _, areaMap := range sdkData.AreaMapList { ruleResult := filterRule(groupRule.Rules[j], areaMap) if ruleResult.Result != "" { @@ -458,7 +473,6 @@ if flag { // 最后成功报警才把符合条件的人脸数据塞进结果标签里 // 配了人脸的算法才把人脸的数据甩出来打标签 logger.Info("------------------------------最后一关") faces := []Arg{} faceFlag := false for j := 0; j < len(groupRule.Rules); j++ { @@ -488,10 +502,10 @@ } if sdkNames != "" { args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText, locations,polygonId}) logger.Info("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result)) logger.Info("-------------------yolo结果标签长度", len(args.RuleResult["yolo"].([]Result))) } if args.RuleResult["face"] != nil { logger.Info("-------------------face结果标签", args.RuleResult["face"].([]Arg)) logger.Info("-------------------face结果标签", len(args.RuleResult["face"].([]Arg))) } return true } else { @@ -689,8 +703,7 @@ //logger.Info("规则的算法id和区域的算法id：", rule.SdkId, "===", am.sdkId) if rule.PolygonId == am.areaId { // 算法和区域都得对的上 if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && rule.SdkArgAlias != "time_rule" { // 只需要过滤阈值，过滤完后数组长度大于0即为报警 if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a972" && rule.SdkArgAlias != "time_rule" && rule.SdkArgAlias != "compareBase"{ argValue, err := strconv.ParseFloat(rule.SdkArgValue, 64) if err != nil { logger.Error("规则配置的阈值非法") @@ -725,11 +738,10 @@ return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort} } if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 人脸检测 logger.Info("------------------------------------------------------------------------------------------------------------------------------------终于等到你") if rule.Operator == "==" || rule.Operator == ">=" || rule.Operator == "<=" || rule.Operator == "<" || rule.Operator == ">" || rule.Operator == "!=" { // 如果是不规矩的连接符统统返回false 规则也只能判断人脸的相似度，所以不存在别的连接符 if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加 logger.Info("--------------------终于等到你的下文") logger.Info("---------------------------终于等到你") var args []*Arg if rule.RuleWithPre == "&&" { args = am.filterData @@ -738,7 +750,7 @@ } // 先清空过滤后的数据，再往里塞本次过滤后的数据 am.filterData = am.filterData[0:0] //logger.Info("-----------------------人脸过滤的args：", args) logger.Info("-----------------------人脸过滤的args里的数量：", len(args)) for _, arg := range args { var formula string if rule.SdkArgAlias == "score" {
