From f8d0f8954335b803a20d7cddbf4bc06e7810cbf5 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期五, 25 十月 2019 17:29:52 +0800
Subject: [PATCH] ---

---
 util/simpleCV.go |   11 +++++++----
 1 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/util/simpleCV.go b/util/simpleCV.go
index f3f5b6d..082100d 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,6 +241,7 @@
 	// 鏌ュ埌鎽勫儚鏈烘墍鏈夌殑鍖哄煙骞剁敾妗�
 	var cameraPolygons []protomsg.CameraPolygon
 	cameraPolygons = cache.GetPolygonsByCameraId(cameraId)
+	logger.Info("鏍规嵁id鏌ュ埌鐨勬姤璀︽锛�",cameraPolygons)
 	for _, polygon := range cameraPolygons {
 		if polygon.Id == polygonId {
 			return polygon

--
Gitblit v1.8.0