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程序与C程序的结合使用
Apr 07 Python
python操作excel文件并输出txt文件的实例
Jul 10 Python
python字符串替换第一个字符串的方法
Jun 26 Python
python3 selenium自动化测试 强大的CSS定位方法
Aug 23 Python
Python tkinter常用操作代码实例
Jan 03 Python
一文读懂Python 枚举
Aug 25 Python
python如何调用百度识图api
Sep 29 Python
基于Python组装jmx并调用JMeter实现压力测试
Nov 03 Python
基于Python实现天天酷跑功能
Jan 06 Python
tensorboard 可视化之localhost:6006不显示的解决方案
May 22 Python
Python还能这么玩之用Python做个小游戏的外挂
Jun 04 Python
Python中np.random.randint()参数详解及用法实例
Sep 23 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
FCKeditor的安装(PHP)
2007/01/13 PHP
ThinkPHP3.2.2的插件控制器功能简述
2014/07/09 PHP
iis6手工创建网站后无法运行php脚本的解决方法
2017/06/08 PHP
PHP封装类似thinkphp连贯操作数据库Db类与简单应用示例
2019/05/08 PHP
浅谈laravel 5.6 安装 windows上使用composer的安装过程
2019/10/18 PHP
Jquery AutoComplete自动完成 的使用方法实例
2010/03/19 Javascript
纯JavaScript实现的完美渐变弹出层效果代码
2010/04/02 Javascript
关于event.cancelBubble和event.stopPropagation()的区别介绍
2011/12/11 Javascript
js 操作select与option(示例讲解)
2013/12/20 Javascript
jquery中使用循环下拉菜单示例代码
2014/09/24 Javascript
浅谈Unicode与JavaScript的发展史
2015/01/19 Javascript
node.js实现复制文本到剪切板的功能
2017/01/23 Javascript
javascript中BOM基础知识总结
2017/02/14 Javascript
jQuery插件FusionCharts绘制的2D帕累托图效果示例【附demo源码】
2017/03/28 jQuery
js实现延迟加载的几种方法
2017/04/24 Javascript
分析javascript中9 个常见错误阻碍你进步
2017/09/18 Javascript
微信小程序实现跟随菜单效果和循环嵌套加载数据
2017/11/21 Javascript
react同构实践之实现自己的同构模板
2019/03/13 Javascript
原生js实现针对Dom节点的CRUD操作示例
2019/08/26 Javascript
[00:10]DOTA2 TI9勇士令状明日上线
2019/05/07 DOTA
在Python中使用pngquant压缩png图片的教程
2015/04/09 Python
python爬虫实现教程转换成 PDF 电子书
2017/02/19 Python
Python实现自动上京东抢手机
2018/02/06 Python
简单实现Python爬取网络图片
2018/04/01 Python
Python面向对象类的继承实例详解
2018/06/27 Python
python爬虫模拟浏览器访问-User-Agent过程解析
2019/12/28 Python
Python基于stuck实现scoket文件传输
2020/04/02 Python
PyQt5 文本输入框自动补全QLineEdit的实现示例
2020/05/13 Python
一款纯css3实现的响应式导航
2014/10/31 HTML / CSS
个人总结与自我评价
2014/09/18 职场文书
庆六一开幕词
2015/01/29 职场文书
2015年教师党员自我评价材料
2015/03/04 职场文书
电影开国大典观后感
2015/06/04 职场文书
《天使的翅膀》读后感3篇
2019/12/20 职场文书
Python语言规范之Pylint的详细用法
2021/06/24 Python
Python几种酷炫的进度条的方式
2022/04/11 Python