pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换


Posted in Python onJanuary 13, 2020

1, 创建pytorch 的Tensor张量:

torch.rand((3,224,224)) #创建随机值的三维张量,大小为(3,224,224)
 
torch.Tensor([3,2]) #创建张量,[3,2]

2, cpu上的tensor和GPU即pytorch创建的tensor的相互转化

b = a.cpu() # GPU → CPU
 
a = b.cuda() #CPU → GPU

3, tensor和numpy的转化

b = a.numpy() # tensor转化为 numpy数组
 
a = b.from_numpy() # numpy数组转化为tensor

4, torch的GPU tensor保存为图片

import scipy.misc
 
scipy.misc.imsave(‘pic_name',img) #img为二维张量,比如(224,224),保存为黑白图

5, 堆叠矩阵,形成彩色图片

img = np.stack((ia,b,c),dim) #堆叠矩阵a,b,c 可用于三通道图像的保存 dim表示要增加的维度,
#比如a,b,c均为(224,224)大小的矩阵,那么令dim=-1,则 img的维度为(224,224,3)

6, 从numpy数组保存图片

from PIL import Image
 
im = Image.fromarray(A)
 
im.save("your_file.jpeg")

7, 读取图片为矩阵:

import matplotlib.image
im = matplotlib.image.imread('0_0.jpg')

8, 保存矩阵为图片:

import numpy as np
import scipy.misc
 
x = np.random.random((600,800,3))
scipy.misc.imsave('meelo.jpg', x)

以上这篇pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python处理圆角图片、圆形图片的例子
Apr 25 Python
python 链接和操作 memcache方法
Mar 04 Python
Python计算两个日期相差天数的方法示例
May 23 Python
Python实现的本地文件搜索功能示例【测试可用】
May 30 Python
使用Python读取二进制文件的实例讲解
Jul 09 Python
Python不使用int()函数把字符串转换为数字的方法
Jul 09 Python
对pandas将dataframe中某列按照条件赋值的实例讲解
Nov 29 Python
python求最大值最小值方法总结
Jun 25 Python
python3中利用filter函数输出小于某个数的所有回文数实例
Nov 24 Python
python opencv根据颜色进行目标检测的方法示例
Jan 15 Python
python函数调用,循环,列表复制实例
May 03 Python
python re模块和正则表达式
Mar 24 Python
python、PyTorch图像读取与numpy转换实例
Jan 13 #Python
pytorch 彩色图像转灰度图像实例
Jan 13 #Python
Ranorex通过Python将报告发送到邮箱的方法
Jan 12 #Python
python opencv实现信用卡的数字识别
Jan 12 #Python
Python 实现递归法解决迷宫问题的示例代码
Jan 12 #Python
Python3.x+迅雷x 自动下载高分电影的实现方法
Jan 12 #Python
tensorflow的计算图总结
Jan 12 #Python
You might like
php实现以只读方式打开文件的方法
2015/03/16 PHP
win平台安装配置Nginx+php+mysql 环境
2016/01/12 PHP
PHP实现原比例生成缩略图的方法
2016/02/03 PHP
基于thinkPHP框架实现留言板的方法
2016/10/17 PHP
在 Laravel 中 “规范” 的开发短信验证码发送功能
2017/10/26 PHP
JQuery的read函数与js的onload不同方式实现
2013/03/18 Javascript
JS组件Form表单验证神器BootstrapValidator
2016/01/26 Javascript
NODE.JS跨域问题的完美解决方案
2016/10/20 Javascript
JS中事件冒泡和事件捕获介绍
2016/12/13 Javascript
EasyUI修改DateBox和DateTimeBox的默认日期格式示例
2017/01/18 Javascript
原生js实现旋转木马轮播图效果
2017/02/27 Javascript
js和jquery中获取非行间样式
2017/05/05 jQuery
详解vue-cli开发环境跨域问题解决方案
2017/06/06 Javascript
Angular.js前台传list数组由后台spring MVC接收数组示例代码
2017/07/31 Javascript
node中Express 动态设置端口的方法
2017/08/04 Javascript
vue脚手架中配置Sass的方法
2018/01/04 Javascript
layui表格checkbox选择全选样式及功能的实例
2018/03/07 Javascript
vue插件mescroll.js实现移动端上拉加载和下拉刷新
2019/03/07 Javascript
产制造追溯系统之通过微信小程序实现移动端报表平台
2019/06/03 Javascript
js实现橱窗展示效果
2020/01/11 Javascript
vue项目中使用bpmn为节点添加颜色的方法
2020/04/30 Javascript
echarts实现晶体球面投影的实例教程
2020/10/10 Javascript
Python图形绘制操作之正弦曲线实现方法分析
2017/12/25 Python
解决Django一个表单对应多个按钮的问题
2019/07/18 Python
高街生活方式全球在线商店:AZBRO
2017/08/26 全球购物
英国老牌潮鞋店:Offspring
2019/08/19 全球购物
大三在校生电子商务求职信
2013/10/29 职场文书
金融专业应届生求职信
2013/11/02 职场文书
办公室主任职责范文
2013/11/08 职场文书
应届行政管理专业个人自我评价
2013/12/28 职场文书
计算机专业自荐信范文
2015/03/26 职场文书
2015年汽车销售员工作总结
2015/07/24 职场文书
缅怀先烈主题班会
2015/08/14 职场文书
《金钱的魔力》教学反思
2016/02/20 职场文书
python3.9之你应该知道的新特性详解
2021/04/29 Python
GO中sync包自由控制并发示例详解
2022/08/05 Golang