(is_train, args)
| 18 | |
| 19 | |
| 20 | def build_dataset(is_train, args): |
| 21 | transform = build_transform(is_train, args) |
| 22 | |
| 23 | root = os.path.join(args.data_path, 'train' if is_train else 'val') |
| 24 | dataset = datasets.ImageFolder(root, transform=transform) |
| 25 | |
| 26 | print(dataset) |
| 27 | |
| 28 | return dataset |
| 29 | |
| 30 | |
| 31 | def build_transform(is_train, args): |
nothing calls this directly
no test coverage detected