def example_track_val_set():
# setup tracker
tracker = IdentityTracker()
# run experiment on validation set
experiment = ExperimentGOT10k(root_dir=ROOT_DIR,
subset='val',
result_dir='results',
report_dir='reports')
experiment.run(tracker, visualize=False)
# report performance
experiment.report([tracker.name])
def example_track_test_set():
# setup tracker
tracker = IdentityTracker()
# run experiment on test set
experiment = ExperimentGOT10k(root_dir=ROOT_DIR,
subset='test',
result_dir='results',
report_dir='reports')
experiment.run(tracker, visualize=False)
# a ".zip" file will be generated ready for submission
# follow the guide to submit your results to
# http://got-10k.aitestunion.com/
experiment.report([tracker.name])
class TestTrackers(unittest.TestCase):
def setUp(self):
self.data_dir = 'data'
self.tracker = IdentityTracker()
def tearDown(self):
pass
def test_identity_tracker(self):
# setup dataset
root_dir = os.path.join(self.data_dir, 'GOT-10k')
dataset = GOT10k(root_dir, subset='val')
# run experiment
img_files, anno = random.choice(dataset)
boxes, times = self.tracker.track(img_files, anno[0], visualize=True)
self.assertEqual(boxes.shape, anno.shape)
self.assertEqual(len(times), len(anno))
def setUp(self):
self.data_dir = 'data'
self.tracker = IdentityTracker()