From 06283a7ec8b35cd85e389f5306ec3a925896e38b Mon Sep 17 00:00:00 2001
From: zhangmeng <zhangmeng@aiotlink.com>
Date: 星期四, 19 十二月 2019 15:13:56 +0800
Subject: [PATCH] udpate
---
work/sdk/humantrack.go | 18 +++++++++++-------
1 files changed, 11 insertions(+), 7 deletions(-)
diff --git a/work/sdk/humantrack.go b/work/sdk/humantrack.go
index b2b8d80..a196757 100644
--- a/work/sdk/humantrack.go
+++ b/work/sdk/humantrack.go
@@ -29,7 +29,7 @@
mapCameraImageIndex map[string]int
recvImageCount int
index int
- images []imageWithID
+ images []*imageWithID
msgs []*work.MsgRS
}
@@ -44,7 +44,7 @@
recvImageCount: 0,
index: 0,
mapCameraImageIndex: make(map[string]int),
- images: make([]*gohumantrack.ImageHumanTracker, batchSize),
+ images: make([]*imageWithID, batchSize),
msgs: make([]*work.MsgRS, batchSize),
}
}
@@ -134,12 +134,12 @@
if i, ok := t.mapCameraImageIndex[rMsg.Msg.Cid]; ok {
if i < t.batchSize {
- t.images[i] = imageWithID{&img, rMsg.Msg.Cid}
+ t.images[i] = &imageWithID{&img, rMsg.Msg.Cid}
t.msgs[i] = &rMsg
}
} else {
if t.index < t.batchSize {
- t.images[t.index] = imageWithID{&img, rMsg.Msg.Cid}
+ t.images[t.index] = &imageWithID{&img, rMsg.Msg.Cid}
t.msgs[t.index] = &rMsg
t.mapCameraImageIndex[rMsg.Msg.Cid] = t.index
}
@@ -157,9 +157,13 @@
}
var pimg []*gohumantrack.ImageHumanTracker
- for k, v := range t.images {
- pimg = append(v.img)
- logo.Infoln("batch~~~~~~Map index: ", k, " camera: ", v.id)
+ for k, v := range t.images[:] {
+ if v != nil{
+ pimg = append(pimg, v.img)
+ logo.Infoln("batch~~~~~~Image index: ", k, " camera: ", v.id)
+ }else{
+ pimg = append(pimg, nil)
+ }
}
res, err := t.tracker.ProcessImagePointer(pimg)
t.recvImageCount = 0
--
Gitblit v1.8.0