(self, img, v)
| 294 | return (self.rng.randn(3) * self.std).astype("float32") |
| 295 | |
| 296 | def _augment(self, img, v): |
| 297 | old_dtype = img.dtype |
| 298 | v = v * self.eigval |
| 299 | v = v.reshape((3, 1)) |
| 300 | inc = np.dot(self.eigvec, v).reshape((3,)) |
| 301 | img = np.add(img, inc) |
| 302 | if self.clip or old_dtype == np.uint8: |
| 303 | img = np.clip(img, 0, 255) |
| 304 | return img.astype(old_dtype) |
| 305 | |
| 306 | |
| 307 | class MinMaxNormalize(PhotometricAugmentor): |