def __init__(self, split,shared_dict):
self.all_videos = get_image_list(args.data_root, split)
#https://github.com/Rudrabha/Wav2Lip/issues/50
#per epoch must be more than 1000 iterations
# if len(self.all_videos) < 500:
# self.all_videos = self.all_videos*((500//len(self.all_videos)) + 1)
self.shared_dict = shared_dict
def __init__(self, split):
self.all_videos = get_image_list('/root/project/public/data/lrs2/lrs2_landmark_210406', split) # make video file list ex)00001
self.all_videos2 = get_image_list('/root/project/public/data/lrs2/lrs2_preprocessed', split)
def __init__(self, split):
self.all_videos = get_image_list(args.data_root, split)
def __init__(self, data_root, split, random=False):
self.all_videos = get_image_list(data_root, split)
if not random:
np.random.seed(23)
def __init__(self, split, shared_dict):
self.all_videos = get_image_list(args.data_root, split)
self.shared_dict = shared_dict
def __init__(self, split, seed=None, shuffle=True):
self.all_videos = get_image_list(args.data_root, split)
self.seed = seed
self.is_test = split == 'val'
self.shuffle = shuffle
def __init__(self, split, seed=None):
self.all_videos = get_image_list(args.data_root, split)
self.seed = seed
