reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-11 d6ed7cfb76e57556064281a0259709265d7d026e
run.go
@@ -17,6 +17,7 @@
   handle   *ReID
   fnLogger func(...interface{})
   gpu int
   ipc string
}
@@ -50,10 +51,6 @@
   if rGPU == -1 {
      rGPU = gogpu.ValidGPU(gpuM + 512)
   }
   if rGPU == -1 {
      fn("Reid SDK Create Error When Find GPU")
      return nil
   }
   handle := NewSDK(rGPU, cfg.Param[sModel])
   if handle == nil {
@@ -65,6 +62,7 @@
      handle:   handle,
      fnLogger: fn,
      gpu: rGPU,
      ipc: cfg.Param[sIPC],
   }
}
@@ -105,6 +103,7 @@
         if feat == nil {
            feat = make([]float32, 1)
         } else {
            s.fnLogger("REID~~~~~~Run Reid Use GPU: ", s.gpu)
            for k := 0; k < 3; k++ {
               s.fnLogger("REID~~~~~~extractor---human_feats------%f", feat[k+2000])
            }