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

相关推荐
一块plus7 分钟前
什么是去中心化 AI?区块链驱动智能的初学者指南
人工智能·后端·算法
txwtech15 分钟前
第10.4篇 使用预训练的目标检测网络
人工智能·计算机视觉·目标跟踪
羊小猪~~27 分钟前
【NLP入门系列四】评论文本分类入门案例
人工智能·自然语言处理·分类
roman_日积跬步-终至千里32 分钟前
【学习线路】机器学习线路概述与内容关键点说明
人工智能·学习·机器学习
都叫我大帅哥35 分钟前
向量数据库Milvus:非结构化数据的救星,AI开发者的瑞士军刀
java·python
静心问道36 分钟前
APE:大语言模型具有人类水平的提示工程能力
人工智能·算法·语言模型·大模型
zskj_zhyl40 分钟前
科技有温度:七彩喜智慧康养平台,为银发生活织就“数字守护网”
人工智能·科技·生活
嘉讯科技HIS系统42 分钟前
嘉讯科技:医疗信息化、数字化、智能化三者之间的关系和区别
大数据·数据库·人工智能·科技·智慧医疗
音视频牛哥44 分钟前
计算机视觉的新浪潮:扩散模型(Diffusion Models)技术剖析与应用前景
人工智能·计算机视觉·ai·音视频·实时音视频·扩散模型
爆改模型1 小时前
【arXiv2025】计算机视觉|即插即用|LBMamba:革新视觉模型效率,性能炸裂
人工智能·计算机视觉