zhangmeng
2019-12-13 f2713d262256f4f0443c64c4c06cac575e391b4b
udpate
1个文件已修改
7 ■■■■ 已修改文件
work/sdk/humantrack.go 7 ●●●● 补丁 | 查看 | 原始文档 | 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
}