From 6f264a4af85ea20c1add552cd9bd92c5d708c9c9 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 14 一月 2020 11:32:11 +0800
Subject: [PATCH] debug
---
run.go | 20 ++++++++++++++++++++
1 files changed, 20 insertions(+), 0 deletions(-)
diff --git a/run.go b/run.go
index 73abf30..488f9b2 100644
--- a/run.go
+++ b/run.go
@@ -1,8 +1,11 @@
package main
import (
+ "bufio"
"context"
+ "fmt"
"io/ioutil"
+ "os"
"unsafe"
"basic.com/libgowrapper/sdkhelper.git"
@@ -101,9 +104,11 @@
}
s.fnLogger("REID~~~~~~Recv Image:", len(i.Data))
+ /////////////////////////////////
if jpg, err := godraw.ToJpeg(i.Data, int(i.Width), int(i.Height), nil); err == nil {
ioutil.WriteFile("./pic.jpg", jpg, 0644)
}
+ /////////////////////////////////
feat := s.handle.Extract2(unsafe.Pointer(&i.Data[0]), int(i.Width), int(i.Height), 3)
if feat == nil {
// feat = make([]float32, 1)
@@ -112,8 +117,23 @@
s.fnLogger("REID~~~~~~extractor---human_feats------%f", feat[k+2000])
}
s.fnLogger("REID~~~~~~Run Reid Use GPU: ", s.gpu)
+
+ /////////////////////////////////
+ if f, err := os.Create("./origin-feat.txt"); err == nil {
+ defer f.Close()
+
+ w := bufio.NewWriter(f)
+ for k, v := range feat {
+ lineStr := fmt.Sprintf("%d->%f", k, v)
+ fmt.Fprintln(w, lineStr)
+ }
+ w.Flush()
+ }
+ /////////////////////////////////
+
}
buf := float32SliceAsByteSlice(feat)
+ ioutil.WriteFile("./feat.txt", buf, 0644)
c.Send(buf)
}
--
Gitblit v1.8.0