def __init__(self, path, exrta_paras_dict): # type: (str) -> None """ Class constructor. :param path: The folder in which UCSD is stored. """ super(UCSDPed2_deepSVDD, self).__init__() self.path = join(path, 'UCSDped2') self.window_size = int(exrta_paras_dict['test_window_size']) # Test directory self.test_dir = join(self.path, 'Test') # Transform self.transform = transforms.Compose( [ToFloatTensor3D(), ToCrops(self.raw_shape, self.crop_shape)]) # Load all test ids self.test_ids = self.load_test_ids() # Other utilities self.cur_len = 0 self.cur_video_id = None self.cur_video_frames = None self.cur_video_gt = None
def __init__(self, path, exrta_paras_dict): # type: (str) -> None """ Class constructor. :param path: The folder in which UCSD is stored. """ super(UCSDPed2_deepSVDD_TRAIN, self).__init__() self.path = join(path, 'UCSDped2') # 这个path通用于 Train or Test self.window_size = int(exrta_paras_dict['train_window_size']) # Train directory self.train_dir = join(self.path, 'Train') # Transform # TODO: Training还需要其他的Data Augmentation 吗? 问下论文作者! self.transform = transforms.Compose( [ToFloatTensor3D(), ToCrops(self.raw_shape, self.crop_shape)]) # 上面考虑加个去背景mog,从shanghaitech那里找代码, TODO # Load all train ids self.train_ids = self.load_train_ids() # Other utilities self.cur_len = 0 self.cur_video_id = None # Train 下面的所有 TrainXXX(XXX:001~016) 目录名 self.cur_video_frames = None
def __init__(self, path): # type: (str) -> None """ Class constructor. :param path: The folder in which UCSD is stored. """ super(UCSDPed1_TRAIN, self).__init__() self.path = join(path, 'UCSDped1') # 这个path通用于 Train or Test # Train directory self.train_dir = join(self.path, 'Train') # Transform # TODO: Training还需要其他的Data Augmentation 吗? 问下论文作者! self.transform = transforms.Compose( [ToFloatTensor3D(), ToCrops(self.raw_shape, self.crop_shape)]) # Load all train ids self.train_ids = self.load_train_ids() # Other utilities self.cur_len = 0 self.cur_video_id = None # Train 下面的所有 TrainXXX(XXX:001~016) 目录名 self.cur_video_frames = None