人脸多区域坐标去重

panlei

2019-07-29 ee960033e881a16f0243a349036b7fdac8975228

 insertdata/insertDataToEs.go

@@ -104,9 +104,10 @@
}

type AlarmRule struct {
   GroupId    string `json:"groupId"`
   AlarmLevel string `json:"alarmLevel"`
   RuleText   string `json:"ruleText"`
   GroupId      string `json:"groupId"`
   AlarmLevel   string `json:"alarmLevel"`
   RuleText     string `json:"ruleText"`
   DefenceState bool   `json:"defenceState"`
}

// 往ES插数据
@@ -126,10 +127,10 @@
      InsertYolo(msg)

   }
   if timeLabel == "12" { // 并非报警数据，只是状态改变的数据
      //ChangeStatusFace(msg)
      ChangeStatusYolo(msg)
   }
   //if timeLabel == "12" { // 并非报警数据，只是状态改变的数据
   //   //ChangeStatusFace(msg)
   //   ChangeStatusYolo(msg)
   //}
}

// 往es中插入人脸数据
@@ -145,8 +146,8 @@
               panic("解压缩图片时出现错误")
            }
            alarmRules := []AlarmRule{}
            alarm := ChangeToString(faceResult.DefenceState,faceResult.AlarmLevel)
            alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText})
            alarm := ChangeToString(faceResult.DefenceState, faceResult.AlarmLevel)
            alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText,faceResult.DefenceState})
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            bigPhotoUrl := make(map[string]interface{})
@@ -177,9 +178,9 @@
            }
            race := getRaceString(face.ThftRes.Race)
            ageDescription := getDescription(face.ThftRes.Age)

            esDataId := uuid.NewV4().String()
            pervideo := PerVideoPicture{
               uuid.NewV4().String(),
               esDataId,
               msg.Cid,
               camera.Addr,
               i.Timestamp,
@@ -222,7 +223,7 @@
            } else {
               logger.Info("插入es返回的信息：", resp1)
               // 发出录像信号
               ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:1})
               ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 1})
            }
         }
      }
@@ -253,8 +254,9 @@
      logger.Error("查询摄像机信息失败")
   }

   esDataId := uuid.NewV4().String()
   pervideo := PerVideoPicture{
      uuid.NewV4().String(),
      esDataId,
      msg.Cid,
      camera.Addr,
      i.Timestamp,
@@ -298,7 +300,7 @@
   } else {
      logger.Info("插入es返回的信息：", resp1)
      // 发出录像信号
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:1 })
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 1})
   }
   //if msg.RuleResult["cacheData"] != nil {
   //   InsertFace(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
@@ -313,8 +315,8 @@
      alarmRules := []AlarmRule{}
      for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
         sdkNames = sdkNames + yoloResult.SdkName
         alarm := ChangeToString(yoloResult.DefenceState,yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
         alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText,yoloResult.DefenceState})
      }
      isAlarm := 0
      resp := make(map[string]interface{})
@@ -372,8 +374,9 @@
         }
         url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1])
      }
      esDataId := uuid.NewV4().String()
      peraction := Personaction{
         uuid.NewV4().String(),
         esDataId,
         msg.Cid,
         camera.Name,
         camera.Addr,
@@ -407,7 +410,7 @@
      } else {
         logger.Debug("插入es返回的数据信息是：", resp1)
         // 发出录像信号
         ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:2 })
         ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 2})
         logger.Warn("__________________________________________往ES插入yolo数据成功")
         //os.Exit(1)
      }
@@ -442,8 +445,9 @@
      logger.Error("查询摄像机信息失败")
   }
   serverIp, err := GetLocalIP()
   esDataId := uuid.NewV4().String()
   peraction := Personaction{
      uuid.NewV4().String(),
      esDataId,
      msg.Cid,
      camera.Name,
      camera.Addr,
@@ -477,7 +481,7 @@
   } else {
      logger.Info("插入es返回的信息：", resp1)
      // 发出录像信号
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:2 })
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 2})
      logger.Warn("__________________________________________往ES插入yolo数据成功")
      //os.Exit(1)
   }
@@ -512,7 +516,7 @@
}

// 把报警等级转化成汉字
func ChangeToString(defenceState bool,i int32) string {
func ChangeToString(defenceState bool, i int32) string {
   alarm := ""
   if defenceState {
      if i == 1 {