def __init__(self, tracker_name='stark_s', para_name='baseline', refine_model_name='ARcm_coco_seg', threshold=0.15): self.THRES = threshold '''create tracker''' '''STARK''' tracker_info = Tracker(tracker_name, para_name, "vot20", None) params = tracker_info.get_parameters() params.visualization = False params.debug = False params.visdom_info = { 'use_visdom': False, 'server': '127.0.0.1', 'port': 8097 } self.stark = tracker_info.create_tracker(params) '''Alpha-Refine''' project_path = os.path.join(os.path.dirname(__file__), '..', '..') refine_root = os.path.join(project_path, 'ltr/checkpoints/ltr/ARcm_seg/') refine_path = os.path.join(refine_root, refine_model_name) '''2020.4.25 input size: 384x384''' self.alpha = ARcm_seg(refine_path, input_sz=384)
def __init__(self, tracker_name='stark_st', para_name='baseline'): # create tracker tracker_info = Tracker(tracker_name, para_name, "vot20", None) params = tracker_info.get_parameters() params.visualization = False params.debug = False self.tracker = tracker_info.create_tracker(params)
def __init__(self, tracker_name='stark_st', para_name='baseline'): # create tracker tracker_info = Tracker(tracker_name, para_name, "vot20lt", None) params = tracker_info.get_parameters() params.visualization = False params.debug = False params.visdom_info = { 'use_visdom': False, 'server': '127.0.0.1', 'port': 8097 } self.tracker = tracker_info.create_tracker(params)