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
---
goreid.go | 2 +-
run.go | 12 ++++++------
csrc/reid.cpp | 6 +++---
3 files changed, 10 insertions(+), 10 deletions(-)
diff --git a/csrc/reid.cpp b/csrc/reid.cpp
index 462ff80..f894176 100644
--- a/csrc/reid.cpp
+++ b/csrc/reid.cpp
@@ -125,9 +125,9 @@
}
}
- for (int k = 0; k < 3; ++k) {
- printf("--extractor---human_feats------%f",feature[k+2000]);
- }
+ // for (int k = 0; k < 3; ++k) {
+ // printf("--extractor---human_feats------%f",feature[k+2000]);
+ // }
printf("\n\n\n\n\n----------------------------------------------------------------------------\n");
diff --git a/goreid.go b/goreid.go
index c81f75f..2784da5 100644
--- a/goreid.go
+++ b/goreid.go
@@ -23,7 +23,7 @@
handle unsafe.Pointer
}
-// NewReID new reid
+// NewSDK new reid
func NewSDK(gpu int, module string) *ReID {
cmodule := C.CString(module)
defer C.free(unsafe.Pointer(cmodule))
diff --git a/run.go b/run.go
index bec3732..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],
@@ -107,7 +103,11 @@
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