From 5827190e783d7b9c3eb7c3ee2c36a14d85310e5a Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期五, 09 八月 2019 14:05:14 +0800 Subject: [PATCH] 只用一个核 --- labelFilter/readyDataForLabel.go | 20 ++++++++++++++------ 1 files changed, 14 insertions(+), 6 deletions(-) diff --git a/labelFilter/readyDataForLabel.go b/labelFilter/readyDataForLabel.go index 136e487..2650c4e 100644 --- a/labelFilter/readyDataForLabel.go +++ b/labelFilter/readyDataForLabel.go @@ -18,7 +18,10 @@ } // 鎶婃暟鎹閰嶅埌label func (label *Label)DataFormatToLabel(result ruleserver.ResultMsg) { + 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 { @@ -27,14 +30,19 @@ label.CameraName = camera.Name label.CameraAddr = camera.Addr label.TaskId = result.Tasklab.Taskid - label.Time = time.Now().String() - for _,faceGroup := range result.RuleResult["face"].([]ruleserver.FaceResult) { - label.Persons = append(label.Persons,faceGroup) - label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel) + 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) + } } - for _,yoloGroup := range result.RuleResult["yolo"].([]ruleserver.Result) { - label.DefenceLevel = append(label.DefenceLevel,yoloGroup.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("------------鏍囩杩囨护鍣ㄨ祴鍊煎畬姣�") } } -- Gitblit v1.8.0