From b3a8546e2c1472d0e669bd169cef091c2530c792 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 13 十二月 2019 17:17:02 +0800
Subject: [PATCH] udpate
---
work/sdk/humantrack.go | 8 +++++++-
work/sdk/ydetect.go | 2 +-
2 files changed, 8 insertions(+), 2 deletions(-)
diff --git a/work/sdk/humantrack.go b/work/sdk/humantrack.go
index 7227f2c..7236e32 100644
--- a/work/sdk/humantrack.go
+++ b/work/sdk/humantrack.go
@@ -42,8 +42,14 @@
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)
- return true
+ if t.tracker != nil {
+ logo.Infoln("Start Success")
+ return true
+ }
+ logo.Infoln("Start Failed")
+ return false
}
// Run impl
diff --git a/work/sdk/ydetect.go b/work/sdk/ydetect.go
index b963872..7305d13 100644
--- a/work/sdk/ydetect.go
+++ b/work/sdk/ydetect.go
@@ -53,11 +53,11 @@
gpu = gogpu.ValidGPU(2048)
}
yolo := gosdk.InitYolo(y.cfg, y.weights, y.name, gpu)
- logo.Infoln("yolo use gpu: ", gpu)
if yolo == nil {
return false
}
+ logo.Infoln("yolo use gpu: ", gpu)
y.yolo = yolo
return true
--
Gitblit v1.8.0