From aafaf19184a4d3ef5bb88a8613d9bc9d7c1f27e5 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期二, 02 七月 2019 13:52:36 +0800 Subject: [PATCH] --- --- main.go | 19 +++++++++++++++++-- 1 files changed, 17 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index 19d3756..e82a43a 100644 --- a/main.go +++ b/main.go @@ -1,25 +1,39 @@ package main import ( + "basic.com/dbapi.git" "basic.com/pubsub/protomsg.git" "basic.com/valib/deliver.git" "bufio" "bytes" + "flag" "fmt" "github.com/golang/protobuf/proto" "gocv.io/x/gocv" "image" "image/color" + "log" "os" + "ruleprocess/cache" "ruleprocess/insertdata" "ruleprocess/ruleserver" "ruleprocess/util" "sync" ) - +var dbIp = flag.String("dbIp","127.0.0.1","dbserver ip") +var dbPort = flag.Int("dbPort",8001,"default dbPort=8001") +var surveyPort = flag.Int("surveyPort",40007,"survey port") //蹇冭烦 +var pubPort = flag.Int("pubPort",50007,"pubsub port") //鏁版嵁鏇存柊 +var initchan = make(chan bool) func main() { + flag.Parse() wg := sync.WaitGroup{} wg.Add(1) + + dbapi.Init(*dbIp,*dbPort) + go cache.Init(initchan,*dbIp,*surveyPort,*pubPort) + fmt.Println("cache init completed!!!",<- initchan)//dbserver鍒濆鍖栧畬姣� + go ruleserver.TimeTicker() nReciever("ipc:///tmp/sdk-2-rules-process.ipc", deliver.PushPull, 1) wg.Wait() @@ -96,7 +110,7 @@ } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) - + log.Println("----------------鐪嬬湅鏈夊嚑涓畻娉曪細",len(m.Tasklab.Sdkinfos)) for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo绠楁硶 if sdkinfo.Sdktype == "Yolo" { arg := ruleserver.SdkData{} @@ -123,6 +137,7 @@ fmt.Println("瑙f瀽YOLO sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err) continue } + log.Println("------褰撳墠绠楁硶name",sdkinfo.SdkName,"--------------鐪嬬湅yolo涓湁鍑犱釜鐩爣锛�",len(yoloParam.Infos)) for _, info := range yoloParam.Infos { if info.Typ == 0 { photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.RcObj), Score: float64(info.Prob)*100,IsYolo:true} -- Gitblit v1.8.0