panlei
2019-06-22 b2dbeaa7eb8ce43bc0b7553b49e9e4c7c342135d
过滤数据长度
1个文件已修改
15 ■■■■ 已修改文件
main.go 15 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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" { // 人脸提取