| | |
| | | "ruleprocess/insertdata" |
| | | "ruleprocess/ruleserver" |
| | | "ruleprocess/util" |
| | | "ruleprocess/logger" |
| | | "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 surveyPort = flag.Int("surveyPort",40007,"survey port") //心跳 |
| | | var pubPort = flag.Int("pubPort",50007,"pubsub port") //数据更新 |
| | | var initchan = make(chan bool) |
| | | func init () { |
| | | var logFile = "/opt/ruleLog/error.log" |
| | | var logSaveDays = 15 |
| | | |
| | | // 日志初始化 |
| | | logger.Config(logFile, logger.DebugLevel) |
| | | logger.SetSaveDays(logSaveDays) |
| | | logger.Info("loginit success !") |
| | | } |
| | | func main() { |
| | | flag.Parse() |
| | | wg := sync.WaitGroup{} |
| | |
| | | wg.Wait() |
| | | //resp,err := DrawPolygonOnImage() |
| | | //if err != nil { |
| | | // log.Println("画框或者上传有问题") |
| | | // logger.Println("画框或者上传有问题") |
| | | //} |
| | | //log.Println(resp["fileUrl"].(string)) |
| | | //logger.Println(resp["fileUrl"].(string)) |
| | | } |
| | | func nReciever(url string, m deliver.Mode, count int) { |
| | | c := deliver.NewServer(m, url) |
| | |
| | | } |
| | | i := protomsg.Image{} |
| | | err = proto.Unmarshal(bdata, &i) |
| | | |
| | | logger.Info("----------------看看有几个算法:",len(m.Tasklab.Sdkinfos)) |
| | | for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo算法 |
| | | if sdkinfo.Sdktype == "Yolo" { |
| | | arg := ruleserver.SdkData{} |
| | |
| | | fmt.Println("解析YOLO sdk数据时出现错误", err) |
| | | continue |
| | | } |
| | | logger.Info("------当前算法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 |
| | | } |
| | | logger.Info("------当前算法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 |
| | | } |
| | | for _,faceExtCom := range faceParam.ExtComp { |
| | | binfos := protomsg.Binfos{} |
| | | err := proto.Unmarshal(faceExtCom.Comp, &binfos) |
| | | if err != nil { |
| | | fmt.Println("解析人脸比对数据时出现错误", err) |
| | | continue |
| | | } |
| | | logger.Info("_________________________________________________________________解析出的人脸的相似的底库数据",binfos) |
| | | } |
| | | logger.Info("------当前算法name",sdkinfo.SdkName,"--------------看看人脸提取中有几个目标:",len(faceParam.ExtComp)) |
| | | for _, extComp := range faceParam.ExtComp { |
| | | baseinfos := protomsg.Binfos{} |
| | | err1 := proto.Unmarshal(extComp.Comp, &baseinfos) |
| | |
| | | //client := &http.Client{Timeout: timeout} |
| | | //resp, err := client.Do(request) |
| | | //if err != nil { |
| | | // log.Fatal(err) |
| | | // logger.Fatal(err) |
| | | // return nil, err |
| | | //} |
| | | //defer func() { |
| | |
| | | // body := &bytes.Buffer{} |
| | | // _, err := body.ReadFrom(resp.Body) |
| | | // if err != nil { |
| | | // log.Fatal(err) |
| | | // logger.Fatal(err) |
| | | // } |
| | | // fmt.Println(resp.StatusCode) |
| | | // //fmt.Println(resp.Header) |