From bdf3ad71583fb4ef100d3819ecdae8fd9f70083e Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期六, 11 一月 2020 09:36:33 +0800
Subject: [PATCH] debug split
---
run.go | 16 +++++++++-------
1 files changed, 9 insertions(+), 7 deletions(-)
diff --git a/run.go b/run.go
index 9cd9ec1..c9fdd9f 100644
--- a/run.go
+++ b/run.go
@@ -15,7 +15,6 @@
type reid struct {
handle *ReID
- list *sdkhelper.LockList
fnLogger func(...interface{})
ipc string
@@ -46,7 +45,7 @@
gpuM := sdkhelper.Atoi(cfg.Param[sGPU])
- rGPU := 0
+ rGPU := gpu
if rGPU == -1 {
rGPU = gogpu.ValidGPU(gpuM + 512)
@@ -56,8 +55,6 @@
return nil
}
- rGPU = 0
-
handle := NewSDK(rGPU, cfg.Param[sModel])
if handle == nil {
fn("Reid SDK Create Error When New SDK")
@@ -66,7 +63,6 @@
return &reid{
handle: handle,
- list: sdkhelper.NewLockList(6),
fnLogger: fn,
ipc: cfg.Param[sIPC],
@@ -90,9 +86,9 @@
msg, err = c.Recv()
if err != nil {
- s.fnLogger("REID~~~~~~Recv Image Error : ", err)
continue
}
+
i := &protomsg.Image{}
err := proto.Unmarshal(msg, i)
if err != nil {
@@ -103,9 +99,15 @@
s.fnLogger("REID~~~~~~protomsg.Image data null")
continue
}
+ s.fnLogger("REID~~~~~~Recv Image:", len(i.Data))
+
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])
+ }
}
buf := float32SliceAsByteSlice(feat)
c.Send(buf)
--
Gitblit v1.8.0