def main():
global args, logger, v_id
args = parser.parse_args()
init_log('global', logging.INFO)
if args.log != "":
add_file_handler('global', args.log, logging.INFO)
logger = logging.getLogger('global')
logger.info(args)
# setup dataset
dataset = load_dataset(args.dataset)
total_lost = 0 # VOT
speed_list = []
trackers = ['STRCF']
for tracker_type in trackers:
for v_id, video in enumerate(dataset.keys(), start=1):
lost, speed = track_vot(tracker_type, dataset[video])
total_lost += lost
speed_list.append(speed)
logger.info('Total Lost: {:d}'.format(total_lost))
logger.info('Mean Speed: {:.2f} FPS'.format(np.mean(speed_list)))
def main():
global args, logger, v_id
args = parser.parse_args()
init_log('global', logging.INFO)
if args.log != "":
add_file_handler('global', args.log, logging.INFO)
logger = logging.getLogger('global')
logger.info(args)
#trackers=['SAMF','STRCF','Staple-CA','DCF','Staple','DSST','ECO-HC','ECO','BACF','CSRDCF']
#trackers=['DCF','Staple','ECO-HC','BACF']
trackers = ['DCF_MIX', 'Staple_MIX'] #,'ECO-HC_MIX','BACF_MIX']
if not os.path.exists("./otb_dataset.npy"):
dataset = DatasetFactory.create_dataset(
name='OTB100',
dataset_root='/mnt/nvme/siamf/testing_dataset/OTB100',
load_img=True)
np.save("./otb_dataset.npy", dataset)
else:
dataset = np.load("./otb_dataset.npy")
#dataset = np.load("./otb_subset.npy")
print('start:')
for tracker_type in trackers:
print("tracker_name:{}".format(tracker_type))
track_otb(tracker_type, dataset)
def main():
global args, logger, v_id
args = parser.parse_args()
init_log('global', logging.INFO)
if args.log != "":
add_file_handler('global', args.log, logging.INFO)
logger = logging.getLogger('global')
logger.info(args)
trackers = ['MKCFup']
dataset = DatasetFactory.create_dataset(name='OTB100',
dataset_root='../dataset/OTB100',
load_img=True)
for tracker_type in trackers:
print(tracker_type)
track_otb(tracker_type, dataset)