import os
from run_tracker import eval_tracking, p_config
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# test ATOM+MU
p = p_config()
p.save_training_data = False
p.tracker = 'ATOM_MU'
p.checkpoint = 20000
p.name = p.tracker
# # test ATOM
# p = p_config()
# p.save_training_data = False
# p.tracker = 'ATOM'
# p.name = p.tracker
eval_tracking('tlp', p=p, mode='all')
eval_tracking('lasot', p=p, mode='test')
eval_tracking('votlt19', p=p, mode='all')
import os
from run_tracker import eval_tracking, p_config
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# test RTMD+MU
p = p_config()
p.tracker = 'RTMD_MU'
p.save_results = False
p.name = p.tracker
eval_tracking('demo', p=p)
import os
from run_tracker import eval_tracking, p_config
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# test DiMP_LTMU
p = p_config()
p.save_results = True
p.name = 'DiMP_LTMU'
eval_tracking('votlt19', p=p)
import os
from run_tracker import eval_tracking, p_config
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# test DiMPMU
p = p_config()
p.tracker = 'Dimp_MU'
p.save_results = False
p.name = p.tracker + '_' + p.model_dir
eval_tracking('demo', p=p, mode='test')
# # test DiMP
# p = p_config()
# p.tracker = 'Dimp'
# p.save_results = False
# p.name = p.tracker
# eval_tracking('demo', p=p, mode='test')
import os
from run_tracker import eval_tracking, p_config
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
p = p_config()
p.tracker = 'Dimp'
p.model_dir = 'dimp_mu_1'
p.name = p.tracker + '_' + p.model_dir
eval_tracking('votlt19', p=p, mode='test')