目录
0.7版本的torchvision.transforms集成了PIL ImageEnhance图像增强模块,对该模块函数进行整理/测试。通过开发工具找到ImageEnhance实现源码,一共由5个类:
_Enhance是一个私有类,作为基类使用,提供通用模块。可以看出使用此类方法将图片进行数据增强,具体增强图片的形式是如下几种:Color、Contrast、Brightness、Sharpness,依次进行说明和效果测试。
1、Color-色彩饱和度调节
色彩饱和度:彩度除以明度,指色彩的鲜艳程度,也称色彩的纯度。值较小时,图像颜色降低,值较大时图像颜色更鲜艳。
from PIL import Image, ImageEnhance
image = Image.open("dog.jpg")
# print(image.size) # (w, h)
enh_bri = ImageEnhance.Color(image)
factor = 0.5
enhance_image = enh_bri.enhance(factor)
enhance_image.save("enhance_image.jpg")
2、Contrast-对比度调节
对比度:白色画面(最亮时)下的亮度除以黑色画面(最暗时)下的亮度。值较小时。由雾的感觉,值较大时更清晰,能简单去雾。
from PIL import Image, ImageEnhance
image = Image.open("dog.jpg")
# print(image.size) # (w, h)
enh_bri = ImageEnhance.Contrast(image)
factor = 0.5
enhance_image = enh_bri.enhance(factor)
enhance_image.save("enhance_image.jpg")
3、Brightness-对比度调节
白色画面(最亮时)下的亮度除以黑色画面(最暗时)下的亮度。factor为浮点型对比度调节参数,为0时返回一张黑图,1时返回原图。建议参数值范围:0.5~1.5,参数从小至大,对应的图像亮度由小变大,返回原图多少倍的亮度图。
from PIL import Image, ImageEnhance
image = Image.open("dog.jpg")
# print(image.size) # (w, h)
enh_bri = ImageEnhance.Brightness(image)
factor = 0.5
enhance_image = enh_bri.enhance(factor)
enhance_image.save("enhance_image.jpg")
4、Sharpness-锐度调节
锐度:反映图像平面清晰度和图像边缘锐利程度的一个指标。值较小时,图像偏模糊,值较大时边缘更加锐利。
from PIL import Image, ImageEnhance
image = Image.open("dog.jpg")
# print(image.size) # (w, h)
enh_bri = ImageEnhance.Sharpness(image)
factor = 0.5
enhance_image = enh_bri.enhance(factor)
enhance_image.save("enhance_image.jpg")
参考:
1、PIL ImageEnhance 模块文档