(train_or_test, cifar_classnum)
| 80 | |
| 81 | |
| 82 | def get_data(train_or_test, cifar_classnum): |
| 83 | isTrain = train_or_test == 'train' |
| 84 | if cifar_classnum == 10: |
| 85 | ds = dataset.Cifar10(train_or_test) |
| 86 | else: |
| 87 | ds = dataset.Cifar100(train_or_test) |
| 88 | if isTrain: |
| 89 | augmentors = [ |
| 90 | imgaug.RandomCrop((30, 30)), |
| 91 | imgaug.Flip(horiz=True), |
| 92 | imgaug.Brightness(63), |
| 93 | imgaug.Contrast((0.2, 1.8)), |
| 94 | imgaug.MeanVarianceNormalize(all_channel=True) |
| 95 | ] |
| 96 | else: |
| 97 | augmentors = [ |
| 98 | imgaug.CenterCrop((30, 30)), |
| 99 | imgaug.MeanVarianceNormalize(all_channel=True) |
| 100 | ] |
| 101 | ds = AugmentImageComponent(ds, augmentors) |
| 102 | ds = BatchData(ds, 128, remainder=not isTrain) |
| 103 | if isTrain: |
| 104 | ds = MultiProcessRunnerZMQ(ds, 5) |
| 105 | return ds |
| 106 | |
| 107 | |
| 108 | def get_config(cifar_classnum): |
no test coverage detected