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 绘图和可视化详细介绍
Feb 11 Python
python生成器,可迭代对象,迭代器区别和联系
Feb 04 Python
python实现监控某个服务 服务崩溃即发送邮件报告
Jun 21 Python
Python递归函数实例讲解
Feb 27 Python
PyQt4编程之让状态栏显示信息的方法
Jun 18 Python
Pandas删除数据的几种情况(小结)
Jun 21 Python
tensor和numpy的互相转换的实现示例
Aug 02 Python
pyhton中__pycache__文件夹的产生与作用详解
Nov 24 Python
python创建n行m列数组示例
Dec 02 Python
解决TensorFlow GPU版出现OOM错误的问题
Feb 03 Python
Python使用pyyaml模块处理yaml数据
Apr 14 Python
python绘制简单直方图(质量分布图)的方法
Apr 21 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 身份证号验证函数
2009/05/07 PHP
初步介绍PHP扩展开发经验分享
2012/09/06 PHP
jquery不支持toggle()高(新)版本的问题解决
2016/09/24 PHP
php compact 通过变量创建数组
2016/11/15 PHP
PHPMailer ThinkPHP实现自动发送邮件功能
2018/06/10 PHP
详解PHP神奇又有用的Trait
2019/03/25 PHP
PHP使用Redis实现Session共享的实现示例
2019/05/12 PHP
如何重写Laravel异常处理类详解
2020/12/20 PHP
Maps Javascript
2007/01/22 Javascript
防止页面被iframe(兼容IE,Firefox火狐)
2010/07/04 Javascript
js 通用javascript函数库整理
2011/08/14 Javascript
javascript日期计算实例分析
2015/06/29 Javascript
jquery自适应布局的简单实例
2016/05/28 Javascript
javascript创建含数字字母的随机字符串方法总结
2016/08/01 Javascript
Vue自定义指令拖拽功能示例
2017/02/17 Javascript
jQuery遍历节点方法汇总(推荐)
2017/05/13 jQuery
JS判断数组那点事
2017/10/10 Javascript
vue 挂载路由到头部导航的方法
2017/11/13 Javascript
React组件refs的使用详解
2018/02/09 Javascript
微信小程序实现的一键复制功能示例
2019/04/24 Javascript
基于layui轮播图满屏是高度自适应的解决方法
2019/09/16 Javascript
[00:10]DOTA2全国高校联赛 以DOTA2会友
2018/05/30 DOTA
Python struct.unpack
2008/09/06 Python
Python装饰器实现几类验证功能做法实例
2017/05/18 Python
python进阶之多线程对同一个全局变量的处理方法
2018/11/09 Python
Python3 实现串口两进程同时读写
2019/06/12 Python
python读取Excel表格文件的方法
2019/09/02 Python
Python warning警告出现的原因及忽略方法
2020/01/31 Python
详解Html5原生拖拽操作
2018/01/12 HTML / CSS
四方通行旅游网:台湾订房、出国旅游
2017/09/20 全球购物
高中考试作弊检讨书
2014/01/14 职场文书
医学院毕业生自荐信范文
2014/03/06 职场文书
2014年会计主管工作总结
2014/12/20 职场文书
校园运动会广播稿
2015/08/19 职场文书
《莫泊桑拜师》教学反思
2016/02/22 职场文书
css3带你实现3D转换效果
2022/02/24 HTML / CSS