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/humantrack.go | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/work/sdk/humantrack.go b/work/sdk/humantrack.go
index 446c8a6..8c5c3fd 100644
--- a/work/sdk/humantrack.go
+++ b/work/sdk/humantrack.go
@@ -21,7 +21,7 @@
flag int
handle interface{}
- fnInit func(int, int, int) interface{}
+ fnInit func(int, int, int, func(...interface{})) interface{}
fnFree func(interface{})
fnRun func(interface{}, []byte, int, int, int) []sdkstruct.FgResult
fnProcess func(interface{}, []sdkstruct.SDKImage) ([]sdkstruct.FgResult, error)
@@ -52,7 +52,7 @@
flag: flag,
handle: nil,
- fnInit: fnInit.(func(int, int, int) interface{}),
+ fnInit: fnInit.(func(int, int, int, func(...interface{})) interface{}),
fnFree: fnFree.(func(interface{})),
fnRun: fnRun.(func(interface{}, []byte, int, int, int) []sdkstruct.FgResult),
fnProcess: fnProc.(func(interface{}, []sdkstruct.SDKImage) ([]sdkstruct.FgResult, error)),
@@ -71,7 +71,7 @@
if gpu == -1 {
gpu = gogpu.ValidGPU(2048)
}
- h := t.fnInit(gpu, t.batchSize, t.flag)
+ h := t.fnInit(gpu, t.batchSize, t.flag, logo.Infoln)
logo.Infoln("HumanTrack USE GPU: ", gpu)
if h == nil {
--
Gitblit v1.8.0