From f0a55fbe2c2757bc64883ae65a717ad1a1ae9072 Mon Sep 17 00:00:00 2001
From: zhangmeng <zhangmeng@aiotlink.com>
Date: 星期一, 16 十二月 2019 10:41:00 +0800
Subject: [PATCH] update cuda10.0/cuda10.2(face)
---
work/sdk/ydetect.go | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/work/sdk/ydetect.go b/work/sdk/ydetect.go
index fadf272..144ae0f 100644
--- a/work/sdk/ydetect.go
+++ b/work/sdk/ydetect.go
@@ -9,7 +9,7 @@
"basic.com/valib/gogpu.git"
- "basic.com/valib/gosdk.git"
+ "analysis/gosdk"
)
type trackInfo struct {
@@ -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