跟着小土堆学习pytorch(三)——常见transform

文章目录

一、Normalize

把图片进行,归一化
output[channel] = (input[channel] - mean[channel]) / std[channel]

这是Normalize,所使用的公式

python 复制代码
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.add_image("Normalize",img_norm)

因为RGB图片是三通道的,所以需要用中括号,在均值、方差处,输入三个参数

二、Resize

如果输入,两个参数,那么直接规定为长和宽,如果只有一个数字,那么就是等比例缩小

python 复制代码
print(img.size)
trans_resize transforms.Resize((512,512))
img PIL -resize -img_resize PIL
img_resize trans_resize(img)
img_resize PIL -totensor -img_resize tensor
img_resize trans_totensor(img_resize)
writer.add_image("Resize",img_resize,0)
print(img_resize)

三、compose

Compose()中的参数需要是一个列表

Python中,列表的表示形式为[数据1,数据2,]

在Compose中,数据需要是transforms类型,所以得到,Compose([transforms参数1,transforms参数2,.)

python 复制代码
# Compose-resize-2
trans_resize_2 = transforms.Resize(512)
# PIL -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)

trans_compose transforms.Compose([trans_resize_2,trans_totensor])

着重讲一下这一行

首先明确三点

  1. compose的作用就是把transform组合一起,按照前面处理完,再传完下一个
  2. Resize对象输入PIL,输出也是PIL
  3. totensor对象输入时PIL,输出时tensor
    注意:trans_resize_2,trans_totensor,这是已经把对象实例化了
    所以这一行的代码,是先放入trans_resize_2处理,再放入trans_totensor

四、RandomCrop

嗯,随机裁剪

sequence---序列---()

list---列表----[]

tuple---元组

RandomCrop可以输入的是序列

compose输入的是列表

五、transform

听到现在为止,transform就是对图片进行一个预处理的过程

关注几个点

1.输入和输出的数据类型

2.多看官方文档

3.关注方法需要什么参数

不知道返回值或者返回值的类型

print()

print(type())

最后转化为tensor,用torchvision看一下效果

相关推荐
-一杯为品-14 分钟前
【51单片机】程序实验5&6.独立按键-矩阵按键
c语言·笔记·学习·51单片机·硬件工程
傻啦嘿哟43 分钟前
如何使用 Python 开发一个简单的文本数据转换为 Excel 工具
开发语言·python·excel
B站计算机毕业设计超人1 小时前
计算机毕业设计SparkStreaming+Kafka旅游推荐系统 旅游景点客流量预测 旅游可视化 旅游大数据 Hive数据仓库 机器学习 深度学习
大数据·数据仓库·hadoop·python·kafka·课程设计·数据可视化
风尚云网1 小时前
风尚云网前端学习:一个简易前端新手友好的HTML5页面布局与样式设计
前端·css·学习·html·html5·风尚云网
IT古董1 小时前
【人工智能】Python在机器学习与人工智能中的应用
开发语言·人工智能·python·机器学习
湫ccc2 小时前
《Python基础》之pip换国内镜像源
开发语言·python·pip
hakesashou2 小时前
Python中常用的函数介绍
java·网络·python
菜鸟的人工智能之路2 小时前
极坐标气泡图:医学数据分析的可视化新视角
python·数据分析·健康医疗
菜鸟学Python2 小时前
Python 数据分析核心库大全!
开发语言·python·数据挖掘·数据分析
小白不太白9502 小时前
设计模式之 责任链模式
python·设计模式·责任链模式