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

相关推荐
酷雷曼VR全景5 分钟前
系统升级丨让VR全景制作更全面、更简单
人工智能·科技·vr·vr全景·酷雷曼
Goboy8 分钟前
Python修仙入门 - 踏入仙门的第一步
后端·python
链上日记15 分钟前
数字资产反诈指南:识别套路,守护WEEX账户安全
人工智能·安全
爱吃青菜的大力水手21 分钟前
软银收购 ABB 机器人业务的背景与 Physical AI 的关联
人工智能·机器人
长桥夜波42 分钟前
【第十七周】机器学习笔记06
人工智能·笔记·机器学习
程序员爱钓鱼44 分钟前
Python编程实战 · 基础入门篇 | 什么是Python
后端·python
golang学习记44 分钟前
Claude Code2.0.12发布插件系统:构建可扩展的 AI 编程工作流
人工智能
程序猿小D1 小时前
【完整源码+数据集+部署教程】遥感图像道路检测分割系统源码和数据集:改进yolo11-CARAFE
python·yolo·计算机视觉·目标跟踪·数据集·yolo11·遥感图像道路检测分割系统
minhuan1 小时前
构建AI智能体:六十二、金融风控系统:基于信息熵和KL散度的异常交易检测
人工智能·信息熵·kl散度·信息论·金融风控
zy_destiny1 小时前
【工业场景】用YOLOv8实现人员打电话识别
人工智能·python·深度学习·yolo·机器学习·计算机视觉·目标跟踪