(self, img, hue)
| 31 | return self._rand_range(*self.range) |
| 32 | |
| 33 | def _augment(self, img, hue): |
| 34 | m = cv2.COLOR_BGR2HSV if not self.rgb else cv2.COLOR_RGB2HSV |
| 35 | hsv = cv2.cvtColor(img, m) |
| 36 | # https://docs.opencv.org/3.2.0/de/d25/imgproc_color_conversions.html#color_convert_rgb_hsv |
| 37 | if hsv.dtype.itemsize == 1: |
| 38 | # OpenCV uses 0-179 for 8-bit images |
| 39 | hsv[..., 0] = (hsv[..., 0] + hue) % 180 |
| 40 | else: |
| 41 | # OpenCV uses 0-360 for floating point images |
| 42 | hsv[..., 0] = (hsv[..., 0] + 2 * hue) % 360 |
| 43 | m = cv2.COLOR_HSV2BGR if not self.rgb else cv2.COLOR_HSV2RGB |
| 44 | img = cv2.cvtColor(hsv, m) |
| 45 | return img |
| 46 | |
| 47 | |
| 48 | class Brightness(PhotometricAugmentor): |
no outgoing calls
no test coverage detected