扩大数据集

最新推荐文章于 2023-11-03 14:49:43 发布

很菜又爱学

最新推荐文章于 2023-11-03 14:49:43 发布

阅读量226

点赞数

分类专栏： python 文章标签：计算机视觉图像识别

本文链接：https://blog.csdn.net/kundezui/article/details/116804601

版权

本文介绍了一种通过图像水平翻转、垂直翻转及不同角度旋转来扩充数据集的方法，以应对图像处理中数据不足的问题，旨在提高模型的泛化能力和降低过拟合风险。

摘要由CSDN通过智能技术生成

在图像处理中，有时候数据集不够，可以考虑如下方式来增广数据集，降低过拟合，提高网络的泛化能力

图像增强

原图
水平翻转（左右翻转）
垂直翻转（上下翻转）
逆时针旋转45°
逆时针旋转90°
逆时针旋转180°
逆时针旋转270°
综合代码
遍历文件夹，并扩充数据集

原图

在这里插入图片描述

水平翻转（左右翻转）

在这里插入图片描述

垂直翻转（上下翻转）

在这里插入图片描述

逆时针旋转45°

在这里插入图片描述

逆时针旋转90°

在这里插入图片描述

逆时针旋转180°

在这里插入图片描述

逆时针旋转270°

在这里插入图片描述

综合代码


import matplotlib.pyplot as plt
from PIL import Image

path = r"C:\Users\qiu\Desktop\hw2\garbage_classify\train_data\img_6877.jpg"

#读取图像
img = Image.open(path)
print(img.size) #原图是800x575像素  ，输入却是575,800
plt.imshow(img),plt.show()

#图像调整为100x100
# imgs = img.resize((72,72),Image.BILINEAR)
# print(imgs.size