| 78 | |
| 79 | |
| 80 | def build_dataset(args): |
| 81 | if not hasattr(args, "split"): |
| 82 | args.split = 0 |
| 83 | if not hasattr(args, "pretrain") or not args.pretrain: |
| 84 | dataset = build_dataset_from_name(args.dataset, args.split) |
| 85 | else: |
| 86 | dataset = build_dataset_pretrain(args) |
| 87 | |
| 88 | if hasattr(dataset, "num_classes") and dataset.num_classes > 0: |
| 89 | args.num_classes = dataset.num_classes |
| 90 | if hasattr(dataset, "num_features") and dataset.num_features > 0: |
| 91 | args.num_features = dataset.num_features |
| 92 | |
| 93 | return dataset |
| 94 | |
| 95 | |
| 96 | def build_dataset_from_path(data_path, dataset=None): |