(dp)
| 97 | aug = imgaug.AugmentorList(augmentors) |
| 98 | |
| 99 | def mapf(dp): |
| 100 | fname, cls = dp |
| 101 | im = cv2.imread(fname, cv2.IMREAD_COLOR) |
| 102 | im = aug.augment(im) |
| 103 | return im, cls |
| 104 | ds = MultiThreadMapData(ds, parallel, mapf, buffer_size=2000, strict=True) |
| 105 | ds = BatchData(ds, batch_size, remainder=True) |
| 106 | ds = MultiProcessRunnerZMQ(ds, 1) |