深度学习笔记 —— 数据增广

最新推荐文章于 2025-05-07 09:26:51 发布

Whisper_yl

最新推荐文章于 2025-05-07 09:26:51 发布

阅读量5.4k

点赞数 3

分类专栏： # 深度学习文章标签：深度学习

本文链接：https://blog.csdn.net/LightInDarkness/article/details/124251507

版权

深度学习专栏收录该内容

23 篇文章

订阅专栏

本文探讨了卷积神经网络中数据增强的重要性，通过左右翻转、上下翻转、随机裁剪、调整亮度和色调等方法增加训练集的多样性，以提高模型的泛化能力。并展示了使用Python和PyTorch库实现这些数据增强技术的代码示例。

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

固定形状？卷积神经网络通常是一个固定的输入。

作用好不好是不一定的。很多时候觉得它有用是因为觉得测试集里会有图片跟这个效果是差不多的。

所以如何选择数据增强？可以从后往前推，想一下部署和测试集里面的图片与训练集的图片会有什么变化，从而确定需要什么样的数据增强方式。

多样性更强，使得泛化性更好。增广的目的是想让训练集更像测试集，如果完全一样就更好了！:)

import torch
import torchvision
from torch import nn
from d2l import torch as d2l
import matplotlib.pyplot as plt


d2l.set_figsize()
img = d2l.Image.open('./img/cat1.jpg')
d2l.plt.imshow(img);


# aug:图片增广的方式
def apply(img, aug, num_rows=2, num_cols=4, scale=1.5):
    Y = [aug(img) for _ in range(num_rows * num_cols)]
    d2l.show_images(Y, num_rows, num_cols, scale=scale)
    plt.show()


# 左右翻转图像
# apply(img, torchvision.transforms.RandomHorizontalFlip())

# 上下翻转图像
# apply(img, torchvision.transforms.RandomVerticalFlip())

# 随机裁剪
shape_aug = torchvision.transforms.RandomResizedCrop(
    (200, 200), scale=(0.1, 1), ratio=(0.5, 2))

# 随机更改图像的亮度
'''
apply(img, torchvision.transforms.ColorJitter(
    brightness=0.5, contrast=0, saturation=0, hue=0))
'''

# 随机更改图像的色调
'''
apply(img, torchvision.transforms.ColorJitter(
    brightness=0, contrast=0, saturation=0, hue=0.5))
'''

# 随机更改图像的亮度、对比度、饱和度和色调
color_aug = torchvision.transforms.ColorJitter(
    brightness=0.5, contrast=0.5, saturation=0.5, hue=0.5)

# 结合多种图像增广方法
augs = torchvision.transforms.Compose([
    torchvision.transforms.RandomHorizontalFlip(), color_aug, shape_aug])
apply(img, augs)