批量gama增强

import imageio
import numpy as np
import matplotlib.pyplot as plt
import pylab
import os

s = 'C:/Users/liupeng/Desktop/A/BB'
for filename in os.listdir(s):  # listdir的参数是文件夹的路径
    path =s + '/' + filename
    # print(filename)  # 1 (1).png
    # dark_image = imageio.imread('C:/Users/liupeng/Desktop/A/AA/1 (4).png')
    dark_image = imageio.imread(path)
    gama_image = (dark_image.astype(np.float32) / 255) ** 0.5  # 0.7
    plt.figure(figsize=(10, 10))
    plt.subplot(1, 2, 1)
    plt.imshow(dark_image / 255)
    plt.subplot(1, 2, 2)
    # plt.imshow(gama_image)
    # pylab.show()
    imageio.imsave(path, gama_image*255)