From 7b20cb1a629526553ed640c42108fe3fba203040 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期五, 27 九月 2019 14:17:28 +0800 Subject: [PATCH] logger改用公共库 --- labelFilter/readyDataForLabel.go | 41 +++++++++++++++++++++++++++++------------ 1 files changed, 29 insertions(+), 12 deletions(-) diff --git a/labelFilter/readyDataForLabel.go b/labelFilter/readyDataForLabel.go index 143302f..67888f0 100644 --- a/labelFilter/readyDataForLabel.go +++ b/labelFilter/readyDataForLabel.go @@ -2,8 +2,9 @@ import ( "ruleprocess/cache" - "ruleprocess/logger" + "basic.com/valib/logger.git" "ruleprocess/ruleserver" + "time" ) type Label struct { @@ -12,21 +13,37 @@ CameraAddr string TaskId string Time string - Persons []ruleserver.Arg - DefenceLevel string + Persons []ruleserver.FaceResult + DefenceLevel []int32 } // 鎶婃暟鎹閰嶅埌label func (label *Label)DataFormatToLabel(result ruleserver.ResultMsg) { - label.CameraId = result.Cid - camera, err := cache.GetCameraById(result.Cid) - if err != nil { - logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭嚭閿�") + logger.Debug("face缁撴灉锛�",result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0) + logger.Debug("yolo缁撴灉锛�",result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0) + if (result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0) || (result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0) { + logger.Info("---------------鏍囩杩囨护鍣ㄨ祴鍊�") + label.CameraId = result.Cid + camera, err := cache.GetCameraById(result.Cid) + if err != nil { + logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭嚭閿�") + } + label.CameraName = camera.Name + label.CameraAddr = camera.Addr + label.TaskId = result.Tasklab.Taskid + label.Time = time.Now().Format("2006-01-02 15:04:05") + if result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0 { + for _,faceGroup := range result.RuleResult["face"].([]ruleserver.FaceResult) { + label.Persons = append(label.Persons,faceGroup) + label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel) + } + } + if result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0 { + for _,yoloGroup := range result.RuleResult["yolo"].([]ruleserver.Result) { + label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel) + } + } + logger.Info("------------鏍囩杩囨护鍣ㄨ祴鍊煎畬姣�") } - label.CameraName = camera.Name - label.CameraAddr = camera.Addr - label.TaskId = result.Tasklab.Taskid - label.Persons = result.RuleResult["face"].([]ruleserver.Arg) - //labelData.DefenceLevel = result. } // 浠庣紦瀛樹腑鏌ュ嚭鎵�鏈夌殑瑙勫垯鏁版嵁 -- Gitblit v1.8.0