调整yolo报警策略

panlei

2019-07-02 04dfa328304ea805d93d04b577058f07b669bf14

 insertdata/insertDataToEs.go

@@ -90,7 +90,7 @@

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

@@ -150,7 +150,7 @@
               face.ThftRes.Smile,
               face.ThftRes.Beauty,
               "不是每个人脸算法都有",
               resp["fileUrl"].(string),
               strings.Split(resp["fileUrl"].(string), "/")[1],
               "暂无集群",
               localConfig.ServerId,
               localConfig.ServerName,
@@ -172,6 +172,8 @@
            err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
            if err != nil {
               log.Println("es can not execute right.")
            } else {
               log.Println("___________________________________往ES插入人脸数据成功")
            }
         }else {
            log.Println("跟底库的相似数据---------：",face.Liker)
@@ -184,7 +186,8 @@
      alarmRules := []AlarmRule{}
      for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
         sdkNames = sdkNames + yoloResult.SdkName
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, yoloResult.AlarmLevel, yoloResult.RuleText})
         alarm := ChangeToString(yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
      }
      isAlarm := ""
      resp := make(map[string]interface{})
@@ -198,7 +201,7 @@
         i := protomsg.Image{}
         err = proto.Unmarshal(bdata, &i)
         //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
         resp, err =util.DrawPolygonOnImage(i)
         resp, err =util.DrawPolygonOnImage(msg.Cid,i)
         if err != nil {
            log.Println("画框或上传图片服务器出错",err)
         }else{
@@ -225,7 +228,7 @@
         uuid.NewV4().String(),
         msg.Cid,
         camera.Name,
         msg.Caddr,
         camera.Addr,
         msg.Tasklab.Taskid,
         msg.Tasklab.Taskname,
         sdkNames,
@@ -254,7 +257,7 @@
      if err != nil {
         log.Println("往ES插入数据失败",err)
      } else {
         log.Println("__________________________________________往ES插入数据成功")
         log.Println("__________________________________________往ES插入数据yolo成功")
      }
   }
}
@@ -286,3 +289,24 @@
   err = errors.New("ipv4 not found")
   return
}

// 把报警等级转化成汉字
func ChangeToString(i int32) string{
   alarm := ""
   if i == 1 {
      alarm = "一级"
   }
   if i == 2 {
      alarm = "二级"
   }
   if i == 3 {
      alarm = "三级"
   }
   if i == 4 {
      alarm = "四级"
   }
   if i == 5 {
      alarm = "五级"
   }
   return alarm
}