| | |
| | | } |
| | | var labelTypes []int // 0为yolo标签,1为face标签 2为两者标签 |
| | | if sdkNames != "" { |
| | | args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, islink,label,}) |
| | | args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]structure.Result), structure.Result{taskId, sdkNames, groupRule.GroupId, groupRule.DefenceState, groupRule.AlarmLevel, groupRule.GroupText, locations, polygonId, islink,label,}) |
| | | labelTypes = append(labelTypes,0) |
| | | //logger.Info("-------------------yolo结果标签长度", len(args.RuleResult["yolo"].([]Result))) |
| | | } |
| | |
| | | // 倒是把打的组规则标签给去掉了啊 |
| | | for _,val := range labelTypes { |
| | | if val == 0 { |
| | | if len(args.RuleResult["yolo"].([]Result)) >= 1 { |
| | | lens := len(args.RuleResult["yolo"].([]Result))-1 |
| | | args.RuleResult["yolo"] = args.RuleResult["yolo"].([]Result)[0:lens] |
| | | if len(args.RuleResult["yolo"].([]structure.Result)) >= 1 { |
| | | lens := len(args.RuleResult["yolo"].([]structure.Result))-1 |
| | | args.RuleResult["yolo"] = args.RuleResult["yolo"].([]structure.Result)[0:lens] |
| | | } |
| | | } |
| | | if val == 1 { |
| | | if len(args.RuleResult["face"].([]FaceResult)) >= 1 { |
| | | lens := len(args.RuleResult["face"].([]FaceResult))-1 |
| | | args.RuleResult["face"] = args.RuleResult["face"].([]FaceResult)[0:lens] |
| | | if len(args.RuleResult["face"].([]structure.FaceResult)) >= 1 { |
| | | lens := len(args.RuleResult["face"].([]structure.FaceResult))-1 |
| | | args.RuleResult["face"] = args.RuleResult["face"].([]structure.FaceResult)[0:lens] |
| | | } |
| | | } |
| | | } |