批量隨機gamma

from PIL import Image as PImage
from os import listdir
import numpy as np
import random
import cv2

#Load images in filefolder
def loadImages(path):
    imagesList = listdir(path)
    loadedImages = []
    for image in imagesList:
        img = PImage.open(path + image)
        loadedImages.append(img)
    return loadedImages

path = "./trainSet/"
imgs = loadImages(path)


#gamma change function
def darkened(img):
    gamma = np.random.uniform(2,5)
    tmp=gamma
    return img ** gamma

i=0
for img in imgs:
    #regular to (0,1)
	img_array_reg = np.array(img)/255.0
	dark_array = (darkened(img_array_reg)*255.0).astype(np.uint8)
	dark_img = cv2.cvtColor(dark_array, cv2.COLOR_BGR2RGB)
	i=i+1
	cv2.imwrite('%s.png'%i, dark_img)