From f2713d262256f4f0443c64c4c06cac575e391b4b Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 13 十二月 2019 17:08:25 +0800
Subject: [PATCH] udpate
---
work/sdk/humantrack.go | 7 ++++++-
1 files changed, 6 insertions(+), 1 deletions(-)
diff --git a/work/sdk/humantrack.go b/work/sdk/humantrack.go
index e8a82a9..7227f2c 100644
--- a/work/sdk/humantrack.go
+++ b/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
}
--
Gitblit v1.8.0