liuxiaolong
2019-06-27 906c2c219c73b90eae9e5040d09d1b32b5d928d9
get facefea from faceExtract sdk
1个文件已修改
37 ■■■■ 已修改文件
service/FaceSdkService.go 37 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
service/FaceSdkService.go
@@ -3,6 +3,7 @@
import (
    "basic.com/pubsub/protomsg.git"
    "basic.com/valib/deliver.git"
    "encoding/base64"
    "fmt"
    "github.com/gogo/protobuf/proto"
    "github.com/pierrec/lz4"
@@ -14,9 +15,13 @@
)
const (
    Url_Service_PUSH = "ipc:///tmp///virtual-faceextract-sdk-pull_2.ipc"
    Url_Service_PULL = "ipc:///tmp///virtual-faceextract-sdk-pull_1.ipc"
    Ipc_Push_Ext = "_2.ipc"
    Ipc_Pull_Ext = "_1.ipc"
    Ipc_Url_Pre = "ipc:///tmp///"
    Virtual_FaceTaskId = "92496BDF-2BFA-98F2-62E8-96DD9866ABD2"
    Virtual_FaceSdkId = "virtual-faceextract-sdk-pull"
    Url_Service_PUSH = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Push_Ext
    Url_Service_PULL = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Pull_Ext
)
var imgPushChan chan protomsg.Recvmsg
var client_push deliver.Deliver
@@ -131,7 +136,31 @@
        rMsg := protomsg.SdkMessage{}
        if err := proto.Unmarshal(resultBytes, &rMsg);err ==nil{
            fmt.Println("received MSG:",rMsg.Cid)
            resultMap[rMsg.Cid] = rMsg
            perId := rMsg.Cid //数据id
            if rMsg.Tasklab !=nil && rMsg.Tasklab.Taskid == Virtual_FaceTaskId {
                sdkInfos := rMsg.Tasklab.Sdkinfos
                fmt.Println("Len(sdkInfos)=",len(sdkInfos))
                //第一个sdk是人脸检测,第二个sdk是人脸提取
                    for _,swt :=range sdkInfos{
                        if swt.Sdktype == "FaceExtract"{
                            fmt.Println("sdkName:",swt.SdkName)
                            var pff protomsg.ParamFaceFeature
                            if err := proto.Unmarshal(swt.Sdkdata, &pff);err !=nil{
                                fmt.Println("ParamFaceFeature unmarshal err:",err)
                            } else {
                                fmt.Println("目标数:",len(pff.ExtComp))
                                for _,fea :=range pff.ExtComp{
                                    base64Fea := base64.StdEncoding.EncodeToString(fea.Feats)
                                    fmt.Println("perId:",perId)
                                    fmt.Println("faceFeature:",base64Fea)
                                }
                            }
                            break
                        }
                    }
            }
            //resultMap[rMsg.Cid] = rMsg
        } else {
            fmt.Println("recv msg Err:",err)
        }