From 906c2c219c73b90eae9e5040d09d1b32b5d928d9 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期四, 27 六月 2019 13:52:42 +0800
Subject: [PATCH] get facefea from faceExtract sdk
---
service/FaceSdkService.go | 45 +++++++++++++++++++++++++++++++++++++++------
1 files changed, 39 insertions(+), 6 deletions(-)
diff --git a/service/FaceSdkService.go b/service/FaceSdkService.go
index fefcae5..2db7587 100644
--- a/service/FaceSdkService.go
+++ b/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,8 +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
@@ -46,7 +52,7 @@
Picdata:bc,
})
fmt.Println("pushed img")
- time.Sleep(10*time.Second)
+ time.Sleep(5*time.Second)
}
}
@@ -92,6 +98,7 @@
func InitService(){
fmt.Println("service init!")
imgPushChan = make(chan protomsg.Recvmsg)
+ resultMap = make(map[string]protomsg.SdkMessage,0)
client_push = deliver.NewClient(deliver.PushPull, Url_Service_PUSH)
client_pull = deliver.NewClient(deliver.PushPull, Url_Service_PULL)
defer func() {
@@ -127,9 +134,35 @@
continue
}
rMsg := protomsg.SdkMessage{}
- if err := proto.Unmarshal(resultBytes, &rMsg);err !=nil{
- fmt.Println("recv MSG:",rMsg)
- resultMap[rMsg.Cid] = rMsg
+ if err := proto.Unmarshal(resultBytes, &rMsg);err ==nil{
+ fmt.Println("received MSG:",rMsg.Cid)
+ perId := rMsg.Cid //鏁版嵁id
+ if rMsg.Tasklab !=nil && rMsg.Tasklab.Taskid == Virtual_FaceTaskId {
+ sdkInfos := rMsg.Tasklab.Sdkinfos
+ fmt.Println("Len(sdkInfos)=",len(sdkInfos))
+ //绗竴涓猻dk鏄汉鑴告娴嬶紝绗簩涓猻dk鏄汉鑴告彁鍙�
+ 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)
}
}
--
Gitblit v1.8.0