Skip to content

Commit

Permalink
Update tcn_hpl/data/vectorize/locs_and_confs.py
Browse files Browse the repository at this point in the history
Co-authored-by: Paul Tunison <[email protected]>
  • Loading branch information
cameron-a-johnson and Purg authored Nov 5, 2024
1 parent 65675a6 commit 4d80561
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tcn_hpl/data/vectorize/locs_and_confs.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@ def get_top_k_indexes_of_one_obj_type(f_dets, k, label_ind):
# Sort labels by score values.
sorted_inds = [i[1] for i in sorted(zip(filtered_scores, filtered_idxs))]
return sorted_inds[:k]

@staticmethod
def append_vector(frame_feat, i, number):
frame_feat[i] = number
Expand Down

0 comments on commit 4d80561

Please sign in to comment.