From 355284a3ac944515951d4c2a7ec5d3964fb13857 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期四, 31 十月 2019 13:48:29 +0800
Subject: [PATCH] 把人体追踪从es最前改为放在规则最后

---
 main.go |   10 ++++++----
 1 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 61e8405..7f9a5d6 100644
--- a/main.go
+++ b/main.go
@@ -7,16 +7,18 @@
 	"net/http"
 	_ "net/http/pprof"
 	"ruleprocess/insertdata"
+	"ruleprocess/labelFilter"
 	"ruleprocess/structure"
 	"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 +26,6 @@
 	"ruleprocess/cache"
 	"ruleprocess/ruleserver"
 	"sync"
-	"github.com/spf13/viper"
 )
 
 var dbIp = flag.String("dbIp", "127.0.0.1", "dbserver ip")
@@ -62,6 +63,7 @@
 	go cache.Init(initchan, *dbIp, *surveyPort, *pubPort)
 	logger.Info("cache init completed!!!", <-initchan) //dbserver鍒濆鍖栧畬姣�
 	ruleserver.Init()
+	labelFilter.Init()
 	go ruleserver.TimeTicker()
 	go ruleserver.StartServer()
 	nReciever("ipc:///tmp/sdk-2-rules-process.ipc", deliver.PushPull, 1)
@@ -106,7 +108,7 @@
 					insertdata.InsertToEs(resultMsg)
 					logger.Debug("鎻掑叆瀹孍s鎵�鐢ㄦ椂闂达細", time.Since(start))
 					//浜嬩欢鎺ㄩ��
-					//go labelFilter.Judge(resultMsg)
+					go labelFilter.PushSomthing(resultMsg)
 				//}(msg)
 			}
 		}
@@ -183,7 +185,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