PyTorch中的transforms图像增强方法

本文详细介绍了PyTorch中的transforms模块用于数据增强的各种方法,包括中心裁剪、随机裁剪、随机旋转、填充图像边缘、调整色彩等,旨在提升模型的泛化能力。
摘要由CSDN通过智能技术生成

#博学谷IT学习技术支持#


一、数据增强(Data Augmentation)

数据增强又称为数据增广,数据扩增,它是对训练集进行变换,使训练集更丰富,从而让模型更具泛化能力。

1.1 中心裁剪

transforms.CenterCrop
功能:从中心裁剪图片
size:所需裁剪图片尺寸

1.2 随机裁剪

transforms.RandomCrop
功能:从图片中随机裁剪初尺寸为size的图片
size:所需裁剪图片尺寸
padding:设置填充大小,当为 int时, 例如padding=a时,图片上下左右均填充a个像素;当为tupple时,(a, b),则左右填充a个像素,上下填充b个像素;(a, b, c, d), 则左填充a个像素,下填充b个像素,右填充c个像素, 上填充d个像素。

pad_if_needed=False(默认), 若图像小于设定的size,则填充。当超出尺寸时,必须打开,不然会报错。
padding_mode:填充模式,有4种,
1、constant:像素由fill设定。
2、edge:像素由图像边沿像素决定。
3、reflect:像素填充,最后一个像素不镜像, 例如: [1, 2, 3, 4] --> [3, 2, 1, 2, 3, 4, 3, 2]
4、symmetric:镜像填充,最后一个像素镜像, 例如:[1, 2, 3, 4] --> [2, 1, 1, 2, 3, 4, 4, 3]
file:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值