work/sdk/ydetect.go | ●●●●● 补丁 | 查看 | 原始文档 | 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