数据增强python代码实现（对比度、亮度、色度和锐度）

from PIL import Image

from PIL import ImageEnhance

import numpy as np

img = Image.open(r"C:\\Users\\Administrator\\Desktop\\2.JPG")

# 提高亮度

enh_bri = ImageEnhance.Brightness(img)

# brightness = np.random.uniform(0.5,2,)

brightness = 1.5

print(brightness)

image_brightened = enh_bri.enhance(brightness)

image_brightened.show("")

 

# 提高色度

enh_col = ImageEnhance.Color(img)

color = np.random.uniform(1.2, 1.5,)

image_colored = enh_col.enhance(color)

image_colored.show("")

# 饱和度

# 对比度

enh_con = ImageEnhance.Contrast(img)

constrast = np.random.uniform(1.2, 2.2)

image_constrast = enh_con.enhance(constrast)

image_constrast.show()

# 灰度系数

# 锐度

enh_sha = ImageEnhance.Sharpness(img)

sharpness = np.random.uniform(1.5, 4)

# print(sharpness)

image_sharped = enh_sha.enhance(sharpness)

image_sharped.show()