()
| 110 | |
| 111 | |
| 112 | def get_data(): |
| 113 | assert args.data |
| 114 | imgs = glob.glob(args.data + '/*.jpg') |
| 115 | ds = ImageFromFile(imgs, channel=3, shuffle=True) |
| 116 | ds = AugmentImageComponent(ds, get_augmentors()) |
| 117 | ds = BatchData(ds, args.batch) |
| 118 | ds = MultiProcessRunnerZMQ(ds, 5) |
| 119 | return ds |
| 120 | |
| 121 | |
| 122 | def sample(model, model_path, output_name='gen/gen'): |
no test coverage detected