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中的tuple元组详细介绍
Feb 02 Python
简单谈谈Python中的反转字符串问题
Oct 24 Python
Python实现替换文件中指定内容的方法
Mar 19 Python
详解Python给照片换底色(蓝底换红底)
Mar 22 Python
python循环定时中断执行某一段程序的实例
Jun 29 Python
Python 3 判断2个字典相同
Aug 06 Python
pytorch之inception_v3的实现案例
Jan 06 Python
Python实现名片管理系统
Feb 14 Python
Python PyQt5模块实现窗口GUI界面代码实例
May 12 Python
Python实现数字的格式化输出
Aug 01 Python
Python Opencv图像处理基本操作代码详解
Aug 31 Python
python软件测试Jmeter性能测试JDBC Request(结合数据库)的使用详解
Jan 26 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实现Socket服务器的代码
2008/04/03 PHP
ThinkPHP里用U方法调用js文件实例
2015/06/18 PHP
CI框架表单验证实例详解
2016/11/21 PHP
PHP微信H5支付开发实例
2018/07/25 PHP
laravel实现于语言包的完美切换方法
2019/09/29 PHP
基于Laravel(5.4版本)的基本增删改查操作方法
2019/10/11 PHP
JS实现浏览器状态栏文字从右向左弹出效果代码
2015/10/27 Javascript
javascript给span标签赋值的方法
2015/11/26 Javascript
jquery if条件语句的写法
2016/05/19 Javascript
使用jQuery实现WordPress中的Ctrl+Enter和@评论回复
2016/05/21 Javascript
Validform表单验证总结篇
2016/10/31 Javascript
jquery Easyui Datagrid实现批量操作(编辑,删除,添加)
2017/02/20 Javascript
Vue.js实战之使用Vuex + axios发送请求详解
2017/04/04 Javascript
微信小程序 实例开发总结
2017/04/26 Javascript
表格展示利器 Bootstrap Table实例代码
2017/09/06 Javascript
基于Bootstrap框架菜鸟入门教程(推荐)
2017/09/17 Javascript
vue2.0 路由不显示router-view的解决方法
2018/03/06 Javascript
详解Vue项目在其他电脑npm run dev运行报错的解决方法
2018/10/29 Javascript
layui form表单提交后实现自动刷新
2019/10/25 Javascript
vue设置一开始进入的页面教程
2019/10/28 Javascript
在Python中使用模块的教程
2015/04/27 Python
深入解析Python设计模式编程中建造者模式的使用
2016/03/02 Python
Python随机读取文件实现实例
2017/05/25 Python
Python中的支持向量机SVM的使用(附实例代码)
2019/06/26 Python
django修改models重建数据库的操作
2020/03/31 Python
浅析PyCharm 的初始设置(知道)
2020/10/12 Python
阻止移动设备(手机、pad)浏览器双击放大网页的方法
2014/06/03 HTML / CSS
毕业生的自我评价分享
2013/12/18 职场文书
博士学位自我鉴定范文
2013/12/26 职场文书
11月升旗仪式讲话稿
2014/02/15 职场文书
安徽导游词
2015/02/12 职场文书
护士个人年终总结
2015/02/13 职场文书
2015年清明节演讲稿范文
2015/03/17 职场文书
浅谈redis五大数据结构和使用场景
2021/04/12 Redis
学会Python数据可视化必须尝试这7个库
2021/06/16 Python
Django数据库(SQlite)基本入门使用教程
2022/07/07 Python