modify byte.trackerupdate, add list first_finded
This commit is contained in:
@ -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
|
||||
'''
|
||||
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user