From f8d0f8954335b803a20d7cddbf4bc06e7810cbf5 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期五, 25 十月 2019 17:29:52 +0800 Subject: [PATCH] --- --- main.go | 8 ++++---- 1 files changed, 4 insertions(+), 4 deletions(-) diff --git a/main.go b/main.go index fa28b33..ddb2b73 100644 --- a/main.go +++ b/main.go @@ -11,12 +11,13 @@ "ruleprocess/util" "time" + "basic.com/valib/logger.git" //"bufio" //"bytes" "flag" "fmt" "github.com/golang/protobuf/proto" - "basic.com/valib/logger.git" + "github.com/spf13/viper" //"gocv.io/x/gocv" //"image" //"image/color" @@ -24,7 +25,6 @@ "ruleprocess/cache" "ruleprocess/ruleserver" "sync" - "github.com/spf13/viper" ) var dbIp = flag.String("dbIp", "127.0.0.1", "dbserver ip") @@ -117,7 +117,7 @@ // 灏嗗閮ㄤ紶杩涙潵鐨剅ect锛坱op,bottom,left,right锛夎浆鍖栦负鑷繁鍐呴儴鐨剅ect(left top width height) func rectFormat(rcobj *protomsg.Rect) structure.Rect { //logger.Info("++++++++++++++++++++++++++++++鏀跺埌鐨剏olo鐨勫尯鍩熷潗鏍囷細",rcobj) - rect := ruleserver.Rect{} + rect := structure.Rect{} rect.X = float64(rcobj.Left) rect.Y = float64(rcobj.Top) rect.Width = float64(rcobj.Right - rcobj.Left) @@ -183,7 +183,7 @@ for _, info := range yoloParam.Infos { if info.Typ == 0 { //logger.Debug("-------------yolo鐨勫潗鏍囨湁鍑犱釜",info.RcObj) - photoMap := structure.PhotoMap{Rects: rectFormat(info.RcObj), Score: float64(info.Prob) * 100, IsYolo: true} + photoMap := structure.PhotoMap{Rects: rectFormat(info.RcObj), Score: float64(info.Prob) * 100, IsYolo: true,Id:info.ObjID} arg.Photo = append(arg.Photo, photoMap) yoloNum++ } -- Gitblit v1.8.0