modify byte.trackerupdate, add list first_finded

This commit is contained in:
王庆刚
2024-05-25 18:35:53 +08:00
parent d6f3693d3f
commit dff029de20
17 changed files with 132 additions and 80 deletions

View File

@ -216,7 +216,7 @@ class Track:
trajdist_min, trajdist_max,
trajlen_rate, trajdist_rate]
def compute_static_fids(self, det_y, STATIC_THRESH = 8):
def pt_state_fids(self, det_y, STATIC_THRESH = 8):
'''
前摄时y一般选择为 box 的 y1 坐标,且需限定商品在购物车内。
inputs
@ -324,12 +324,15 @@ class doTracks:
for t_id in self.trackID:
# print(f"The ID is: {t_id}")
idx = np.where(track_ids == t_id)[0]
box = self.bboxes[idx, :]
box = self.bboxes[idx, :]
assert len(set(box[:, 7])) == len(box), "Please check!!!"
lboxes.append(box)
return lboxes
'''
def classify(self):
tracks = self.tracks
@ -354,10 +357,11 @@ class doTracks:
self.Static.extend(static_tracks)
'''剔除静止目标后的 tracks'''
# 剔除静止目标后的 tracks
tracks = self.sub_tracks(tracks, static_tracks)
return tracks
'''