From cd982114434e2fa2617ccdf02ea49e162c09a464 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期一, 16 十二月 2019 15:39:45 +0800
Subject: [PATCH] bug fixed
---
work/sdk/ydetect.go | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/work/sdk/ydetect.go b/work/sdk/ydetect.go
index 737b41a..922186c 100644
--- a/work/sdk/ydetect.go
+++ b/work/sdk/ydetect.go
@@ -28,7 +28,7 @@
cond bool
handle interface{}
- fnInit func(string, string, string, int) interface{}
+ fnInit func(string, string, string, int, func(...interface{})) interface{}
fnFree func(interface{})
fnRun func(interface{}, string, []byte, int, int, int, float32, int) ([]sdkstruct.CObjTrackInfo, []sdkstruct.CObjTrackInfo)
}
@@ -57,7 +57,7 @@
cond: false,
handle: nil,
- fnInit: fnInit.(func(string, string, string, int) interface{}),
+ fnInit: fnInit.(func(string, string, string, int, func(...interface{})) interface{}),
fnFree: fnFree.(func(interface{})),
fnRun: fnRun.(func(interface{}, string, []byte, int, int, int, float32, int) ([]sdkstruct.CObjTrackInfo, []sdkstruct.CObjTrackInfo)),
}
@@ -70,7 +70,7 @@
if gpu == -1 {
gpu = gogpu.ValidGPU(2048)
}
- h := y.fnInit(y.cfg, y.weights, y.name, gpu)
+ h := y.fnInit(y.cfg, y.weights, y.name, gpu, logo.Infoln)
logo.Infoln("YOLO USE GPU: ", gpu)
if h == nil {
--
Gitblit v1.8.0