pytorch 把图片数据转化成tensor的操作


Posted in Python onMarch 04, 2021

摘要:

在图像识别当中,一般步骤是先读取图片,然后把图片数据转化成tensor格式,再输送到网络中去。本文将介绍如何把图片转换成tensor。

一、数据转换

把图片转成成torch的tensor数据,一般采用函数:torchvision.transforms。通过一个例子说明,先用opencv读取一张图片,然后在转换;注意一点是:opencv储存图片的格式和torch的储存方式不一样,opencv储存图片格式是(H,W,C),而torch储存的格式是(C,H,W)。

import torchvision.transforms as transforms
import cv2 as cv
img = cv.imread('image/000001.jpg')
print(img.shape)  # numpy数组格式为(H,W,C)
transf = transforms.ToTensor()
img_tensor = transf(img) # tensor数据格式是torch(C,H,W)
print(img_tensor.size())

pytorch 把图片数据转化成tensor的操作

注意:使用torchvision.transforms时要注意一下,其子函数 ToTensor() 是没有参数输入的,以下用法是会报错的

img_tensor = transforms.ToTensor(img)

必须是先定义和赋值转换函数,再调用并输入参数,正确用法:

img = cv.imread('image/000001.jpg')
transf = transforms.ToTensor()
img_tensor = transf(img)

再转换过程中正则化

在使用 transforms.ToTensor() 进行图片数据转换过程中会对图像的像素值进行正则化,即一般读取的图片像素值都是8 bit 的二进制,那么它的十进制的范围为 [0, 255],而正则化会对每个像素值除以255,也就是把像素值正则化成 [0.0, 1.0]的范围。通过例子理解一下:

import torchvision.transforms as transforms
import cv2 as cv
img = cv.imread('image/000001.jpg')
transf = transforms.ToTensor()
img_tensor = transf(img)
print('opencv', img)
print('torch', img_tensor)

pytorch 把图片数据转化成tensor的操作

pytorch 把图片数据转化成tensor的操作

三、自行修改正则化的范围

使用transforms.Compose函数可以自行修改正则化的范围,下面举个例子正则化成 [-1.0, 1.0]

transf2 = transforms.Compose(
  [
    transforms.ToTensor(),
    transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
  ]
)
img_tensor2 = transf2(img)
print(img_tensor2)

pytorch 把图片数据转化成tensor的操作

计算方式就是:

C=(C-mean)/ std

C为每个通道的所有像素值,彩色图片为三通道图像(BGR),所以mean和std是三个数的数组。

使用transforms.ToTensor()时已经正则化成 [0,0, 0,1]了,那么(0.0 - 0.5)/0.5=-1.0,(1.0 - 0.5)/0.5=1.0,所以正则化成 [-1.0, 1.0]

补充:Python: 记录一个关于图片直接转化为pytorch.tensor和numpy.array的不同之处的问题

img = Image.open(img_path).convert("RGB")
img2 = torchvision.transforms.functional.to_tensor(img)
print(img2)
img1 = np.array(img)
print(img1)

输出是这样的:

pytorch 把图片数据转化成tensor的操作

不仅shape不一样,而且值也是不一样的。

解释如下:

tensor = torch.from_numpy(np.asarray(PIL.Image.open(path))).permute(2, 0, 1).float() / 255
tensor = torchvision.transforms.functional.to_tensor(PIL.Image.open(path)) # 两种方法是一样的

PIL.Image.open()得到HWC格式,直接使用numpy 去转换得到(h,w,c)格式,而用to_tensor得到(c,h,w)格式且值已经除了255。

byte()相当于to(torch.uint8),tensor.numpy()是把tensor 转化为numpy.array格式。

在这里需要注意的是PIL和OPENCV的图像读取得到的格式都是HWC格式,一般模型训练使用的是CHW格式, H为Y轴是竖直方向,W为X轴水平方向。

且torchvision.transforms.functional.to_tensor()对所有输入都是有变换操作。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。如有错误或未考虑完全的地方,望不吝赐教。

Python 相关文章推荐
python定时检查某个进程是否已经关闭的方法
May 20 Python
Python利用前序和中序遍历结果重建二叉树的方法
Apr 27 Python
浅谈Python基础之I/O模型
May 11 Python
深入浅出学习python装饰器
Sep 29 Python
Tensorflow使用支持向量机拟合线性回归
Sep 07 Python
python实现从文件中读取数据并绘制成 x y 轴图形的方法
Oct 14 Python
Python 中包/模块的 `import` 操作代码
Apr 22 Python
django-利用session机制实现唯一登录的例子
Mar 16 Python
windows上彻底删除jupyter notebook的实现
Apr 13 Python
容易被忽略的Python内置类型
Sep 03 Python
如何基于Python按行合并两个txt
Nov 03 Python
Pytorch如何切换 cpu和gpu的使用详解
Mar 01 Python
pytorch 计算Parameter和FLOP的操作
Mar 04 #Python
从Pytorch模型pth文件中读取参数成numpy矩阵的操作
Mar 04 #Python
python 如何用urllib与服务端交互(发送和接收数据)
Mar 04 #Python
python 求两个向量的顺时针夹角操作
Mar 04 #Python
python 制作磁力搜索工具
Mar 04 #Python
python抢购软件/插件/脚本附完整源码
Mar 04 #Python
Python 求向量的余弦值操作
Mar 04 #Python
You might like
php mysql数据库操作分页类
2008/06/04 PHP
PHP中mb_convert_encoding与iconv函数的深入解析
2013/06/21 PHP
输入值/表单提交参数过滤有效防止sql注入的方法
2013/12/25 PHP
php检测用户是否用手机(Mobile)访问网站的类
2014/01/09 PHP
PHP中的事务使用实例
2015/05/26 PHP
php使用Jpgraph绘制饼状图的方法
2015/06/10 PHP
用js判断浏览器是否是IE的比较好的办法
2007/05/08 Javascript
JQuery入门——移除绑定事件unbind方法概述及应用
2013/02/05 Javascript
30分钟就入门的正则表达式基础教程
2013/02/25 Javascript
jQuery控制输入框只能输入数值的小例子
2013/03/20 Javascript
基于Bootstrap使用jQuery实现简单可编辑表格
2016/05/04 Javascript
js实现的简练高效拖拽功能示例
2016/12/21 Javascript
关于Node.js中Buffer的一些你可能不知道的用法
2017/03/28 Javascript
EasyUI的DataGrid每行数据添加操作按钮的实现代码
2017/08/22 Javascript
JS实现验证码倒计时的注册页面
2018/01/02 Javascript
p5.js绘制旋转的正方形
2019/10/23 Javascript
使用PreloadJS加载图片资源的基础方法详解
2020/02/03 Javascript
iview实现图片上传功能
2020/06/29 Javascript
Vue-cli 移动端布局和动画使用详解
2020/08/10 Javascript
python 3.6.4 安装配置方法图文教程
2018/09/18 Python
python print出共轭复数的方法详解
2019/06/25 Python
pytorch 固定部分参数训练的方法
2019/08/17 Python
Python+Selenium+phantomjs实现网页模拟登录和截图功能(windows环境)
2019/12/11 Python
Django 解决model 反向引用中的related_name问题
2020/05/19 Python
Matplotlib.pyplot 三维绘图的实现示例
2020/07/28 Python
Python新建项目自动添加介绍和utf-8编码的方法
2020/12/26 Python
优秀教师事迹简介
2014/02/02 职场文书
护理专业自荐书
2014/06/04 职场文书
文艺晚会策划方案
2014/06/11 职场文书
税务干部群众路线教育实践活动对照检查材料
2014/09/20 职场文书
运动会稿件100字
2014/09/24 职场文书
小学教师岗位职责
2015/04/02 职场文书
原告离婚代理词
2015/05/23 职场文书
培训班开班主持词
2015/07/02 职场文书
《用字母表示数》教学反思
2016/02/17 职场文书
JavaScript实现复选框全选功能
2021/04/11 Javascript