From 117152166646b90756805a141f0e791243b09e04 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 13 十二月 2019 17:18:43 +0800
Subject: [PATCH] update
---
work/sdk/ydetect.go | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/work/sdk/ydetect.go b/work/sdk/ydetect.go
index fadf272..9b641c9 100644
--- a/work/sdk/ydetect.go
+++ b/work/sdk/ydetect.go
@@ -53,11 +53,12 @@
gpu = gogpu.ValidGPU(2048)
}
yolo := gosdk.InitYolo(y.cfg, y.weights, y.name, gpu)
- logo.Infoln("yolo use gpu: ", gpu)
if yolo == nil {
+ logo.Infoln("yolo start failed: ", gpu)
return false
}
+ logo.Infoln("yolo use gpu: ", gpu)
y.yolo = yolo
return true
--
Gitblit v1.8.0