def get_transform(self, img): w = np.random.uniform(self.intensity_min, self.intensity_max) return BlendTransform(src_image=0, src_weight=1 - w, dst_weight=w)
def get_transform(self, img): assert img.shape[-1] == 3, "Saturation only works on RGB images" w = np.random.uniform(self.intensity_min, self.intensity_max) grayscale = img.dot([0.299, 0.587, 0.114])[:, :, np.newaxis] return BlendTransform(src_image=grayscale, src_weight=1 - w, dst_weight=w)
def get_transform(self, img): assert img.shape[-1] == 3, "Saturation only works on RGB images" weights = np.random.normal(scale=self.scale, size=3) return BlendTransform( src_image=self.eigen_vecs.dot(weights * self.eigen_vals), src_weight=1.0, dst_weight=1.0 )
def get_transform(self, img): w = np.random.uniform(self.intensity_min, self.intensity_max) pure_gray = np.zeros_like(img) pure_gray[:, :, 0] = 0.5 return BlendTransform(src_image=pure_gray, src_weight=1 - w, dst_weight=w)