tensor转换为图片_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,数组等的转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

tensor转换为图片_pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换相关推荐

pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换 1, 创建pytorch 的Tensor张量: torch.rand((3,224,224)) #创建随机值的三维张量,大小为 ...
pytorch 张量tensor 转为 jpg 图片
在用pytorch训练网络的时候,每一张图片都被转为了张量形式的数据,有时候我们需要将这些张量转为jpg格式的图片,应该如何转变呢?我们只要将张量tensor转为PIL格式的数据,然后再将数据保存起来 ...
图片转化为向量Tensor的方法
图片转化为向量Tensor的方法两种方法报错1: 结果 pytorch中比较所得两个tensor是否相等源代码解析 ToTensor() to_tensor(pic) 图像数据读取 PIL与CV ...
将tensor转换为image
将图片转换为tensor的时候,图片发生了几个变化; 首先图片的通道维度发生了变化,从h,w,c变化成了b,c,h,w 图像的像素范围发生了变化,从0-255变化成了0-1 图片的数据类型发生了变化, ...
什么是张量 (tensor)？
在TensorFlow 官方文档中文版中学习cifar10时,对张量Tensor的概念一知半解,特别在理解代码如何批量读取图片,如何实现卷积的细节困扰了我好几天,在查询网上的资料后总结如下: 一.Te ...
Pytorch入门之张量tensor
一.张量tensor 张量的三个特征:秩.轴.形状张量的秩是指索引的个数,轴是指每一个维度的最大的索引的值,张量的形状提供了维度和索引的数量关系. 经常需要对张量进行重塑 t.reshape(1,9 ...
Pytorch张量tensor的使用
1. 张量Tensor Tensors张量: 张量的概念类似于Numpy中的ndarray数据结构, 最大的区别在于Tensor可以利用GPU的加速功能. 张量是一个统称,其中包含很多类型: [各 ...
tensor数据类型转换_PyTorch的tensor数据类型及其相关转换
1 基本类型 PyTorch的数据类型为各式各样的tensor,tensor可以理解为高维矩阵,与Numpy中的array类似. PyTorch中的tensor包括CPU上的数据类型和GPU上的数据类 ...
判定两个tensor维度相同_Pytorch之Tensor大详解
导语深度学习框架Pytorch发展势头惊人,这点小编也深有体会,翻翻Github上深度学习的开源代码,发现用Pytorch真的多了不少,所以小编最近也正在入坑Pytorch,顺便写写文章做些总结.认 ...

tensor转换为图片_pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换

tensor转换为图片_pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换相关推荐

最新文章

热门文章