From 1e995e0b8237159c1683cedcedd96ef35ce1ba30 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期一, 09 十二月 2019 17:13:32 +0800
Subject: [PATCH] update
---
goyolo.go | 13 +++++++++----
1 files changed, 9 insertions(+), 4 deletions(-)
diff --git a/goyolo.go b/goyolo.go
index 09cf99d..11a8bdd 100644
--- a/goyolo.go
+++ b/goyolo.go
@@ -60,9 +60,10 @@
// SDKImage sdk image
type SDKImage struct {
- Data []byte
- Width int
- Height int
+ Data []byte
+ Width int
+ Height int
+ Channel int
}
// NewYolo init yolo sdk
@@ -108,12 +109,16 @@
data := img.Data
w := img.Width
h := img.Height
+ c := img.Channel
+ if c == 0 {
+ c = 3
+ }
var count C.int
var cobjinfo unsafe.Pointer
ret := C.detect(y.handle,
- unsafe.Pointer(&data[0]), C.int(w), C.int(h), 3,
+ unsafe.Pointer(&data[0]), C.int(w), C.int(h), C.int(c),
C.float(thrsh), C.int(umns),
&cobjinfo, &count)
--
Gitblit v1.8.0