(self, args, split='train')
| 186 | |
| 187 | class SequenceDataset(data.Dataset): |
| 188 | def __init__(self, args, split='train'): |
| 189 | root = os.path.join(args.training_data_dir, args.dataset, split) |
| 190 | if not os.path.exists(root): |
| 191 | raise colored('[Error: ]', 'red') + 'file path {:s} is not exist!'.format(root) |
| 192 | |
| 193 | self.images_list = glob(os.path.join(root,'*/*/*/*/*.jpg')) |
| 194 | |
| 195 | def random_select(self): |
| 196 | index = random.randint(0, len(self.images_list)-1) |
nothing calls this directly
no outgoing calls
no test coverage detected