bug fixed

zhangmeng

2019-12-16 cd982114434e2fa2617ccdf02ea49e162c09a464

 work/sdk/ydetect.go

@@ -28,7 +28,7 @@
   cond  bool

   handle interface{}
   fnInit func(string, string, string, int) interface{}
   fnInit func(string, string, string, int, func(...interface{})) interface{}
   fnFree func(interface{})
   fnRun  func(interface{}, string, []byte, int, int, int, float32, int) ([]sdkstruct.CObjTrackInfo, []sdkstruct.CObjTrackInfo)
}
@@ -57,7 +57,7 @@
      cond:    false,

      handle: nil,
      fnInit: fnInit.(func(string, string, string, int) interface{}),
      fnInit: fnInit.(func(string, string, string, int, func(...interface{})) interface{}),
      fnFree: fnFree.(func(interface{})),
      fnRun:  fnRun.(func(interface{}, string, []byte, int, int, int, float32, int) ([]sdkstruct.CObjTrackInfo, []sdkstruct.CObjTrackInfo)),
   }
@@ -70,7 +70,7 @@
   if gpu == -1 {
      gpu = gogpu.ValidGPU(2048)
   }
   h := y.fnInit(y.cfg, y.weights, y.name, gpu)
   h := y.fnInit(y.cfg, y.weights, y.name, gpu, logo.Infoln)
   logo.Infoln("YOLO USE GPU: ", gpu)

   if h == nil {