(self, batch_size, num_threads, device_id, seed, image_dir)
| 106 | |
| 107 | class FlippingPipeline(CommonPipeline): |
| 108 | def __init__(self, batch_size, num_threads, device_id, seed, image_dir): |
| 109 | super().__init__(batch_size, num_threads, device_id, seed, image_dir) |
| 110 | self.flip = ops.Flip(horizontal=1) |
| 111 | |
| 112 | def define_graph(self): |
| 113 | images, labels = self.load() |