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 30 Python
在Django中限制已登录用户的访问的方法
Jul 23 Python
详解python中xlrd包的安装与处理Excel表格
Dec 16 Python
python 读入多行数据的实例
Apr 19 Python
python开发游戏的前期准备
May 05 Python
python 计算数据偏差和峰度的方法
Jun 29 Python
windows下python虚拟环境virtualenv安装和使用详解
Jul 16 Python
Python while true实现爬虫定时任务
Jun 08 Python
python3:excel操作之读取数据并返回字典 + 写入的案例
Sep 01 Python
多个版本的python共存时使用pip的正确做法
Oct 26 Python
python定义具名元组实例操作
Feb 28 Python
总结Python使用过程中的bug
Jun 18 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实现WEB动态网页静态
2006/10/09 PHP
php 禁止页面缓存输出
2009/01/07 PHP
php获取ip及网址的简单方法(必看)
2017/04/01 PHP
Prototype使用指南之array.js
2007/01/10 Javascript
javascript 鼠标悬浮图片显示原图 移出鼠标后原图消失(多图)
2009/12/28 Javascript
写自已的js类库需要的核心代码
2012/07/16 Javascript
jquery实现用户信息修改验证输入方法汇总
2015/07/18 Javascript
JavaScript的new date等日期函数在safari中遇到的坑
2016/10/24 Javascript
js实现简单的二级联动效果
2017/03/09 Javascript
详谈Angular 2+ 的表单(一)之模板驱动型表单
2017/04/25 Javascript
JS文件中加载jquery.js的实例代码
2018/05/05 jQuery
JS获取子节点、父节点和兄弟节点的方法实例总结
2018/07/06 Javascript
ES6知识点整理之函数对象参数默认值及其解构应用示例
2019/04/17 Javascript
layui动态渲染生成select的option值方法
2019/09/23 Javascript
javascript刷新父页面方法汇总详解
2019/10/10 Javascript
微信小程序自定义组件components(代码详解)
2019/10/21 Javascript
js blob类型url的视频下载问题的解决
2019/11/29 Javascript
Python实现队列的方法
2015/05/26 Python
ansible作为python模块库使用的方法实例
2017/01/17 Python
使用Python实现图像标记点的坐标输出功能
2019/08/14 Python
Python queue队列原理与应用案例分析
2019/09/27 Python
使用matlab或python将txt文件转为excel表格
2019/11/01 Python
python队列原理及实现方法示例
2019/11/27 Python
对pytorch的函数中的group参数的作用介绍
2020/02/18 Python
Python的Django框架实现数据库查询(不返回QuerySet的方法)
2020/05/19 Python
使用Python webdriver图书馆抢座自动预约的正确方法
2021/03/04 Python
澳大利亚游乐场设备品牌:Lifespan Kids
2019/05/24 全球购物
学生会离职感言
2014/02/11 职场文书
《草原》教学反思
2014/02/15 职场文书
体育系毕业生求职自荐信
2014/04/16 职场文书
建材投资建议书
2014/05/16 职场文书
人事行政专员岗位职责
2014/07/23 职场文书
小学校园文化建设汇报材料
2014/08/19 职场文书
工作总结与自我评价
2014/09/18 职场文书
小学数学继续教育研修日志
2015/11/13 职场文书
eval(cmd)与eval($cmd)的区别与联系
2021/07/07 PHP