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