work/sdk/ydetect.go
@@ -9,7 +9,7 @@ "basic.com/valib/gogpu.git" "basic.com/valib/gosdk.git" "analysis/gosdk" ) type trackInfo struct { @@ -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