panlei
2019-07-03 11ee4d898c98d940943a005f8b6bcba9d04f7541
main.go
@@ -12,6 +12,7 @@
   "gocv.io/x/gocv"
   "image"
   "image/color"
   "log"
   "os"
   "ruleprocess/cache"
   "ruleprocess/insertdata"
@@ -21,8 +22,8 @@
)
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()
@@ -109,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{}
@@ -136,6 +137,7 @@
               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}
@@ -166,13 +168,13 @@
         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)
@@ -200,13 +202,13 @@
         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)