(self, img)
| 237 | a.reset_state() |
| 238 | |
| 239 | def get_transform(self, img): |
| 240 | check_dtype(img) |
| 241 | assert img.ndim in [2, 3], img.ndim |
| 242 | |
| 243 | from .transform import LazyTransform, TransformList |
| 244 | # The next augmentor requires the previous one to finish. |
| 245 | # So we have to use LazyTransform |
| 246 | tfms = [] |
| 247 | for idx, a in enumerate(self.augmentors): |
| 248 | if idx == 0: |
| 249 | t = a.get_transform(img) |
| 250 | else: |
| 251 | t = LazyTransform(a.get_transform) |
| 252 | |
| 253 | if isinstance(t, TransformList): |
| 254 | tfms.extend(t.tfms) |
| 255 | else: |
| 256 | tfms.append(t) |
| 257 | return TransformList(tfms) |
| 258 | |
| 259 | def __str__(self): |
| 260 | repr_each_aug = ",\n".join([" " + repr(x) for x in self.augmentors]) |
nothing calls this directly
no test coverage detected