From fe389ae00476e8aed1be85262be1dcb6ffc083d3 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期六, 11 一月 2020 11:34:04 +0800
Subject: [PATCH] debug split
---
run.go | 9 ++++-----
1 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/run.go b/run.go
index c9fdd9f..89cfb1f 100644
--- a/run.go
+++ b/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],
}
}
@@ -103,11 +101,12 @@
feat := s.handle.Extract2(unsafe.Pointer(&i.Data[0]), int(i.Width), int(i.Height), 3)
if feat == nil {
- feat = make([]float32, 1)
+ // feat = make([]float32, 1)
} else {
for k := 0; k < 3; k++ {
s.fnLogger("REID~~~~~~extractor---human_feats------%f", feat[k+2000])
}
+ s.fnLogger("REID~~~~~~Run Reid Use GPU: ", s.gpu)
}
buf := float32SliceAsByteSlice(feat)
c.Send(buf)
--
Gitblit v1.8.0