work/sdk/ydetect.go
@@ -53,11 +53,11 @@ gpu = gogpu.ValidGPU(2048) } yolo := gosdk.InitYolo(y.cfg, y.weights, y.name, gpu) logo.Infoln("yolo use gpu: ", gpu) if yolo == nil { return false } logo.Infoln("yolo use gpu: ", gpu) y.yolo = yolo return true