udpate

zhangmeng

2019-12-19 1118eaddcadb8b7c4a5da084a46c0c4fc18040e1

 work/sdk/humantrack.go

@@ -13,6 +13,11 @@
   "basic.com/valib/gogpu.git"
)

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

// HumanTracker track
type HumanTracker struct {
   tracker   *gohumantrack.HumanTracker
@@ -24,7 +29,7 @@
   mapCameraImageIndex map[string]int
   recvImageCount      int
   index               int
   images              []*gohumantrack.ImageHumanTracker
   images              []*imageWithID
   msgs                []*work.MsgRS
}

@@ -39,7 +44,7 @@
      recvImageCount:      0,
      index:               0,
      mapCameraImageIndex: make(map[string]int),
      images:              make([]*gohumantrack.ImageHumanTracker, batchSize),
      images:              make([]*imageWithID, batchSize),
      msgs:                make([]*work.MsgRS, batchSize),
   }
}
@@ -129,12 +134,12 @@

   if i, ok := t.mapCameraImageIndex[rMsg.Msg.Cid]; ok {
      if i < t.batchSize {
         t.images[i] = &img
         t.images[i] = &imageWithID{&img, rMsg.Msg.Cid}
         t.msgs[i] = &rMsg
      }
   } else {
      if t.index < t.batchSize {
         t.images[t.index] = &img
         t.images[t.index] = &imageWithID{&img, rMsg.Msg.Cid}
         t.msgs[t.index] = &rMsg
         t.mapCameraImageIndex[rMsg.Msg.Cid] = t.index
      }
@@ -143,15 +148,28 @@
   t.recvImageCount++

   if t.recvImageCount < t.batchSize+t.batchSize/2 {
      logo.Infoln("batch~~~~~~Current Index: ", t.index)
      return
   }

   res, err := t.tracker.ProcessImagePointer(t.images[:])
   t.mapCameraImageIndex = nil
   for k, v := range t.mapCameraImageIndex {
      logo.Infoln("batch~~~~~~Map index: ", v, " camera: ", k)
   }

   var pimg []*gohumantrack.ImageHumanTracker
   for k, v := range t.images[:] {
      if v != nil {
         pimg = append(pimg, v.img)
         logo.Infoln("batch~~~~~~Image index: ", k, " camera: ", v.id, " image address: ", v.img)
      } else {
         pimg = append(pimg, nil)
      }
   }

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

   if err != nil {
      logo.Infoln("batch~~~~~~Track Failed: ", err)
      logo.Infoln("batch~~~~~~Track Image Count: ", t.index, " Failed: ", err)
      ejectResult(nil, rMsg, out)
      return
   }