pytorch 中transforms的使用详解
transforms
按住Ctrl查看transforms的源码可以知道，transforms就是一个python文件，里面定义了很多类，每一个类都是一个工具
在结构那里，可以看到有很多的类
ToTensor
Convert a
PIL Imageornumpy.ndarrayto tensor. This transform does not support torchscript
通过ToTensor来学习transforms如何使用以及为什么使用tensor数据类型
transforms使用
transforms里面每一个类都可以看成是一个模具，我们可以用里面的模具做出一个具体的工具，如何用这个具体的工具来实现具体的功能
比如ToTensor的使用:
from torchvision import transforms from PIL import Image img_path = "data/train/ants_image/0013035.jpg" img = Image.open(img_path) tensor_trans = transforms.ToTensor()#模具（也就是这个类的对象） tensor_img = tensor_trans(img)#实现ToTensor的功能，将一个input（PIL Image）转化成tensor print(tensor_img)
为什么需要tensor数据类型呢？
在使用tensorboard里面常用的add_image时，里面的第二个参数是图片的数据类型，这个数据类型，可以是torch.Tensor, numpy.array, or string/blobname，上一篇博客用的是numpy.array，这里，其实可以直接得到tensor类型后直接用
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter
from PIL import Image
img_path = "data/train/ants_image/0013035.jpg"
img = Image.open(img_path)
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
writer = SummaryWriter("logs")
writer.add_image("Tensor_image", tensor_img)
writer.close()
常见的transforms
内置方法__call__()
可以发现基本上transforms里面的每一个类都有一个内置方法__call__()，这个方法和普通的方法的区别其实就是，普通方法一般是类的对象通过.的方式调用，但是call函数不需要，可以直接用对象加括号的形式调用
一个Person类，内置方法__call__和hello都有一个参数name，然后两个方法都输出name，一个通过person(“”)形式调用，一个通过person.hello(“”)调用
Normalize
Normalize a tensor image with mean and standard deviation.
这个方法进行归一化的时候，传入的参数是有两个列表一个是均值，一个是标准差，每个列表的n表示维度，是根据输入的channel数量决定的，比如我们的图片是rgb那n=3，它能将每个信道的输入进行归一化
根据公式可以知道计算的结果其实就是
代码示例：
from PIL import Image
from torchvision import transforms
img_path = "data/train/ants_image/0013035.jpg"
img = Image.open(img_path)
trans_totensor = transforms.ToTensor()
img_tensor = trans_totensor(img)
print(img_tensor[0][0][0])
trans_norm = transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])
img_norm = trans_norm(img_tensor)
print(img_norm[0][0][0])
writer = SummaryWriter("logs")
writer.add_image("Normalize", img_norm)
writer.close()
输出：
tensor(0.3137)
tensor(-0.3725)
Resize
Resize the input image to the given size
参数：
可以给一个（H,W）这样的参数，改变图片的大小，也可以指定一个int，改变长和宽的比例
代码示例
print(img.size) trans_resize = transforms.Resize((512, 512)) img_resize = trans_resize(img)# 参数和返回值都是 img PIL print(img_resize)
输出结果：
变成了正方形
Compose
Composes several transforms together. This transform does not support torchscript.
可以将第一种类型转化为第二种，参数一的类型做输入，参数二的类型做输出，输入一定要对应，不然就会报错
代码示例
trans_totensor = transforms.ToTensor()
trans_resize_2 = transforms.Resize(512)
# PIL -> tensor
trans_compose = transforms.Compose([trans_resize_2, trans_totensor])
img_resize_2 = trans_compose(img)
writer.add_image("Resize", img_resize_2, 1)
到此这篇关于pytroch中transforms的使用详解的文章就介绍到这了,更多相关pytroch transforms的使用内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
解决Python 命令行执行脚本时,提示导入的包找不到的问题
今天小编就为大家分享一篇解决Python 命令行执行脚本时,提示导入的包找不到的问题，具有很好的参考价值，希望对大家有所帮助，一起跟随小编过来看看吧2019-01-01
这篇文章主要介绍了IntelliJ 中配置 Anaconda过程，本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-06-06
这篇文章主要介绍了Python面向对象程序设计,结合实例形式总结分析了Python面向对象程序设计中比较常见的类定义、实例化、继承、私有变量等相关使用技巧与操作注意事项,需要的朋友可以参考下2019-01-01
这篇文章主要给大家介绍了关于Python导入引用其他文件的函数的相关资料,文中通过代码以及图文介绍的非常详细,对大家学习或者使用Python具有一定的参考借鉴价值,需要的朋友可以参考下2024-01-01
这篇文章主要介绍了如何用C代码给Python写扩展库(Cython)，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-05-05
这篇文章主要介绍了Python爬虫之网络请求，文章基于Python展开对网络请求的相关介绍，需要的小伙伴可以参考一下2022-04-04
这篇文章主要介绍了matplotlib绘图实例演示标记路径，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下2018-01-01
这篇文章主要介绍了Pandas之MultiIndex对象的示例详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-06-06
在本篇文章里小编给大家分享的是关于python语言是否免费的相关知识点，需要的朋友们可以学习下。2020-06-06
这篇文章主要介绍了matplotlib事件处理基础(事件绑定、事件属性)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-02-02
最新评论