Exemplo n.º 1
0
 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)
Exemplo n.º 2
0
 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)
Exemplo n.º 3
0
 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)