【PyTorch】中transforms函数详解

本文介绍在pytorch框架下进行迁移学习时,利用torchvision中Compose函数整合训练图像。torchvision.transforms是图像预处理包,Compose可整合多个步骤。还详细介绍了transforms中的多个函数,如Resize、Normalize等的功能。

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

转载自:        https://www.jianshu.com/p/13e31d619c15

在pytorch框架下进行迁移学习是往往需要将用到训练的图像进行整合,此时就需要用到torchvision中Compose函数,下面对其中的内部参数进行讲解

torchvision.transforms是pytorch中的图像预处理包

一般用Compose把多个步骤整合到一起:

比如说

transforms.Compose([

transforms.CenterCrop(10),

transforms.ToTensor(),

])

这样就把两个步骤整合到一起

接下来介绍transforms中的函数

Resize:把给定的图片resize到given size

Normalize:Normalized an tensor image with mean and standard deviation

ToTensor:convert a PIL image to tensor (H*W*C) in range [0,255] to a torch.Tensor(C*H*W) in the range [0.0,1.0]

ToPILImage: convert a tensor to PIL image

Scale:目前已经不用了,推荐用Resize

CenterCrop:在图片的中间区域进行裁剪

RandomCrop:在一个随机的位置进行裁剪

RandomHorizontalFlip:以0.5的概率水平翻转给定的PIL图像

RandomVerticalFlip:以0.5的概率竖直翻转给定的PIL图像

RandomResizedCrop:将PIL图像裁剪成任意大小和纵横比

Grayscale:将图像转换为灰度图像

RandomGrayscale:将图像以一定的概率转换为灰度图像

FiceCrop:把图像裁剪为四个角和一个中心

TenCrop

Pad:填充

ColorJitter:随机改变图像的亮度对比度和饱和度

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值