Args: ds (DataFlow): input DataFlow. augmentors (AugmentorList): a list of :class:`imgaug.ImageAugmentor` instance to be applied in order. index: tuple of indices of the image components. coords_index: tuple of indices of the coordinates compo
(self, ds, augmentors, index=(0, 1), coords_index=(), copy=True, catch_exceptions=False)
| 183 | """ |
| 184 | |
| 185 | def __init__(self, ds, augmentors, index=(0, 1), coords_index=(), copy=True, catch_exceptions=False): |
| 186 | """ |
| 187 | Args: |
| 188 | ds (DataFlow): input DataFlow. |
| 189 | augmentors (AugmentorList): a list of :class:`imgaug.ImageAugmentor` instance to be applied in order. |
| 190 | index: tuple of indices of the image components. |
| 191 | coords_index: tuple of indices of the coordinates components. |
| 192 | copy, catch_exceptions: same as in :class:`AugmentImageComponent` |
| 193 | """ |
| 194 | if isinstance(augmentors, AugmentorList): |
| 195 | self.augs = augmentors |
| 196 | else: |
| 197 | self.augs = AugmentorList(augmentors) |
| 198 | self.ds = ds |
| 199 | self._exception_handler = ExceptionHandler(catch_exceptions) |
| 200 | self._copy = copy |
| 201 | self._index = index |
| 202 | self._coords_index = coords_index |
| 203 | |
| 204 | super(AugmentImageComponents, self).__init__(ds, self._aug_mapper) |
| 205 | |
| 206 | def reset_state(self): |
| 207 | self.ds.reset_state() |
nothing calls this directly
no test coverage detected