work/sdk/humantrack.go | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
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 }