From 90757df2ff936c675b0eb5a5486b50a62e796a37 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期五, 25 十月 2019 18:19:24 +0800 Subject: [PATCH] --- --- util/simpleCV.go | 12 ++++++++---- 1 files changed, 8 insertions(+), 4 deletions(-) diff --git a/util/simpleCV.go b/util/simpleCV.go index f3f5b6d..9abc5bd 100644 --- a/util/simpleCV.go +++ b/util/simpleCV.go @@ -11,6 +11,7 @@ "ruleprocess/cache" "basic.com/valib/logger.git" "ruleprocess/ruleserver" + "ruleprocess/structure" "strings" ) @@ -155,7 +156,7 @@ } } -func DrawPolygonOnImageForYolo(cameraId string, img protomsg.Image, results []ruleserver.Result,url string) (maps map[string]interface{}, err0 error) { +func DrawPolygonOnImageForYolo(cameraId string, img protomsg.Image, results []structure.Result,url string) (maps map[string]interface{}, err0 error) { rook, _ := gocv.NewMatFromBytes(int(img.Height), int(img.Width), gocv.MatTypeCV8UC3, img.Data) //rook := gocv.IMRead("/home/user/workspace/ruleprocess/util/105.jpg",gocv.IMReadColor) @@ -168,7 +169,7 @@ for _,result := range results { polygonIds := strings.Split(result.AlarmPolygon,",") logger.Info("-----------------------鐪嬬湅鎶ヨ鍖哄煙id锛�",polygonIds) - for i := 0; i < len(polygonIds)-1; i++ { + for i := 0; i < len(polygonIds); i++ { polygon := getPolygonById(polygonIds[i],cameraId) if polygon.Polygon != "[]" && polygon.Polygon != ""{ logger.Debug("鎵�鐢诲尯鍩燂細",polygon.Polygon) @@ -187,7 +188,7 @@ return } -func DrawPolygonOnImageForFace(cameraId string, img protomsg.Image, results []ruleserver.FaceResult,url string) (maps map[string]interface{}, err0 error) { +func DrawPolygonOnImageForFace(cameraId string, img protomsg.Image, results []structure.FaceResult,url string) (maps map[string]interface{}, err0 error) { rook, _ := gocv.NewMatFromBytes(int(img.Height), int(img.Width), gocv.MatTypeCV8UC3, img.Data) //rook := gocv.IMRead("/home/user/workspace/ruleprocess/util/105.jpg",gocv.IMReadColor) @@ -199,8 +200,9 @@ for _,result := range results { polygonIds := strings.Split(result.AlarmPolygon,",") logger.Info("-----------------------鐪嬬湅鎶ヨ鍖哄煙id锛�",polygonIds) - for i := 0; i < len(polygonIds)-1; i++ { + for i := 0; i < len(polygonIds); i++ { polygon := getPolygonById(polygonIds[i],cameraId) + logger.Info("----鏌ュ埌鐨勬姤璀︽锛�",polygon) if polygon.Polygon != "[]" && polygon.Polygon != ""{ logger.Debug("鎵�鐢诲尯鍩燂細",polygon.Polygon) DrawAPolygon(&rook,polygon.Polygon,yellow) @@ -239,7 +241,9 @@ // 鏌ュ埌鎽勫儚鏈烘墍鏈夌殑鍖哄煙骞剁敾妗� var cameraPolygons []protomsg.CameraPolygon cameraPolygons = cache.GetPolygonsByCameraId(cameraId) + logger.Info("鏍规嵁id鏌ュ埌鐨勬姤璀︽锛�",cameraPolygons) for _, polygon := range cameraPolygons { + logger.Info("鏌ュ埌鐨勫尯鍩焛d锛�",polygon.Id,"--瑕佸尮閰嶇殑鏁版嵁id锛�",polygonId) if polygon.Id == polygonId { return polygon } -- Gitblit v1.8.0