From 114ee011af25b727409541dbab8c7ac85f266c0c Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期四, 19 十二月 2019 15:08:02 +0800
Subject: [PATCH] update
---
work/sdk/humantrack.go | 24 +++++++++++++++++++-----
1 files changed, 19 insertions(+), 5 deletions(-)
diff --git a/work/sdk/humantrack.go b/work/sdk/humantrack.go
index 94e45ac..b2b8d80 100644
--- a/work/sdk/humantrack.go
+++ b/work/sdk/humantrack.go
@@ -13,6 +13,11 @@
"basic.com/valib/gogpu.git"
)
+type imageWithID struct {
+ img *gohumantrack.ImageHumanTracker
+ id string
+}
+
// HumanTracker track
type HumanTracker struct {
tracker *gohumantrack.HumanTracker
@@ -24,7 +29,7 @@
mapCameraImageIndex map[string]int
recvImageCount int
index int
- images []*gohumantrack.ImageHumanTracker
+ images []imageWithID
msgs []*work.MsgRS
}
@@ -129,12 +134,12 @@
if i, ok := t.mapCameraImageIndex[rMsg.Msg.Cid]; ok {
if i < t.batchSize {
- t.images[i] = &img
+ t.images[i] = imageWithID{&img, rMsg.Msg.Cid}
t.msgs[i] = &rMsg
}
} else {
if t.index < t.batchSize {
- t.images[t.index] = &img
+ t.images[t.index] = imageWithID{&img, rMsg.Msg.Cid}
t.msgs[t.index] = &rMsg
t.mapCameraImageIndex[rMsg.Msg.Cid] = t.index
}
@@ -147,8 +152,17 @@
return
}
- res, err := t.tracker.ProcessImagePointer(t.images[:])
- t.recvImageCount = 0
+ for k, v := range t.mapCameraImageIndex {
+ logo.Infoln("batch~~~~~~Map index: ", v, " camera: ", k)
+ }
+
+ var pimg []*gohumantrack.ImageHumanTracker
+ for k, v := range t.images {
+ pimg = append(v.img)
+ logo.Infoln("batch~~~~~~Map index: ", k, " camera: ", v.id)
+ }
+ res, err := t.tracker.ProcessImagePointer(pimg)
+ t.recvImageCount = 0
if err != nil {
logo.Infoln("batch~~~~~~Track Failed: ", err)
--
Gitblit v1.8.0