pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
作者:小星星* 时间:2023-08-12 07:44:34
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)
来源:https://blog.csdn.net/zsx1713366249/article/details/93619101
标签:pytorch,tensor,图片,CPU,GPU,数组


猜你喜欢
如何正确合理的建立MYSQL数据库索引
2010-10-25 20:08:00
一篇文章弄懂Python中的内建函数
2023-01-18 00:36:36

讲解SQL Server危险扩展存储删除和恢复
2008-12-09 14:30:00
Python PaddlePaddle机器学习之求解线性模型
2023-04-19 08:35:14
在SQL触发器或存储过程中获取在程序登录的用户
2024-01-29 09:30:40
Python基础之模块详解
2023-02-24 23:23:52

pyramid配置session的方法教程
2021-04-26 09:23:37
Python实现归一化算法详情
2023-05-11 12:18:33

浅谈Python 递归算法指归
2023-01-12 06:27:50

松子学ASP—Server篇
2008-04-22 14:14:00
win10系统下python3安装及pip换源和使用教程
2023-07-10 19:48:40

Python 制作自动化翻译工具
2022-08-17 05:34:50

css基础教程属性篇之盒子模型
2008-07-29 12:05:00

jQuery中ajax和post处理json的不同示例对比
2024-04-16 08:54:10
vue单页面在微信下只能分享落地页的解决方案
2024-05-09 10:52:19
在ASP.NET 2.0中操作数据之五十:为GridView控件添加Checkbox
2023-06-26 19:18:09

Typescript3.9 常用新特性一览(推荐)
2024-04-18 10:56:16
vue中provide和inject的用法及说明(vue组件爷孙传值)
2024-05-21 10:15:26

手把手教你vscode配置golang开发环境的步骤
2024-04-23 09:35:34

python爬虫selenium模块详解
2023-04-11 20:32:43