def vot_lt_2020_local3():
trackers = trackerlist('dimp', 'dimp50', None) + \
trackerlist('dimp', 'dimp50_vot', None) + \
trackerlist('dimp', 'dimp50_vot19', None)
dataset = get_dataset('vot')
return trackers, dataset
def uav_test():
# Run DiMP18, ATOM and ECO on the UAV dataset
trackers = trackerlist('dimp', 'dimp18', range(1)) + \
trackerlist('dimp', 'dimp18_vot', range(1))
dataset = get_dataset('uav')
return trackers, dataset
def vot_lt_2020_local4():
trackers = trackerlist('dimp', 'prdimp18', None)+ \
trackerlist('dimp', 'prdimp50', None)+ \
trackerlist('dimp', 'super_dimp', None)
dataset = get_dataset('vot')
return trackers, dataset
def main():
parser = argparse.ArgumentParser(
description='Generate success and precision plots')
parser.add_argument('tracker_name', type=str)
parser.add_argument('tracker_param', type=str)
parser.add_argument('--run_id', type=int, default=None)
parser.add_argument('--dataset', type=str)
args = parser.parse_args()
trackers = []
display_name = args.tracker_param + str(args.run_id)
trackers.extend(
trackerlist(args.tracker_name, args.tracker_param, args.run_id,
display_name))
dataset = get_dataset(args.dataset)
report_name = args.tracker_param + '_' + args.dataset
plot_results(trackers,
dataset,
report_name,
merge_results=True,
plot_types=('success', 'prec'),
skip_missing_seq=True,
force_evaluation=True,
plot_bin_gap=0.05,
exclude_invalid_frames=False)
def eval_atom_compression_lasot_trackingnet():
# run atom_cfkd on LaSOT, TrackingNet, VOT2018
trackers = trackerlist('atom', 'atom_compression')
dataset = get_dataset('lasot', 'trackingnet')
return trackers, dataset
def transt_readout_test_encoder_mult_from_front():
trackers = trackerlist('transt_readout_test_encoder_mult_from_front',
'default', range(1))
dataset = get_dataset(
'got10k_test') # , 'lasot') # _val, 'got10k_ltrval')
# dataset = get_dataset('got10k_test') # _val, 'got10k_ltrval')
return trackers, dataset
def eval_drnet_cfkd_lasot_trackingnet():
# run atom_cfkd on LaSOT, TrackingNet, VOT2018
trackers = trackerlist('drnet', 'drnet_cfkd')
dataset = get_dataset('lasot', 'trackingnet')
return trackers, dataset
def transt_readout_test_encoder_mult_control():
trackers = trackerlist('transt_circuit_control', 'default', range(1))
# dataset = get_dataset('got10k_test', 'lasot') # _val, 'got10k_ltrval')
# dataset = get_dataset('lasot') # _val, 'got10k_ltrval')
# dataset = get_dataset('got10k_test') # _val, 'got10k_ltrval')
dataset = get_dataset('got10k_test')
return trackers, dataset
def main():
trackers = []
trackers.extend(
trackerlist('dimpRank', 'dimpRank18', range(0, 1), 'dimpRank'))
# trackers.extend(trackerlist('ECO', 'default_deep', range(1), 'ECO'))
# trackers.extend(trackerlist('dimpFMFAttnAw_49', 'dimp50', range(1), 'TRAT'))
# trackers.extend(trackerlist('MDNet', 'default', range(1), 'MDNet'))
dataset = get_dataset('otb')
sequence = [dataset['Basketball']]
playback_results(trackers, sequence[0])
def vot_lt_2020_all_test():
# Run DiMP18, ATOM and ECO on the UAV dataset
trackers = trackerlist('dimp', 'dimp18', range(1)) + \
trackerlist('dimp', 'dimp18_vot', range(1))+ \
trackerlist('dimp', 'dimp50', range(1))+ \
trackerlist('dimp', 'dimp50_vot', range(1))+ \
trackerlist('dimp', 'dimp50_vot19', range(1))+ \
trackerlist('dimp', 'prdimp18', range(1))+ \
trackerlist('dimp', 'prdimp50', range(1))+ \
trackerlist('dimp', 'super_dimp', range(1))
dataset = get_dataset('vot')
return trackers, dataset
def transt_control_normal():
trackers = trackerlist('transt_control_normal', 'default', range(1))
dataset = get_dataset('got10k_test') # _val, 'got10k_ltrval')
return trackers, dataset
from pytracking.evaluation import Tracker, get_dataset, trackerlist, Sequence
from pytracking.analysis.playback_results import playback_results
trackers = []
#
# trackers.extend(trackerlist('TratCatAttn31-32', 'default2', range(0,1), 'TratCatAttn32_1'))
# trackers.extend(trackerlist('TratCatAttn31-32', 'default2', range(1,2), 'TratCatAttn32_2'))
# trackers.extend(trackerlist('TratCatAttn31-32', 'default2', range(2,3), 'TratCatAttn32_3'))
# trackers.extend(trackerlist('TratCatAttn31-32', 'default2', range(3,4), 'TratCatAttn32_4'))
# trackers.extend(trackerlist('TratCatAttn31-32', 'default2', range(4,5), 'TratCatAttn32_5'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_49', range(1,2), 'dimp50_4_2'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_49', range(2,3), 'dimp50_46_1'))
trackers.extend(trackerlist('RankingFast', 'default', range(1), 'RafnkinFast'))
filter_criteria = None
dataset = get_dataset('otb')
# sequence = [dataset['Basketball']]
# print_per_sequence_results(trackers, dataset, 'nfs', merge_results=True, filter_criteria=filter_criteria, force_evaluation=False)
# playback_results(trackers, sequence[0])
# plot_results(trackers, dataset, 'uav', merge_results=True, plot_types=('success', 'prec', 'norm_prec'),
# skip_missing_seq=False, force_evaluation=True, plot_bin_gap=0.05, exclude_invalid_frames=False)
print_results(trackers,
dataset,
'otb',
merge_results=True,
plot_types=('success', 'prec', 'norm_prec'))
@author: mlcv
"""
import os
import sys
import matplotlib.pyplot as plt
plt.rcParams['figure.figsize'] = [14, 12]
sys.path.append('../..')
from pytracking.analysis.plot_results import plot_results, print_results, print_per_sequence_results
from pytracking.evaluation import Tracker, get_dataset, trackerlist, Sequence
from pytracking.analysis.playback_results import playback_results
trackers = []
trackers.extend(trackerlist('ECO', 'default_deep', range(0, 1), 'ECO'))
trackers.extend(trackerlist('UPDT', 'default', range(0, 9), 'UPDT'))
trackers.extend(trackerlist('MDNet', 'default', range(0, 1), 'MDNet'))
trackers.extend(trackerlist('CCOT', 'default', range(0, 1), 'CCOT'))
# trackers.extend(trackerlist('DaSiamRPN', 'default', range(0,1), 'DaSiamRPN'))
# trackers.extend(trackerlist('SiamRPN++', 'default', range(0,1), 'SiamRPN++'))
trackers.extend(trackerlist('dimp', 'dimp50', range(0, 5), 'DiMP50'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_49', range(1,2), 'TRAT'))
trackers.extend(trackerlist('TRAT', 'trat', range(0, 3), 'TRAT'))
# trackers.extend(trackerlist('atom_R50_ours', 'default', range(0,5), 'ATOM50*'))
trackers.extend(trackerlist('atom', 'default', range(0, 5), 'ATOM'))
# trackers.extend(trackerlist('siamban_results', 'SiamBAN_LaSOT', range(0,1), 'SiamBAN'))
# trackers.extend(trackerlist('Roam', 'LaSOT-ROAM++', range(0,1), 'Roam++'))
# trackers.extend(trackerlist('Roam', 'LaSOT-ROAM', range(0,1), 'Roam'))
import os
import sys
import matplotlib.pyplot as plt
plt.rcParams['figure.figsize'] = [14, 8]
from pytracking.analysis.plot_results import plot_results, print_results, print_per_sequence_results
from pytracking.evaluation import Tracker, get_dataset, trackerlist
from pytracking.analysis.playback_results import playback_two_results, playback_results
trackers = []
trackers.extend(trackerlist('bh', 'default', 1, 'bh'))
dataset = get_dataset('otb')
"""
plot_results(trackers, dataset, 'OTB', merge_results=True, plot_types=('success', 'prec'),
skip_missing_seq=False, force_evaluation=True, plot_bin_gap=0.05, exclude_invalid_frames=False)
"""
for seq in dataset:
playback_two_results(trackers, seq)
# trackers.extend(trackerlist('dimp', 'dimp50', range(0,5), 'dimp_R50'))
# trackers.extend(trackerlist('dimpSF', 'dimp50', range(0,5), 'dimpSF_R50'))
# trackers.extend(trackerlist('dimpSFAttn', 'dimp50', range(0,5), 'dimpSFAttn_R50'))
# trackers.extend(trackerlist('dimpSF2F', 'dimp50', range(0,5), 'dimpSF2F'))
# trackers.extend(trackerlist('dimpFMFAttn', 'dimp50', range(0,5), 'dimpFMFAttn'))
# trackers.extend(trackerlist('dimpFMFConAttn', 'dimp50', range(0,5), 'dimpFMFConAttn'))
# trackers.extend(trackerlist('dimpFMFZipAttn', 'dimp50', range(0,5), 'dimpFMFZipAttn'))
# trackers.extend(trackerlist('dimpFMFAttn_k3', 'dimp50', range(0,5), 'dimpFMFAttn_k3'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50', range(0,3), 'dimpFMFAttnAw_48'))
# trackers.extend(trackerlist('dimpFMFAttnAw_k5_50', 'dimp50', range(0,5), 'dimpFMFAttnAw_k5'))
# trackers.extend(trackerlist('dimpFMF', 'dimp50', range(0,5), 'dimpFMF'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_45', range(0,3), 'dimp50_45'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_46', range(0,3), 'dimp50_46'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_47', range(0,3), 'dimp50_47'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_48', range(0,3), 'dimp50_48'))
trackers.extend(trackerlist('Ranking', 'default', range(1, 2), 'Ranking'))
# trackers.extend(trackerlist('Ranking_aspect', 'default', range(0,3), 'Ranking_aspect'))
# trackers.extend(trackerlist('trat', 'default', range(0,5), 'TRAT_R50'))
# trackers.extend(trackerlist('trat_CF', 'default', range(0,5), 'TRAT_R50_CF_sum'))
# trackers.extend(trackerlist('trat_CF_average', 'default', range(0,5), 'TRAT_R50_CF_average'))
# trackers.extend(trackerlist('trat_CF_attn_ECA_k3', 'default', range(0,5), 'TRAT_R50_CF_ECA_k3'))
# trackers.extend(trackerlist('trat_CF_attn_last', 'default', range(0,5), 'TRAT_R50_CF_ECA_last'))
# trackers.extend(trackerlist('TRATSF', 'default', range(0,5), 'TRATSF_R50'))
# trackers.extend(trackerlist('TRATSFAttn_Eca_R50_k5_dimp', 'default', range(0,5), 'TRATSFAttn_Eca_R50_k5_dimp'))
# trackers.extend(trackerlist('TRATSFAttn_k3', 'default', range(0,5), 'TRATSFAttn_Eca_R50_k3'))
# trackers.extend(trackerlist('TRATSFAttn_k5', 'default', range(0,5), 'TRATSFAttn_Eca_R50_k5'))
# trackers.extend(trackerlist('TratCatAttn_AllTrained', 'default', range(0,5), 'TratCatAttn_AllTrained'))
# trackers.extend(trackerlist('TratCatAttn_Trained', 'default', range(0,5), 'TratCatAttn'))
# trackers.extend(trackerlist('TratCatAttn_LR0.0075', 'default', range(0,5), 'TratCatAttn_LR0.0075'))
# trackers.extend(trackerlist('TratCatAttn32_LR-085-115', 'default', range(0,5), 'TratCatAttn32_LR-085'))
import sys
import matplotlib.pyplot as plt
plt.rcParams['figure.figsize'] = [14, 12]
sys.path.append('../..')
from pytracking.analysis.plot_resultsNew import plot_results, print_results, print_per_sequence_results
from pytracking.evaluation import Tracker, get_dataset, trackerlist, Sequence
from pytracking.analysis.playback_results import playback_results
trackers = []
# trackers.extend(trackerlist('Ranking', 'default', range(0,5), 'RankingSF'))
# trackers.extend(trackerlist('RankingFlow', 'default', range(0,3), 'RankingFlow'))
# trackers.extend(trackerlist('Ranking', 'default', range(2,3), 'RankingSF1'))
# trackers.extend(trackerlist('Ranking', 'default2', range(4,5), 'RankingAF'))
# trackers.extend(trackerlist('Ranking', 'default3', range(0,5), 'H_Ranking'))
trackers.extend(trackerlist('RFSF', 'default', range(0, 3),
'RankingAF')) # 1 mdnet
trackers.extend(trackerlist('RFSF', 'default2', range(0, 3), 'RankingSF'))
# trackers.extend(trackerlist('Ranking', 'default', range(0,5), 'RankingSF'))
# trackers.extend(trackerlist('Ranking', 'default2', range(0,5), 'RankingAF'))
trackers.extend(trackerlist('ECO', 'default_deep', range(0, 1), 'ECO'))
# trackers.extend(trackerlist('UPDT', 'default', range(0,9), 'UPDT'))
# trackers.extend(trackerlist('CCOT', 'default', range(0,1), 'CCOT'))
# trackers.extend(trackerlist('atom', 'default', range(0,5), 'ATOM'))
trackers.extend(trackerlist('MDNet', 'default', range(0, 1), 'MDNet'))
# trackers.extend(trackerlist('DaSiamRPN', 'default', range(0,1), 'DaSiamRPN'))
# trackers.extend(trackerlist('SiamRPN++', 'default', range(0,1), 'SiamRPN++'))
# trackers.extend(trackerlist('dimp', 'dimp50', range(0,5), 'DiMP50'))
# trackers.extend(trackerlist('dimpFMFAttnAw', 'dimp50_49', range(1,2), 'TRAT'))
def dimp_lasot():
# Run three runs of ATOM on NFS and UAV datasets
trackers = trackerlist('dimp', 'dimp18', range(2))
dataset = get_dataset('lasot')
return trackers, dataset
def transt_readout_test_v1():
trackers = trackerlist('transt_readout_test_v1', 'default', range(1))
dataset = get_dataset('got10k_test') # _val, 'got10k_ltrval')
return trackers, dataset
def trsiam_otb():
trackers = trackerlist('trdimp', 'trsiam', range(1))
dataset = get_dataset('otb')
return trackers, dataset
def trsiam_trackingnet():
trackers = trackerlist('trdimp', 'trsiam', range(1))
dataset = get_dataset('trackingnet')
return trackers, dataset
def vot_lt_2020_local2():
trackers = trackerlist('dimp', 'dimp18', None) + \
trackerlist('dimp', 'dimp18_vot', None)
dataset = get_dataset('vot')
return trackers, dataset
def eco_lasot():
# Run three runs of ATOM on NFS and UAV datasets
trackers = trackerlist('eco', 'default', range(1))
dataset = get_dataset('lasot')
return trackers, dataset
def transt_readout_mixed():
# Evaluate GOT on a circuit KYS I trained
trackers = trackerlist('transt_readout_mixed', 'default', range(1))
dataset = get_dataset('got10k_test') # _val, 'got10k_ltrval')
return trackers, dataset
def trsiam_lasot():
trackers = trackerlist('trdimp', 'trsiam_lasot', range(1))
dataset = get_dataset('lasot')
return trackers, dataset
def rtranst():
trackers = trackerlist('rtranst', 'default', range(1))
dataset = get_dataset('got10k_test') # _val, 'got10k_ltrval')
return trackers, dataset
def Ranking():
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# Run three runs of ATOM on NFS and UAV datasets
trackers = trackerlist('Ranking', 'default', range(2))
dataset = get_dataset('otb')
return trackers, dataset
def atom_nfs_uav():
# Run three runs of ATOM on NFS and UAV datasets
trackers = trackerlist('atom', 'default', range(3))
dataset = get_dataset('nfs', 'uav')
return trackers, dataset
def dimp50_lasot():
trackers = trackerlist('dimp', 'super_dimp', range(1))
dataset = get_dataset('lasot')
return trackers, dataset
