使用Keras进行数据增强可生成几乎为白色的图像

from keras.preprocessing.image import ImageDataGenerator
import numpy as np

# 创建ImageDataGenerator对象
datagen = ImageDataGenerator(brightness=2.0)

# 加载原始图像
image = np.ones((100, 100, 3))  # 创建一个全白的图像

# 将图像转换为4D张量，以符合Keras的输入要求
image = np.expand_dims(image, axis=0)

# 生成增强后的图像
augmented_images = datagen.flow(image, batch_size=1)

# 获取增强后的图像
augmented_image = augmented_images.next()

# 将增强后的图像转换为3D张量
augmented_image = augmented_image[0].astype(np.uint8)

# 显示增强后的图像
plt.imshow(augmented_image)
plt.axis('off')
plt.show()