From 5d78b36e851a5862302f2e40c99a4477020c79bc Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期五, 20 九月 2019 13:46:52 +0800 Subject: [PATCH] 查找此人时先切图 --- service/FaceSdkService.go | 113 +++++++++++++++++++++++++++++++++++++++----------------- 1 files changed, 79 insertions(+), 34 deletions(-) diff --git a/service/FaceSdkService.go b/service/FaceSdkService.go index bd23acf..aacbc18 100644 --- a/service/FaceSdkService.go +++ b/service/FaceSdkService.go @@ -2,50 +2,95 @@ import ( "basic.com/pubsub/protomsg.git" - "basic.com/valib/deliver.git" - "encoding/json" - "fmt" "github.com/gogo/protobuf/proto" + "github.com/pkg/errors" + "gocv.io/x/gocv" + "image" + "time" + "basic.com/valib/logger.git" ) -const ( - Url_Service_PUSH = "tcp:///tmp///webserver-2.ipc" - Url_Service_PULL = "tcp:///tmp///webserver-1.ipc" -) -var imgPushChan chan protomsg.Recvmsg - -func PushImgMsg(is protomsg.Recvmsg){ - imgPushChan <- is +type FaceSdkService struct { + File []byte + Id int64 + Result []*protomsg.ResultFaceDetect + DeadTime time.Duration } -var resultMap map[string]protomsg.SdkMessage +const ( + faceExtractWebCID = "virtual-face-extract-web-camera-id" +) +func GetFaceFeaFromSdk(fileBytes []byte,deadTime time.Duration) ([]*protomsg.ResultFaceDetect,error,*protomsg.Image){ + t1 := time.Now() + s := newFaceSdkService(fileBytes, deadTime) + i, err := s.readFromUploadImg() + logger.Debug("ReadFromUploadImg鐢ㄦ椂:", time.Since(t1)) + if err !=nil{ + logger.Debug("readFromUploadImg err:",err) + return nil,err,i + } + imgBytes, err := proto.Marshal(i) + if err !=nil { + logger.Debug("i marshal err:",err) + return nil,err,i + } + detectResults, err := DoFDetectWithRpc(&protomsg.FaceDetectRequest{ReqParam: imgBytes}, deadTime) + logger.Debug("鎻愬彇face鍏辩敤鏃讹細", time.Since(t1)) + if err !=nil { + logger.Debug("DoFDetectWithRpc err:",err) + return nil,err,i + } else { + return detectResults,nil,i + } +} -func InitService(){ - imgPushChan = make(chan protomsg.Recvmsg) - client_push := deliver.NewClient(deliver.PushPull, Url_Service_PUSH) - client_pull := deliver.NewClient(deliver.PushPull, Url_Service_PULL) - for { - select { - case is := <- imgPushChan: - b, _ := json.Marshal(is) - client_push.Send(b) +func newFaceSdkService(fileBytes []byte, deadTime time.Duration) *FaceSdkService{ + return &FaceSdkService{ + File:fileBytes, + Id:time.Now().UnixNano(), + DeadTime:deadTime, + } +} + +func (s *FaceSdkService) readFromUploadImg() (*protomsg.Image,error){ + picMat, err := gocv.IMDecode(s.File, gocv.IMReadColor) + if err !=nil { + logger.Debug("gocv.IMDecode err:",err) + return nil,err + } + logger.Debug("picMat.Data.len:", len(picMat.ToBytes())) + newMat := gocv.NewMat() + size := 1024 + if picMat.Rows() > size || picMat.Cols() > size { + fx := float64(size)/float64(picMat.Rows()) + fy := float64(size)/float64(picMat.Cols()) + ff := fx + if fx > fy{ + ff = fy } + gocv.Resize(picMat,&newMat, image.Pt(0,0), ff, ff, gocv.InterpolationDefault) + picMat = newMat } - //鎺ユ敹浜鸿劯鎻愬彇缁撴灉 - for { - resultBytes, err := client_pull.Recv() - if err !=nil{ - fmt.Println("pull err:",err) - continue - } - rMsg := protomsg.SdkMessage{} - if err := proto.Unmarshal(resultBytes, &rMsg);err !=nil{ - fmt.Println(rMsg) - resultMap[rMsg.Cid] = rMsg - } + defer picMat.Close() + if picMat.Empty() { + logger.Debug("file not exist") + return nil,errors.New("picMat is empty") } + height := int32(picMat.Rows()) + width := int32(picMat.Cols()) + data := picMat.ToBytes() + timeUnix := time.Now().Unix() + formatTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05") -} \ No newline at end of file + return &protomsg.Image{ + Width: width, + Height: height, + Timestamp: formatTimeStr, + Data: data, + Id: s.Id, + Cid: faceExtractWebCID, + },nil +} -- Gitblit v1.8.0