(train_or_test)
| 120 | |
| 121 | |
| 122 | def get_data(train_or_test): |
| 123 | isTrain = train_or_test == 'train' |
| 124 | augs = fbresnet_augmentor(isTrain) |
| 125 | |
| 126 | meta = dataset.ILSVRCMeta() |
| 127 | pp_mean = meta.get_per_pixel_mean() |
| 128 | augs.append(imgaug.MapImage(lambda x: x - pp_mean[16:-16, 16:-16])) |
| 129 | |
| 130 | ds = get_imagenet_dataflow(args.data, train_or_test, BATCH_SIZE, augs) |
| 131 | return ds |
| 132 | |
| 133 | |
| 134 | def get_config(): |
no test coverage detected