panlei
2019-06-22 b2dbeaa7eb8ce43bc0b7553b49e9e4c7c342135d
main.go
@@ -2,13 +2,13 @@
import (
   "fmt"
   "ruleprocess/ruleserver"
   "ruleprocess/insertdata"
   "ruleprocess/ruleserver"
   "sync"
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/deliver.git"
   "github.com/golang/protobuf/proto"
   "sync"
)
func main() {
@@ -83,6 +83,9 @@
   fmt.Println("从mongos中拿到的数据包长度为:", len(msg))
   for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo算法
      if sdkinfo.Sdktype == "Yolo" {
         fmt.Println("数据长度为:", len(sdkinfo.Sdkdata))
         if len(sdkinfo.Sdkdata) > 1 {
            // 大于1才有数据
         yoloParam := protomsg.ParamYoloObj{}
         err = proto.Unmarshal(sdkinfo.Sdkdata, &yoloParam)
         if err != nil {
@@ -93,9 +96,14 @@
            photoMap := ruleserver.PhotoMap{Rects:rectFormat(info.RcObj), Score: float64(info.Prob)}
            arg.Photo = append(arg.Photo, photoMap)
         }
         } else {
            continue
         }
      }
      if sdkinfo.Sdktype == "FaceDetect" { // 人脸检测
         fmt.Println("数据长度为:", len(sdkinfo.Sdkdata))
         if len(sdkinfo.Sdkdata) > 1 {
         faceParam := protomsg.ParamFacePos{}
         err = proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
         if err != nil {
@@ -106,6 +114,9 @@
            photoMap := ruleserver.PhotoMap{Rects:ruleserver.Rect{-1,-1,-1,-1}, Score: float64(info.Pos.Quality)}
            arg.Photo = append(arg.Photo, photoMap)
         }
         } else {
            continue
         }
      }
      //if sdkinfo.Sdktype == "FaceExtract" { // 人脸提取