def Contrast(x, alpha):
# alpha (1.0, 2.0)
return kornia.adjust_contrast(x, alpha)
def _transform(self, img: torch.Tensor,
param: torch.Tensor) -> torch.Tensor:
return kornia.adjust_contrast(img, param)
plt.axis('off')
plt.show()
#############################
# Show original
imshow(x_rgb)
#############################
# Adjust Brightness
x_brightness: torch.Tensor = kornia.adjust_brightness(x_rgb, 0.6)
imshow(x_brightness)
#############################
# Adjust Contrast
x_contrast: torch.Tensor = kornia.adjust_contrast(x_rgb, 0.2)
imshow(x_contrast)
#############################
# Adjust Gamma
x_gamma: torch.Tensor = kornia.adjust_gamma(x_rgb, gamma=3., gain=1.5)
imshow(x_gamma)
#############################
# Adjust Saturation
x_saturated: torch.Tensor = kornia.adjust_saturation(x_rgb, 0.2)
imshow(x_saturated)
#############################
# Adjust Hue
x_hue: torch.Tensor = kornia.adjust_hue(x_rgb, 0.5)
def Contrast(x, v):
return kornia.adjust_contrast(x, v)