[pytorch学习笔记]permute()和reshape()图像操作小结

本文总结了PyTorch中图像处理函数permute()和reshape()的使用方法。permute()用于改变图像张量的维度顺序,特别指出在从本地读取RGB图像后如何调整其维度以便正确显示。reshape()则用于调整张量的形状。
Qwen3-32B-Chat 私有部署镜像 | RTX4090D 24G 显存 CUDA12.4 优化版

本镜像基于 RTX 4090D 24GB 显存 + CUDA 12.4 + 驱动 550.90.07 深度优化,内置完整运行环境与 Qwen3-32B 模型依赖,开箱即用。

permute()和reshape()图像操作小结

permute()

1.首先要对RGB图像基础有一定的了解

详见:关于三通道彩色图像的存储方式理解

2.permute()是如何变换的

这是我们常见的permute()变换,在这张图中,其他作者用的是随机数,与实际的像素值有所差距,但并不影响理解。

请添加图片描述

RGB三种颜色的灰度值被permute()分别提取出来,此时被提取出来的只是灰度(白色->灰色->黑色),后面要加上不同颜色的通道才能真正的显示出来。

相关资料:pytorch中permute()函数用法实例详解

值得注意的是:

在使用pytorch从本地读取图像的时候,默认把图像的RGB三种颜色的灰度值提取出来,这时如果想要把图像绘制在界面上的话,请使用permute(1,2,0)将原来的张量转换一下。

reshape()

在这里插入图片描述
在这里插入图片描述
参考资料:Python中reshape函数参数-1的意思?

您可能感兴趣的与本文相关的镜像

Qwen3-32B-Chat 私有部署镜像 | RTX4090D 24G 显存 CUDA12.4 优化版

Qwen3-32B-Chat 私有部署镜像 | RTX4090D 24G 显存 CUDA12.4 优化版

Qwen
文本生成
Qwen3

本镜像基于 RTX 4090D 24GB 显存 + CUDA 12.4 + 驱动 550.90.07 深度优化,内置完整运行环境与 Qwen3-32B 模型依赖,开箱即用。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值