def main_vot18lt():
tracker = build_tracker()
experiment = ExperimentVOT(root_dir=VOT18_LT_ROOT_DIR,
version='LT2018',
result_dir=RESULT_DIR,
report_dir=REPORT_DIR,
experiments="unsupervised",
start_idx=args.start_idx,
end_idx=args.end_idx)
experiment.run(tracker, visualize=args.visualize_experiment)
def main_vot15():
root_dir = VOT15_ROOT_DIR
experiments = "supervised"
tracker = build_tracker()
experiment = ExperimentVOT(root_dir=root_dir,
version=2015,
result_dir=RESULT_DIR,
report_dir=REPORT_DIR,
experiments=experiments,
start_idx=args.start_idx,
end_idx=args.end_idx)
experiment.run(tracker, visualize=args.visualize_experiment)
def test_vot(self):
root_dir = os.path.join(self.data_dir, 'vot2018')
# run experiment
experiment = ExperimentVOT(root_dir)
experiment.run(self.tracker, visualize=False)
# report performance
experiment.report([self.tracker.name])
def main_vot18_threestage():
tracker = build_tracker()
root_dir = VOT18_ROOT_DIR
experiment = ExperimentVOT(root_dir=root_dir,
version=2018,
result_dir=RESULT_DIR,
report_dir=REPORT_DIR,
experiments="supervised",
start_idx=args.start_idx,
end_idx=args.end_idx)
experiment.run(tracker, visualize=args.visualize_experiment)
experiment.report([tracker.name])
def init_experiment(
dataset_type: DatasetType, dataset_dir_path: str, results_dir_path: str,
reports_dir_path: str):
params = dict(
root_dir=dataset_dir_path, result_dir=results_dir_path,
report_dir=reports_dir_path)
if dataset_type == DatasetType.OTB13:
return ExperimentOTB(version=2013, **params)
elif dataset_type == DatasetType.OTB15:
return ExperimentOTB(version=2015, **params)
elif dataset_type == DatasetType.GOT10k:
return ExperimentGOT10k(subset='val', **params)
elif dataset_type == DatasetType.VOT15:
return ExperimentVOT(version=2015, **params)
elif dataset_type == DatasetType.UAV123:
return ExperimentUAV123(**params)
else:
raise ValueError(f"unsupported dataset type {dataset_type}")
def main_vot18(reset=True):
root_dir = VOT18_ROOT_DIR
if reset:
experiments = "supervised"
else:
experiments = "unsupervised"
tracker = build_tracker()
experiment = ExperimentVOT(root_dir=root_dir,
version=2018,
result_dir=RESULT_DIR,
report_dir=REPORT_DIR,
experiments=experiments,
start_idx=args.start_idx,
end_idx=args.end_idx)
experiment.run(tracker, visualize=args.visualize_experiment)
experiment.report([tracker.name])
def example_track_val_set():
# setup tracker
tracker = IdentityTrackerRGBD()
# run experiment on validation set
experiment = ExperimentVOT(
root_dir=ROOT_DIR,
version='RGBD2019',
read_image=True,
#list_file=None,
experiments='unsupervised',
result_dir='results',
report_dir='reports')
experiment.run(tracker, visualize=False)
# report performance
experiment.report([tracker.name])
print('Finished')
otb_labels.append("OTB2015")
otb = ExperimentTColor128(
"/home/heonsong/Disk2/Dataset/TColor128",
result_dir=root + "/%s" % result_folder,
report_dir=root + "/%s" % report_folder,
)
otbs.append(otb)
otb_labels.append("TColor128")
vots = []
vot_labels = []
vot = ExperimentVOT(
"/home/heonsong/Disk2/Dataset/VOT2018",
version=2018,
result_dir=root + "/%s" % result_folder,
report_dir=root + "/%s" % report_folder,
)
vot.repetitions = 1
vots.append(vot)
vot_labels.append("VOT2018")
# evaluation performance
for otb, label in zip(otbs, otb_labels):
otb_eval(otb, trackers)
print("Evaluated %s" % label)
for vot, label in zip(vots, vot_labels):
vot_eval(vot, trackers)
vot_un_eval(vot, trackers)
print("Evaluated %s" % label)
parser.add_argument('--run_id', type=int, default=None)
parser.add_argument('--debug', type=int, default=0, help='Debug level.')
args = parser.parse_args()
TrTracker = Tracker(args.tracker_name, args.tracker_param, args.run_id)
class GOT_Tracker(GOT_Tracker):
def __init__(self):
super(GOT_Tracker, self).__init__(name='GOT_Tracker')
self.tracker = TrTracker.tracker_class(TrTracker.get_parameters())
def init(self, image, box):
image = np.array(image)
self.tracker.initialize(image, box)
def update(self, image):
image = np.array(image)
self.box = self.tracker.track(image)
return self.box
if __name__ == '__main__':
# setup tracker
tracker = GOT_Tracker()
# run experiments on VOT
experiment = ExperimentVOT('/home/wangning/Tracking/VOT-2018/workspace/sequences', version=2018)
experiment.run(tracker, visualize=False)
# report performance
experiment.report([tracker.name])
result_dir=root_path + "/results",
report_dir=root_path + "/reports",
)
run_otb(root_path, otb, aaa, trackers)
if "OTB2015" in datasets:
otb = ExperimentOTB(
root_path + "/OTB",
version=2015,
result_dir=root_path + "/results",
report_dir=root_path + "/reports",
)
run_otb(root_path, otb, aaa, trackers)
if "VOT2018Super" in datasets:
vot = ExperimentVOT(
root_path + "/VOT2018",
version=2018,
result_dir=root_path + "/results",
report_dir=root_path + "/reports",
)
vot.repetitions = 1
run_supervised(root_path, vot, aaa, trackers)
if "VOT2018Un" in datasets:
vot = ExperimentVOT(
root_path + "/VOT2018",
version=2018,
result_dir=root_path + "/results",
report_dir=root_path + "/reports",
)
vot.repetitions = 1
run_unsupervised(root_path, vot, aaa, trackers)
if "VOT2018LT" in datasets:
vot = ExperimentVOT(