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)
def run_video(tracker_name,
tracker_param,
videofile,
optional_box=None,
debug=None,
save_results=False):
"""Run the tracker on your webcam.
args:
tracker_name: Name of tracking method.
tracker_param: Name of parameter file.
debug: Debug level.
"""
tracker = Tracker(tracker_name, tracker_param, "video")
tracker.run_video(videofilepath=videofile,
optional_box=optional_box,
debug=debug,
save_results=save_results)