update

zhangmeng

2019-12-19 cae1319f90bfb88fdac02ed07a38396e7ca4378a

 work/sdk/humantrack.go

@@ -1,6 +1,7 @@
package sdk

import (
   "analysis/goconv"
   "analysis/logo"
   "analysis/work"
   "context"
@@ -15,6 +16,8 @@

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

@@ -122,6 +125,14 @@

   imgW, imgH := int(i.Width), int(i.Height)

   var imgData []byte
   rw, rh := 1280, 720
   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
   }
   img := gohumantrack.ImageHumanTracker{
      Data:    i.Data,
      Width:   imgW,
@@ -134,12 +145,12 @@

   if i, ok := t.mapCameraImageIndex[rMsg.Msg.Cid]; ok {
      if i < t.batchSize {
         t.images[i] = &imageWithID{&img, rMsg.Msg.Cid}
         t.images[i] = &imageWithID{&img, fx, fy, rMsg.Msg.Cid}
         t.msgs[i] = &rMsg
      }
   } else {
      if t.index < t.batchSize {
         t.images[t.index] = &imageWithID{&img, rMsg.Msg.Cid}
         t.images[t.index] = &imageWithID{&img, fx, fy, rMsg.Msg.Cid}
         t.msgs[t.index] = &rMsg
         t.mapCameraImageIndex[rMsg.Msg.Cid] = t.index
      }
@@ -178,7 +189,7 @@
      if t.images[i] == nil {
         continue
      }
      hr := convert2ProtoHumanTrackResult(res[i])
      hr := convert2ProtoHumanTrackResultWithScale(res[i], t.images[i].fx, t.images[i].fy)
      result := protomsg.HumanTrackResult{Result: hr}
      data, err := proto.Marshal(&result)
      if err != nil {
@@ -238,3 +249,26 @@
   }
   return res
}

func convert2ProtoHumanTrackResultWithScale(obj gohumantrack.FgResult, fx, fy float32) []*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),
      }
      pr := &protomsg.HumanTrack{
         RcHuman:    &rect,
         Confidence: r.Confidence,
         X:          (int32)((float32)(r.X)/fx,
         Y:          (int32)((float32)(r.Y)/fy,
         Id:         r.ID,
         Feature:    r.Feature[:],
      }
      res = append(res, pr)
   }
   return res
}