八、TorchVision中Transforms的使用

PyTorch torchvision Transforms详解与实战
本文详细介绍了PyTorch torchvision库中的Transforms模块,包括其功能、常用类如ToTensor、ToPILImage、Normalize、Resize和CenterCrop等。通过实例展示了如何使用ToTensor将图片转换为tensor类型,并解释了为何在神经网络中需要tensor数据类型。此外,还演示了如何利用tensorboard显示tensor类型的图片。

一、Transforms结构

1、功能

    对图片进行一些变换

2、引入库

from torchvision import transforms

3、结构

按住Ctrl,点击transforms,出现__init__.py,再按住Ctrl,点击trasforms,出现transforms.py,里面有各种类及各种方法。

transforms其实是一个.py文件,当作一个工具箱的话,里面还有很多分隔区和工具等。Traansforms.py文件中包含许多class,class内又有许多方法。

(1)最常用的是Totensor类,作用是把一个PIL/Numpy.ndarray类型的图片转化为tensor类型

(2)ToPILImage(object)方法:把一个图片转化为PILImage类型

(3)Normalize (object)正则化

(4)Resize(object):进行尺寸的变换

(5)CenterCrop(object):对图片进行中心的裁剪

二、Transform如何使用:以ToTensor()类为例

1、使用步骤

2、代码实战 

用PIL Image打开一个图片,然后转变为tensor类型进行输出。

from torchvision import transforms
from PIL import I
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值