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