| 167 | |
| 168 | |
| 169 | class ComposeX: |
| 170 | def __init__(self, augmentations): |
| 171 | self.augmentations = augmentations |
| 172 | |
| 173 | def __call__(self, data): |
| 174 | # img, mask = Image.fromarray(img, mode='RGB'), Image.fromarray(mask, mode='L') |
| 175 | |
| 176 | for a in self.augmentations: |
| 177 | data = a(data) |
| 178 | if data is None: |
| 179 | return None |
| 180 | return data |
| 181 | |
| 182 | |
| 183 | class LoadImageFromFile: |