liuxiaolong
2019-08-22 f4e8f206a6760bdc31734dfcb1c65916b5b76311
service/FaceSdkService.go
@@ -34,23 +34,25 @@
)
func GetFaceFeaFromSdk(fileBytes []byte,deadTime time.Duration) ([]*protomsg.ResultFaceDetect,error,*protomsg.Image){
   tStart :=time.Now()
   t1 := time.Now()
   s := NewFaceSdkService(fileBytes, deadTime)
   i, err := s.ReadFromUploadImg()
   logger.Debug("ReadFromUploadImg用时:", time.Since(tStart))
   logger.Debug("ReadFromUploadImg用时:", time.Since(t1))
   t1 = time.Now()
   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))
   logger.Debug("ImgCompress用时:", time.Since(t1))
   t1 = time.Now()
   if err !=nil {
      logger.Debug("ImgCompress err:",err)
      return nil,err,i
   }
   s.PushImgMsg(bc)
   t1 := time.Now()
   logger.Debug("PushImgMsg用时:", time.Since(t1))
   t1 = time.Now()
   s.GetFaceFea()
   logger.Debug("GetFaceFea用时:", time.Since(t1))
   if s.Result == nil{
@@ -89,25 +91,27 @@
}
func (s *FaceSdkService) ReadFromUploadImg() (*protomsg.Image,error){
   //defer (*(s.File)).Close()
   //imgB, err := ioutil.ReadAll(*(s.File))
   //if err !=nil{
   //   logger.Debug("File.Read err:",err)
   //   return nil,err
   //}
   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()
   if picMat.Rows() > 1000 || picMat.Cols() > 1000 {
      gocv.Resize(picMat,&newMat, image.Pt(0,0), 0.5,  0.5, gocv.InterpolationDefault)
   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
   }
   defer picMat.Close()
   if picMat.Empty() {
      logger.Debug("file not exist")
      return nil,errors.New("picMat is empty")
@@ -129,18 +133,13 @@
}
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
@@ -150,7 +149,6 @@
         return nil,errors.New("compressed len is 0")
      }
      bc = bc[:n]
      logger.Debug("bc用时:", time.Since(t1))
      return bc,nil
   }
}
@@ -280,7 +278,6 @@
      }
      rMsg := protomsg.SdkMessage{}
      if err := proto.Unmarshal(resultBytes, &rMsg); err == nil {
         logger.Debug("received MSG:", rMsg.Cid)
         i := protomsg.Image{}
         bdata, err := util.UnCompress(rMsg.Data)
         if err !=nil {