def initialize(self, img_RGB, mask):
region = rect_from_mask(mask)
self.H, self.W, _ = img_RGB.shape
gt_bbox_np = np.array(region).astype(np.float32)
'''Initialize dimp for specific video'''
_ = self.base_tracker.initialize_seq(img_RGB, gt_bbox_np)
'''initilize refinement module for specific video'''
self.alpha.initialize(img_RGB, np.array(gt_bbox_np))
def initialize(self, img_RGB, mask):
region = rect_from_mask(mask)
self.H, self.W, _ = img_RGB.shape
gt_bbox_np = np.array(region).astype(np.float32)
'''Initialize dimp for specific video'''
gt_bbox_torch = torch.from_numpy(gt_bbox_np)
init_info = {}
init_info['init_bbox'] = gt_bbox_torch
_ = self.base_tracker.initialize(img_RGB, init_info)
def initialize(self, img_RGB, mask):
region = rect_from_mask(mask)
self.H, self.W, _ = img_RGB.shape
gt_bbox_np = np.array(region).astype(np.float32)
'''Initialize dimp for specific video'''
_ = self.base_tracker.init(img_RGB, region)