(self, args, split)
| 206 | |
| 207 | class ValidateData(data.Dataset): |
| 208 | def __init__(self, args, split): |
| 209 | self.root = os.path.join(args.validate_data_dir, split) |
| 210 | if not os.path.exists(self.root): |
| 211 | raise colored('[Error: ]', 'red') + 'file path {:s} is not exist!'.format(self.root) |
| 212 | |
| 213 | self.images_dir = os.path.join(self.root, 'images') |
| 214 | self.flows_dir = os.path.join(self.root, 'flows') |
| 215 | self.masks_dir = os.path.join(self.root, 'masks') |
| 216 | |
| 217 | |
| 218 | def __getitem__(self, index): |
nothing calls this directly
no outgoing calls
no test coverage detected