def load_annotations(self):
assert self.data_type == 'mot'
gt_filename = os.path.join(
self.data_root, self.seq_name, 'gt', 'gt.txt')
self.gt_frame_dict = read_results(
gt_filename, self.data_type, is_gt=True)
self.gt_ignore_frame_dict = read_results(
gt_filename, self.data_type, is_ignore=True)
def eval_file(self, filename):
self.reset_accumulator()
result_frame_dict = read_results(filename, self.data_type, is_gt=False)
frames = sorted(list(set(self.gt_frame_dict.keys()) | set(result_frame_dict.keys())))
for frame_id in frames:
trk_objs = result_frame_dict.get(frame_id, [])
trk_tlwhs, trk_ids = unzip_objs(trk_objs)[:2]
self.eval_frame(frame_id, trk_tlwhs, trk_ids, rtn_events=False)
return self.acc
def load_detrac_annotations(self):
assert self.data_type == 'detrac'
gt_filename = os.path.join(self.data_root,
'DETRAC-Test-Annotations-XML',
self.seq_name + '.xml')
self.gt_frame_dict = read_results(gt_filename,
self.data_type,
is_gt=True,
is_ignore=False)
self.gt_ignore_frame_dict = dict()
