From 31a9d64a0cc99011263c4516dfaac5c9879a0c45 Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期四, 27 六月 2019 16:51:33 +0800 Subject: [PATCH] test --- service/FaceSdkService.go | 49 +++++++++++++++++++++++-------------------------- 1 files changed, 23 insertions(+), 26 deletions(-) diff --git a/service/FaceSdkService.go b/service/FaceSdkService.go index 2db7587..2737ff4 100644 --- a/service/FaceSdkService.go +++ b/service/FaceSdkService.go @@ -8,9 +8,7 @@ "github.com/gogo/protobuf/proto" "github.com/pierrec/lz4" "github.com/satori/go.uuid" - "image" - "io/ioutil" - "os" + "gocv.io/x/gocv" "time" ) @@ -29,9 +27,9 @@ func TestPushImgMsg() { InitService() - proImg := readImgFile() - if b, err := proto.Marshal(&proImg);err !=nil{ + i := readImgFile() + if b, err := proto.Marshal(&i);err !=nil{ fmt.Println("protoImage marshal err") return } else { @@ -64,28 +62,27 @@ formatTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05") filePath := "/home/user/workspace/timg.jpg" - file, err := os.Open(filePath) - defer file.Close() - if err !=nil{ - fmt.Println("image not exist") - return i - } else { - img, _, err := image.Decode(file) - bytes, err := ioutil.ReadFile(filePath) - if err !=nil { - return i - } - b := img.Bounds() - width := b.Max.X - height := b.Max.Y - i = protomsg.Image{ - Width:int32(width), - Height:int32(height), - Timestamp:formatTimeStr, - Data:bytes, - } + picMat := gocv.IMRead(filePath, gocv.IMReadColor) + + defer picMat.Close() + + if picMat.Empty() { + fmt.Println("file not exist") return i } + width := int32(picMat.Rows()) + height := int32(picMat.Cols()) + data := picMat.DataPtrUint8() + fmt.Printf("width:%d,height:%d,data.length:%d,timestamp:%s",width,height,len(data),formatTimeStr) + i = protomsg.Image{ + Width: width, + Height: height, + Timestamp: formatTimeStr, + Data: data, + Id: timeUnix, + } + fmt.Println("gocv read img completed") + return i } func PushImgMsg(is protomsg.Recvmsg){ @@ -140,10 +137,10 @@ 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) + fmt.Println("sdkData.len:",len(swt.Sdkdata)) var pff protomsg.ParamFaceFeature if err := proto.Unmarshal(swt.Sdkdata, &pff);err !=nil{ fmt.Println("ParamFaceFeature unmarshal err:",err) -- Gitblit v1.8.0