图像增强-深度学习-批量对比度增强-数据增广

数据集增强/数据增广

下面展示一些 内联代码片。

from PIL import Image
from PIL import ImageEnhance
import imageio
import os
import time
import glob
from scipy import ndimage
import cv2 as cv

#------------------单张图片对比度增强-------------------
# img = Image.open('G:\数据集出图\对比度增强\数据集原图5.png')
# img.show()

# 对比度增强
# enh_con = ImageEnhance.Contrast(img)
# contrast = 2    #调整contrast改变对比度参数
# img_contrasted = enh_con.enhance(contrast)
# img_contrasted.show()
# img_contrasted.save("G:\硕士毕业文件夹\BaiduNetdiskWorkspace\论文出图\数据集出图\对比度增强\数据集原图5_duibidu.png")

# -------------------批量对比度增强------------------------

input_dir = 'G:\数据集出图\对比度增强\\'
out_dir = 'G:\数据集出图\对比度增强\\'
a = os.listdir(input_dir)

for i in a:
    print(i)
    I = Image.open(input_dir + i)
    enh_con = ImageEnhance.Contrast(I)
    contrast = 2
    #调整contrast改变对比度参数
    img_contrasted = enh_con.enhance(contrast)
    # L = I.convert('L')
    img_contrasted.save(out_dir + i)