zhangmeng
2019-12-16 f0a55fbe2c2757bc64883ae65a717ad1a1ae9072
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