reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-11 d6ed7cfb76e57556064281a0259709265d7d026e
debug split
1个文件已修改
7 ■■■■■ 已修改文件
run.go 7 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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])
                }