()
| 63 | |
| 64 | |
| 65 | def get_data(): |
| 66 | d1 = dataset.SVHNDigit('train') |
| 67 | d2 = dataset.SVHNDigit('extra') |
| 68 | data_train = RandomMixData([d1, d2]) |
| 69 | data_test = dataset.SVHNDigit('test', shuffle=False) |
| 70 | |
| 71 | augmentors = [ |
| 72 | imgaug.Resize((40, 40)), |
| 73 | imgaug.Brightness(30), |
| 74 | imgaug.Contrast((0.5, 1.5)), |
| 75 | ] |
| 76 | data_train = AugmentImageComponent(data_train, augmentors) |
| 77 | data_train = BatchData(data_train, 128) |
| 78 | data_train = MultiProcessRunner(data_train, 5, 5) |
| 79 | |
| 80 | augmentors = [imgaug.Resize((40, 40))] |
| 81 | data_test = AugmentImageComponent(data_test, augmentors) |
| 82 | data_test = BatchData(data_test, 128, remainder=True) |
| 83 | return data_train, data_test |
| 84 | |
| 85 | |
| 86 | if __name__ == '__main__': |
no test coverage detected
searching dependent graphs…