pytorch保存张量为图片

这里用到的是torchvision中的save_image。

废话不多说,直接来代码:

python 复制代码
import torch
from torchvision.utils import save_image
B, C, H, W = 64, 3, 32, 32
input_tensor = torch.randn(B, C, H, W) 

save_image(input_tensor, "hh.png", nrow=8)

首先,这里我们要明确 ( B , C , H , W ) (B, C, H, W) (B,C,H,W)的意思,我们有64张3通道的图片需要可视化。那么,我设定每排放8张。实现效果如图:

参考链接:https://pytorch.org/vision/stable/generated/torchvision.utils.save_image.html

相关推荐
橡晟3 小时前
深度学习入门:让神经网络变得“深不可测“⚡(二)
人工智能·python·深度学习·机器学习·计算机视觉
墨尘游子3 小时前
神经网络的层与块
人工智能·python·深度学习·机器学习
Leah01053 小时前
什么是神经网络,常用的神经网络,如何训练一个神经网络
人工智能·深度学习·神经网络·ai
倔强青铜33 小时前
苦练Python第18天:Python异常处理锦囊
开发语言·python
PyAIExplorer4 小时前
图像亮度调整的简单实现
人工智能·计算机视觉
企鹅与蟒蛇4 小时前
Ubuntu-25.04 Wayland桌面环境安装Anaconda3之后无法启动anaconda-navigator问题解决
linux·运维·python·ubuntu·anaconda
autobaba4 小时前
编写bat文件自动打开chrome浏览器,并通过selenium抓取浏览器操作chrome
chrome·python·selenium·rpa
Striker_Eureka4 小时前
DiffDet4SAR——首次将扩散模型用于SAR图像目标检测,来自2024 GRSL(ESI高被引1%论文)
人工智能·目标检测
Rvelamen5 小时前
LLM-SECURITY-PROMPTS大模型提示词攻击测评基准
人工智能·python·安全
【本人】5 小时前
Django基础(一)———创建与启动
后端·python·django