(self, img, _)
| 320 | self._init(locals()) |
| 321 | |
| 322 | def _augment(self, img, _): |
| 323 | img = img.astype('float32') |
| 324 | if self.all_channel: |
| 325 | minimum = np.min(img) |
| 326 | maximum = np.max(img) |
| 327 | else: |
| 328 | minimum = np.min(img, axis=(0, 1), keepdims=True) |
| 329 | maximum = np.max(img, axis=(0, 1), keepdims=True) |
| 330 | img = (self.max - self.min) * (img - minimum) / (maximum - minimum) + self.min |
| 331 | return img |