| | |
|---|
| | | import ( |
|---|
| | | "basic.com/pubsub/protomsg.git" |
|---|
| | | "basic.com/valib/deliver.git" |
|---|
| | | "encoding/base64" |
|---|
| | | "fmt" |
|---|
| | | "github.com/gogo/protobuf/proto" |
|---|
| | | "github.com/pierrec/lz4" |
|---|
| | |
|---|
| | | ) |
|---|
| | | |
|---|
| | | 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 |
|---|
| | |
|---|
| | | 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) |
|---|
| | | } |
|---|
