Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
mikel-brostrom authored Aug 12, 2023
1 parent 30734eb commit ab7307a
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -293,13 +293,14 @@ while True:
xyxys = tracks[:, 0:4].astype('int') # float64 to int
ids = tracks[:, 4].astype('int') # float64 to int
confs = tracks[:, 5]
clss = tracks[:, 6]
ind = tracks[:, 7]
clss = tracks[:, 6].astype('int') # float64 to int
inds = tracks[:, 7].astype('int') # float64 to int

# in case you have segmentations or poses alongside with your detections you can use
# the ind variable in order to identify which track is associated to each seg or pose by:
# segs = segs[ind]
# poses = poses[ind]
# segs = segs[inds]
# poses = poses[inds]
# you can then zip them together: zip(tracks, poses)

# print bboxes with their associated id, cls and conf
if ts.shape[0] != 0:
Expand Down

0 comments on commit ab7307a

Please sign in to comment.