from picMat to newMat use resize

liuxiaolong

2019-08-06 92c1e11f87617b0e67f6918423a428248e2015e0

 service/FaceSdkService.go

@@ -8,9 +8,11 @@
   "github.com/pkg/errors"
   "github.com/satori/go.uuid"
   "gocv.io/x/gocv"
   "image"
   "sync"
   "time"
   "webserver/extend/logger"
   "webserver/extend/util"
)

type FaceSdkService struct {
@@ -32,19 +34,25 @@
)

func GetFaceFeaFromSdk(fileBytes []byte,deadTime time.Duration) ([]*protomsg.ResultFaceDetect,error,*protomsg.Image){
   tStart :=time.Now()
   s := NewFaceSdkService(fileBytes, deadTime)
   i, err := s.ReadFromUploadImg()
   logger.Debug("ReadFromUploadImg用时：", time.Since(tStart))
   if err !=nil{
      logger.Debug("readFromUploadImg err:",err)
      return nil,err,i
   }
   tStart = time.Now()
   bc, err := ImgCompress(i)
   logger.Debug("ImgCompress用时：", time.Since(tStart))
   if err !=nil {
      logger.Debug("ImgCompress err:",err)
      return nil,err,i
   }
   s.PushImgMsg(bc)
   t1 := time.Now()
   s.GetFaceFea()
   logger.Debug("GetFaceFea用时:", time.Since(t1))
   if s.Result == nil{
      return nil,errors.New("no fea"),i
   } else {
@@ -93,6 +101,12 @@
      logger.Debug("gocv.IMDecode err:",err)
      return nil,err
   }
   newMat := gocv.NewMat()
   if picMat.Rows() > 1000 || picMat.Cols() > 1000 {
      gocv.Resize(picMat,&newMat, image.Pt(0,0), 0.5,  0.5, gocv.InterpolationDefault)
      picMat = newMat
   }

   defer picMat.Close()
   if picMat.Empty() {
      logger.Debug("file not exist")
@@ -115,13 +129,18 @@
}

func ImgCompress(i *protomsg.Image) ([]byte,error){
   t1 := time.Now()
   if b, err := proto.Marshal(i); err != nil {
      logger.Debug("protoImage marshal err")
      return nil,err
   } else {
      logger.Debug("proto.Marshal用时：", time.Since(t1))
      t1 = time.Now()
      bc := make([]byte, len(b))
      ht := make([]int, 64<<10)
      n, err := lz4.CompressBlock(b, bc, ht)
      logger.Debug("lz4.CompressBlock用时：", time.Since(t1))
      t1 = time.Now()
      if err != nil {
         logger.Debug(err)
         return nil,err
@@ -131,6 +150,7 @@
         return nil,errors.New("compressed len is 0")
      }
      bc = bc[:n]
      logger.Debug("bc用时：", time.Since(t1))
      return bc,nil
   }
}
@@ -213,7 +233,7 @@
   return coms,ok
}

func (f *FeaResult) Delete(id string){
func (f *FeaResult) Delete(id int64){
   f.Lock.Lock()
   defer f.Lock.Unlock()
   delete(f.FaceM,id)
@@ -262,7 +282,12 @@
      if err := proto.Unmarshal(resultBytes, &rMsg); err == nil {
         logger.Debug("received MSG:", rMsg.Cid)
         i := protomsg.Image{}
         err := proto.Unmarshal(rMsg.Data, &i)
         bdata, err := util.UnCompress(rMsg.Data)
         if err !=nil {
            logger.Debug("uncompress err:",err)
            continue
         }
         err = proto.Unmarshal(bdata, &i)
         if err !=nil {
            continue
         }