udpate

zhangmeng

2019-12-13 f2713d262256f4f0443c64c4c06cac575e391b4b

 work/sdk/humantrack.go

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

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

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

   t.tracker = gohumantrack.NewHumanTracker(t.gpu, t.batchSize, t.flag)
   if gpu == -1 {
      gpu = gogpu.ValidGPU(2048)
   }
   t.tracker = gohumantrack.NewHumanTracker(gpu, t.batchSize, t.flag)
   return true
}