zhangmeng
2019-12-13 117152166646b90756805a141f0e791243b09e04
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