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