使用数据增强技术对已有样本进行扩充

本文介绍了在样本量有限的情况下，如何利用Python的PIL库对图像进行数据增强，包括水平翻转、垂直翻转和三种不同角度的旋转，以防止过拟合。提供的代码示例展示了如何实现这些操作并保存处理后的图像。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

当数据集中的样本量较少时，直接将其输入网络进行训练可能会导致过拟合，所以需要对已有样本通过使用数据增强技术进行扩充。扩充时主要有翻转和旋转两种操作，其中翻转包含水平翻转、上下翻转两种，旋转包含逆时针旋转 $90∘90^{\circ}$ 、 $180∘180^{\circ}$ 、 $270∘270^{\circ}$ 三种。

具体代码如下所示。

# -*- coding: utf-8 -*-

from PIL import Image
import os

file_dir = 'E:/Remote Sensing/' # 原始图片路径
save_dir = 'E:/Remote Sensing/'  # 保存路径
img_name = 'airplane.jpg'
img_path = file_dir + img_name  # 批量读取图片
im = Image.open(img_path)
LEFT_RIGHT = im.transpose(Image.FLIP_LEFT_RIGHT)  # 左右翻转
TOP_BOTTOM = im.transpose(Image.FLIP_TOP_BOTTOM)  # 上下翻转
rotated_90 = im.transpose(Image.ROTATE_90)  # 逆时针旋转90度
rotated_180 = im.transpose(Image.ROTATE_180)  # 逆时针旋转180度
rotated_270 = im.transpose(Image.ROTATE_270)  # 逆时针旋转270度
LEFT_RIGHT.save(save_dir + 'LEFT_RIGHT_'+img_name)  # 保存左右翻转后的图片
TOP_BOTTOM.save(save_dir + 'TOP_BOTTOM_'+img_name)  # 保存上下翻转后的图片
rotated_90.save(save_dir + 'rotated_90_'+img_name)  # 保存逆时针旋转90度后的图片
rotated_180.save(save_dir + 'rotated_180_'+img_name)  # 保存逆时针旋转180度后的图片
rotated_270.save(save_dir + 'rotated_270_'+img_name)  # 保存逆时针旋转270度后的图片