File size: 3,441 Bytes
b3f019f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | import os
import sys
import argparse
prj_path = os.path.join(os.path.dirname(__file__), '..')
if prj_path not in sys.path:
sys.path.append(prj_path)
from lib.test.evaluation import get_dataset
from lib.test.evaluation.running import run_dataset
from lib.test.evaluation.tracker import Tracker, trackerlist
from lib.test.evaluation.environment import env_settings
def run_tracker(tracker_name, tracker_param, run_id=None, dataset_name='otb', sequence=None, debug=0, threads=0,ckpt_path="",
num_gpus=8):
"""Run tracker on sequence or dataset.
args:
tracker_name: Name of tracking method.
tracker_param: Name of parameter file.
run_id: The run id.
dataset_name: Name of dataset (otb, nfs, uav, tpl, vot, tn, gott, gotv, lasot).
sequence: Sequence number or name.
debug: Debug level.
threads: Number of threads.
"""
dataset = get_dataset(dataset_name)
# dataset = dataset[:2]
if sequence is not None:
dataset = [dataset[sequence]]
env = env_settings()
checkpoints_path = ckpt_path
results_dir = os.path.join(env.save_dir, tracker_param)
if os.path.exists(results_dir) ==False:
os.makedirs(results_dir)
# trackers = [Tracker(tracker_name, tracker_param, dataset_name, run_id)]
trackers = trackerlist(name=tracker_name, parameter_name=tracker_param, dataset_name=dataset_name,
run_ids=run_id)
results_dir_item = os.path.join(results_dir,dataset_name )
trackers[0].results_eval_dir = results_dir_item
trackers[0].results_dir = results_dir_item
trackers[0].checkpoints_path = checkpoints_path
print("checkpoints_path_item:", checkpoints_path)
print("results_dir_item:", results_dir_item)
run_dataset(dataset, trackers, debug, threads, num_gpus=num_gpus)
def main():
os.environ['CUDA_VISIBLE_DEVICES'] = os.environ.get('CUDA_VISIBLE_DEVICES', os.environ.get('GPUS', '0'))
parser = argparse.ArgumentParser(description='Run tracker on sequence or dataset.')
parser.add_argument('--tracker_name', type=str, default="atctrack", help='Name of tracking method.')
parser.add_argument('--tracker_param', type=str, default="atctrack_base", help='Name of config file.') # atctrack_base atctrack_large
parser.add_argument('--runid', type=int, default=None, help='The run id.')
parser.add_argument('--dataset_name', type=str, default="tnl2k", help='Name of dataset (lasot_lang , tnl2k, otb99_lang, lasot_extension_subset_lang, videocube_test_tiny).')
parser.add_argument('--ckpt_path', type=str, default="./ATCTrack_b.pth.tar",
help='Name of dataset (lasot_lang , tnl2k, otb99_lang, lasot_extension_subset_lang, videocube_test_tiny).')
parser.add_argument('--sequence', type=str, default=None, help='Sequence number or name.')
parser.add_argument('--debug', type=int, default=0, help='Debug level.')
parser.add_argument('--threads', type=int, default=1, help='Number of threads.')
parser.add_argument('--num_gpus', type=int, default=1)
args = parser.parse_args()
try:
seq_name = int(args.sequence)
except:
seq_name = args.sequence
run_tracker(args.tracker_name, args.tracker_param, args.runid, args.dataset_name, seq_name, args.debug,
args.threads,args.ckpt_path, num_gpus=args.num_gpus)
if __name__ == '__main__':
main()
|