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 |   19 +++++++++----------
 1 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/run.go b/run.go
index bec3732..89cfb1f 100644
--- a/run.go
+++ b/run.go
@@ -15,9 +15,9 @@
 
 type reid struct {
 	handle   *ReID
-	list     *sdkhelper.LockList
 	fnLogger func(...interface{})
 
+	gpu int
 	ipc string
 }
 
@@ -46,17 +46,11 @@
 
 	gpuM := sdkhelper.Atoi(cfg.Param[sGPU])
 
-	rGPU := 0
+	rGPU := gpu
 
 	if rGPU == -1 {
 		rGPU = gogpu.ValidGPU(gpuM + 512)
 	}
-	if rGPU == -1 {
-		fn("Reid SDK Create Error When Find GPU")
-		return nil
-	}
-
-	rGPU = 0
 
 	handle := NewSDK(rGPU, cfg.Param[sModel])
 	if handle == nil {
@@ -66,9 +60,9 @@
 
 	return &reid{
 		handle:   handle,
-		list:     sdkhelper.NewLockList(6),
 		fnLogger: fn,
 
+		gpu: rGPU,
 		ipc: cfg.Param[sIPC],
 	}
 }
@@ -107,7 +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)
+			} 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