up

zhangmeng

2019-12-19 d8d2bc7b4cf8e4a1f2b2a325d2ca02f0eb409d9c

 work/sdk/humantrack.go

@@ -16,8 +16,8 @@

type imageWithID struct {
   img *gohumantrack.ImageHumanTracker
   fx  float32
   fy  float32
   fx  float64
   fy  float64
   id  string
}

@@ -130,13 +130,13 @@
   fx, fy := 1.0, 1.0
   if imgW != rw || imgH != rh {
      imgData = goconv.ResizeBGR(i.Data, imgW, imgH, rw, rh)
      fx = (float32)(rw)/(float32)imgW
      fy = (float32)(rh)/(float32)imgH
      fx = (float64)(rw) / (float64)(imgW)
      fy = (float64)(rh) / (float64)(imgH)
   }
   img := gohumantrack.ImageHumanTracker{
      Data:    i.Data,
      Width:   imgW,
      Height:  imgH,
      Data:    imgData,
      Width:   rw,
      Height:  rh,
      Channel: 3,
   }

@@ -176,7 +176,7 @@
      }
   }

   res, err := t.tracker.ProcessImagePointer(pimg)
   res, err := t.tracker.ProcessImagePointer(pimg, rw, rh, 3)
   t.recvImageCount = 0

   if err != nil {
@@ -250,21 +250,21 @@
   return res
}

func convert2ProtoHumanTrackResultWithScale(obj gohumantrack.FgResult, fx, fy float32) []*protomsg.HumanTrack {
func convert2ProtoHumanTrackResultWithScale(obj gohumantrack.FgResult, fx, fy float64) []*protomsg.HumanTrack {
   res := []*protomsg.HumanTrack{}
   for i := 0; i < int(obj.FgNum); i++ {
      r := obj.Fginfo[i]
      rect := protomsg.Rect{
         Left:   (int32)((float32)(r.Left)/fx),
         Right:  (int32)((float32)(r.Right)/fy),
         Top:    (int32)((float32)(r.Top)/fx),
         Bottom: (int32)((float32)(r.Bottom)/fy),
         Left:   (int32)((float64)(r.Left) / fx),
         Right:  (int32)((float64)(r.Right) / fy),
         Top:    (int32)((float64)(r.Top) / fx),
         Bottom: (int32)((float64)(r.Bottom) / fy),
      }
      pr := &protomsg.HumanTrack{
         RcHuman:    &rect,
         Confidence: r.Confidence,
         X:          (int32)((float32)(r.X)/fx,
         Y:          (int32)((float32)(r.Y)/fy,
         X:          (int32)((float64)(r.X) / fx),
         Y:          (int32)((float64)(r.Y) / fy),
         Id:         r.ID,
         Feature:    r.Feature[:],
      }