zhangmeng
2019-12-13 117152166646b90756805a141f0e791243b09e04
update
1个文件已修改
3 ■■■■ 已修改文件
work/sdk/ydetect.go 3 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
work/sdk/ydetect.go
@@ -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