图像数据增强代码
时间: 2023-08-06 16:20:05 浏览: 46
以下是使用 Keras 的图像数据增强代码示例:
```python
from keras.preprocessing.image import ImageDataGenerator
# 定义数据增强器
datagen = ImageDataGenerator(
rotation_range=30, # 随机旋转范围
width_shift_range=0.2, # 随机水平平移范围
height_shift_range=0.2, # 随机垂直平移范围
shear_range=0.2, # 随机错切变换范围
zoom_range=0.2, # 随机缩放范围
horizontal_flip=True, # 随机水平翻转
fill_mode='nearest') # 填充模式
# 使用数据增强器生成增强后的图像数据
img = load_img('example.jpg')
x = img_to_array(img)
x = x.reshape((1,) + x.shape)
i = 0
for batch in datagen.flow(x, batch_size=1):
plt.figure(i)
imgplot = plt.imshow(array_to_img(batch[0]))
i += 1
if i % 4 == 0:
break
```
此代码使用了 Keras 的 `ImageDataGenerator` 类来定义数据增强器,并使用 `flow` 方法对图像数据进行增强。可以通过参数来调整数据增强的方式,如随机旋转、平移、缩放、翻转等,从而扩充数据集以提高深度学习模型的泛化能力。