派生自 libgowrapper/yolo

zhangmeng
2019-12-09 1e995e0b8237159c1683cedcedd96ef35ce1ba30
update
1个文件已修改
13 ■■■■■ 已修改文件
goyolo.go 13 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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)