udpate

zhangmeng

2019-12-17 4a745a94f3eee82624e1b211b48920c0b7fcd6d2

 work/sdk/humantrack.go

@@ -11,6 +11,7 @@
   "analysis/gohumantrack"

   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/gogpu.git"
)

// HumanTracker track
@@ -36,9 +37,19 @@
      logo.Errorln("ONLY SUPPORT BATCH SIZE = 1")
      return false
   }
   gpu := t.gpu

   t.tracker = gohumantrack.NewHumanTracker(t.gpu, t.batchSize, t.flag)
   return true
   if gpu == -1 {
      gpu = gogpu.ValidGPU(2048)
   }
   logo.Infof("start gpu %d, batch %d, flag %d\n", gpu, t.batchSize, t.flag)
   t.tracker = gohumantrack.NewHumanTracker(gpu, t.batchSize, t.flag)
   if t.tracker != nil {
      logo.Infoln("Start Success")
      return true
   }
   logo.Infoln("Start Failed")
   return false
}

// Run impl
@@ -87,6 +98,11 @@
            data = nil
         }
         ejectResult(data, rMsg, out)
         var id, name string
         if rMsg.Msg.Tasklab != nil {
            id, name = rMsg.Msg.Tasklab.Taskid, rMsg.Msg.Tasklab.Taskname
         }
         logo.Infoln("CAMERAID: ", rMsg.Msg.Cid, " TASKID: ", id, " TASKNAME: ", name, " Human Track COUNT: ", len(hr[0]))

         /////////////////////////////////////
         sc++