| | |
| | | "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} |
| | |
| | | arg.KeepRight = false |
| | | arg.IsStatic = false |
| | | if len(sdkinfo.Sdkdata) > 1 { |
| | | fmt.Println("----------------------------------------------------",m.Caddr) |
| | | faceParam := protomsg.ParamFacePos{} |
| | | err = proto.Unmarshal(sdkinfo.Sdkdata, &faceParam) |
| | | if err != nil { |
| | | fmt.Println("解析FACE sdk数据时出现错误", err) |
| | | continue |
| | | } |
| | | log.Println("------当前算法name",sdkinfo.SdkName,"--------------看看人脸检测中有几个目标:",len(faceParam.Faces)) |
| | | for _, info := range faceParam.Faces { |
| | | photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.Pos.RcFace), Score: float64(info.Pos.Quality)*100, IsYolo:false,ThftRes:*(info.Result)} |
| | | arg.Photo = append(arg.Photo, photoMap) |
| | |
| | | arg.KeepRight = false |
| | | arg.IsStatic = false |
| | | if len(sdkinfo.Sdkdata) > 1 { |
| | | fmt.Println("----------------------------------------------------",m.Caddr) |
| | | faceParam := protomsg.ParamFaceFeature{} |
| | | err = proto.Unmarshal(sdkinfo.Sdkdata, &faceParam) |
| | | if err != nil { |
| | | fmt.Println("解析FACE sdk数据时出现错误", err) |
| | | continue |
| | | } |
| | | log.Println("------当前算法name",sdkinfo.SdkName,"--------------看看人脸提取中有几个目标:",len(faceParam.ExtComp)) |
| | | for _, extComp := range faceParam.ExtComp { |
| | | baseinfos := protomsg.Binfos{} |
| | | err1 := proto.Unmarshal(extComp.Comp, &baseinfos) |