在使用PIL做数据增强的时候,需要对图片进行格式转换,Image.fromarray,这条指令无法处理float型数据,如下的报错
所以需要类型转换为uint8。
import numpy as np
from PIL.ImageEnhance import *
from PIL import Image
gamma = np.random.uniform(0.7, 1.5)
img_1=np.array(range(300)).reshape(10,10,3).astype(np.float32)
a=img_1.astype(np.uint8)
b = Contrast(Image.fromarray(a)).enhance(gamma)
c=np.array(b,np.uint8)