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版微信红包分配算法
May 04 Python
python使用reportlab实现图片转换成pdf的方法
May 22 Python
Python读大数据txt
Mar 28 Python
Python3非对称加密算法RSA实例详解
Dec 06 Python
对python中字典keys,values,items的使用详解
Feb 03 Python
python单线程文件传输的实例(C/S)
Feb 13 Python
使用matplotlib中scatter方法画散点图
Mar 19 Python
Python发展史及网络爬虫
Jun 19 Python
为什么从Python 3.6开始字典有序并效率更高
Jul 15 Python
python从内存地址上加载python对象过程详解
Jan 08 Python
Python 动态变量名定义与调用方法
Feb 09 Python
pycharm2020.2 配置使用的方法详解
Sep 16 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字符串处理的10个简单方法
2010/06/30 PHP
php计算到指定日期还有多少天的方法
2015/04/14 PHP
PHP模型Model类封装数据库操作示例
2019/03/14 PHP
ThinkPHP框架整合微信支付之刷卡模式图文详解
2019/04/10 PHP
JS多物体 任意值 链式 缓冲运动
2012/08/10 Javascript
用JavaScript实现动画效果的方法
2013/07/20 Javascript
js从Cookies里面取值的简单实现
2014/06/30 Javascript
使用javascript实现雪花飘落的效果
2015/01/13 Javascript
通过Jquery.cookie.js实现展示浏览网页的历史记录超管用
2015/10/23 Javascript
JavaScript+Java实现HTML页面转为PDF文件保存的方法
2016/05/30 Javascript
详解javascript事件绑定使用方法
2016/10/20 Javascript
jQuery实现的无缝广告图片左右滚动功能详解
2016/12/24 Javascript
利用Vconsole和Fillder进行移动端抓包调试方法
2019/03/05 Javascript
详解vue中this.$emit()的返回值是什么
2019/04/07 Javascript
vue抽出组件并传值实例
2020/07/31 Javascript
vue中的.$mount('#app')手动挂载操作
2020/09/02 Javascript
[00:52]黑暗之门更新 新英雄孽主驾临DOTA2
2016/08/24 DOTA
[00:12]2018DOTA2亚洲邀请赛 Somnus丶M出阵单挑
2018/04/06 DOTA
详解Python中DOM方法的动态性
2015/04/11 Python
Python itertools模块详解
2015/05/09 Python
Django 连接sql server数据库的方法
2018/06/30 Python
pandas DataFrame 删除重复的行的实现方法
2019/01/29 Python
解决django前后端分离csrf验证的问题
2019/02/03 Python
在Python 字典中一键对应多个值的实例
2019/02/03 Python
Python中使用threading.Event协调线程的运行详解
2020/05/02 Python
Python 合并拼接字符串的方法
2020/07/28 Python
英国男女奢华内衣和泳装购物网站:Figleaves
2017/01/28 全球购物
英国文具、办公用品和科技商店:Ryman
2018/09/27 全球购物
名人珠宝设计师:Melinda Maria Jewelry
2019/03/06 全球购物
AP澳洲中文网:澳洲正品直邮,包税收件无忧
2019/07/12 全球购物
人力资源部副职的竞聘演讲稿
2014/01/07 职场文书
求职简历自荐信
2014/06/18 职场文书
2016年党建工作简报
2015/11/26 职场文书
民警忠诚教育心得体会
2016/01/23 职场文书
Python基于Tkinter开发一个爬取B站直播弹幕的工具
2021/05/06 Python
Python标准库之typing的用法(类型标注)
2021/06/02 Python