pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
更新时间：2020年01月13日 09:08:13 作者：小星星*
今天小编就为大家分享一篇pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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,数组等的转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了sklearn中make_blobs的用法详情，sklearn中的make_blobs函数主要是为了生成数据集的，下面文章的学习内容，需要的小伙伴可以参考一下2022-01-01
今天小编就为大家分享一篇python实现逆滤波与维纳滤波示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
这篇文章主要介绍了基于Python实现剪切板实时监控方法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-09-09
python结合shell自动创建kafka的连接器实战教程
这篇文章主要介绍了python结合shell自动创建kafka的连接器,需要安装连接oracle的python包，获取oracle表信息，本文给大家介绍的非常详细，需要的朋友可以参考下2022-04-04
每天都是浪漫七夕！这篇文章主要为大家详细介绍了python爱心表白，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-08-08
非极大值抑制（Non-Maximum Suppression，NMS），顾名思义就是抑制不是极大值的元素，可以理解为局部最大搜索。这个局部代表的是一个邻域，邻域有两个参数可变，一是邻域的维数，二是邻域的大小2021-06-06
这篇文章主要介绍了ORM Django 终端打印 SQL 语句实现解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
这篇文章主要介绍了Python3之数据指纹MD5校验与对比，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-06-06
这篇文章主要介绍了详解python中executemany和序列的使用方法的相关资料,需要的朋友可以参考下2017-08-08
python3实现语音转文字(语音识别)和文字转语音(语音合成)
这篇文章主要介绍了python3实现语音转文字(语音识别)和文字转语音(语音合成)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-10-10
最新评论