eladb3 / ORViT

"Object-Region Video Transformers”, Herzig et al., CVPR 2022
Apache License 2.0
42 stars 12 forks source link

__init__() got an unexpected keyword argument 'trackers_count_start' #2

Closed Randle-Github closed 2 years ago

Randle-Github commented 2 years ago

Hi, In ORViT/slowfast/utils/LinkBoxes/epickitchens.py line 116: def get_vid_boxes_linked(vid, dboxes, clear_dups_threshold, dboxes_sorted): n_sorted = _get_dboxes_sorted_length(dboxes_sorted) osort = Sort(clear_dups_threshold=clear_dups_threshold, trackers_count_start=n_sorted) where it asserts 'init() got an unexpected keyword argument 'trackers_count_start' '. In ORViT/slowfast/utils/LinkBoxes/sort.py line 228: class Sort(object): def __init__(self, max_age=1, min_hits=3, iou_threshold=0.3, clear_dups_threshold = 0.7): """ Sets key parameters for SORT """ self.max_age = max_age self.min_hits = min_hits self.iou_threshold = iou_threshold self.clear_dups_threshold = clear_dups_threshold self.trackers = [] self.frame_count = 0 self.trackers_count = 0 Maybe there're some mistakes.

eladb3 commented 2 years ago

It's indeed a mistake, fixed it, thanks! FYI you can download the epic-kitchens processed boxes file now, see slowfast/datasets/DATASET.md