| | |
| | | IsStatic bool // 是否静止 |
| | | ImageWidth int // 摄像机拍摄的图像宽 像素 |
| | | ImageHeight int // 摄像机拍摄的图像高 像素 |
| | | AreaMapList []*AreaMap // 本sdk提取的数据按照区域划分后的数据集合 |
| | | AreaMapList []*AreaMap // 本sdk提取的数据按照区域划分后的数据集合 |
| | | } |
| | | |
| | | // 从算法模块儿拿来的对一帧图像各个算法提取的数据集合 |
| | |
| | | // 根据id从map中拿到区域 |
| | | var cameraPolygons []*protomsg.CameraPolygon |
| | | for _, item := range data { |
| | | if item.CameraId == cameraId { |
| | | if item.CameraId == cameraId && item.Type != "line" { |
| | | // 需要根据比例把前台画的区域的坐标转化为相应摄像机拍摄的图像的大小 x坐标分别*image.width/页面区域宽 y坐标分别*image.height/页面区域高 |
| | | // 前台宽高固定 |
| | | if item.Type == "rect" { |
| | | cameraPolygons = append(cameraPolygons, &item) |
| | | } |
| | | cameraPolygons = append(cameraPolygons, &item) |
| | | } |
| | | } |
| | | //log.Println("根据摄像机id查到的区域", cameraPolygons, "--区域数量为:", len(cameraPolygons)) |
| | |
| | | // cameraPolygons := GetPolygons(arg.CameraId) |
| | | // list := AreaMapList{} |
| | | for _, polygon := range cameraPolygons { |
| | | log.Println("++++++在这儿看一下区域啊",polygon) |
| | | log.Println("++++++在这儿看一下区域啊", *polygon) |
| | | areaMap := &AreaMap{cameraId: cameraId, areaId: polygon.Id, areaJson: polygon.Polygon, triggerLine: polygon.TriggerLine, directionLine: polygon.DirectionLine} |
| | | // 为每个摄像机区域填充数据 |
| | | areaMap.CountAreaObjs(arg) |
| | |
| | | func RunRule(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string) bool { |
| | | resultSplice := []*LittleRuleResult{} |
| | | // 先过完条件规则 |
| | | log.Println("---------------------看下这个组规则:",groupRule.Rules) |
| | | log.Println("---------------------看下这个组规则:", groupRule.Rules) |
| | | for j := 0; j < len(groupRule.Rules); j++ { |
| | | for _, sdkData := range args.Sdkdata { |
| | | if sdkData.SdkId == "812b674b-2375-4589-919a-5c1c3278a975" { |
| | | log.Println("入侵的算法数据",sdkData) |
| | | log.Println("入侵的算法数据", sdkData) |
| | | } |
| | | for _, areaMap := range sdkData.AreaMapList { |
| | | ruleResult := filterRule(groupRule.Rules[j], areaMap) |
| | |
| | | sdkName := "" |
| | | for j := 0; j < len(groupRule.Rules); j++ { |
| | | for _, sdkData := range args.Sdkdata { |
| | | if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo{ // 只记录yolo符合规则的sdk |
| | | if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo { // 只记录yolo符合规则的sdk |
| | | sdkName = sdkName + "," + sdkData.SdkName |
| | | } |
| | | } |
| | |
| | | // } |
| | | for _, obj := range arg.Photo { |
| | | log.Println("-------------------看看areajson-------------", a.areaJson) |
| | | log.Println("------------------看看sdkData的Photo数据----------------",obj,"----顺便看看占比-----:",PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale)) |
| | | log.Println("------------------看看sdkData的Photo数据----------------", obj, "----顺便看看占比-----:", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale)) |
| | | if threshold <= obj.Score && size <= float64(obj.Rects.Width*obj.Rects.Height) && intersectionper <= PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale) { |
| | | // 这步要备齐表达式里所需要的所有参数 |
| | | a.targetNum++ |
| | |
| | | sdkName := "" |
| | | for j := 0; j < len(groupRule.Rules); j++ { |
| | | for _, sdkData := range args.Sdkdata { |
| | | if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo{ // 只记录yolo符合规则的sdk |
| | | if groupRule.Rules[j].SdkId == sdkData.SdkId && sdkData.IsYolo { // 只记录yolo符合规则的sdk |
| | | sdkName = sdkName + "," + sdkData.SdkName |
| | | } |
| | | } |
| | |
| | | } |
| | | } else { |
| | | // 处理的都是yolo数据 |
| | | log.Println("规则中的sdkId为:",rule.SdkId,"--区域数据集合中的sdkId为:",am.sdkId,"--规则中的区域Id为:",rule.PolygonId,"--区域数据集合中的区域Id为:",am.areaId,) |
| | | log.Println("规则中的sdkId为:", rule.SdkId, "--区域数据集合中的sdkId为:", am.sdkId, "--规则中的区域Id为:", rule.PolygonId, "--区域数据集合中的区域Id为:", am.areaId, ) |
| | | if rule.SdkId == am.sdkId && rule.PolygonId == am.areaId { // 首先这条规则得是这个算法的规则,其次规则所对应的区域id要跟区域数据的id对的上 |
| | | if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值,占比,尺寸等过滤条件,如果再有,还可以再加 |
| | | var args []Arg |