| | |
| | | "gocv.io/x/gocv" |
| | | "image" |
| | | "image/color" |
| | | "log" |
| | | "os" |
| | | "ruleprocess/cache" |
| | | "ruleprocess/insertdata" |
| | |
| | | ) |
| | | 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 surveyPort = flag.Int("surveyPort",40007,"survey port") //心跳 |
| | | var pubPort = flag.Int("pubPort",50007,"pubsub port") //数据更新 |
| | | var initchan = make(chan bool) |
| | | func main() { |
| | | flag.Parse() |
| | |
| | | } |
| | | 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{} |
| | |
| | | fmt.Println("解析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} |