Pytorch显示图片

python 复制代码
import torch
import torchvision
import matplotlib.pyplot as plt

batch_size = 512

train_loader = torch.utils.data.DataLoader(
    torchvision.datasets.MNIST('mnist data',train=True,download=True,
                               transform=torchvision.transforms.Compose([
                               torchvision.transforms.ToTensor(),
                               torchvision.transforms.Normalize((0.1307,),(0.3081,))])),
                                batch_size=batch_size,shuffle=True)

test_loader = torch.utils.data.DataLoader(
    torchvision.datasets.MNIST('mnist data/',train=False,download=True,
                               transform=torchvision.transforms.Compose([
                               torchvision.transforms.ToTensor(),
                               torchvision.transforms.Normalize((0.1307,),(0.3081,))])),
                                batch_size=batch_size,shuffle=False)

x,y = next(iter(train_loader))
plt.imshow(x[0].permute(1, 2, 0))  # 将通道维度调整到最后
plt.axis('off')  # 关闭坐标轴
plt.show()
相关推荐
说私域5 分钟前
基于开源AI智能名片链动2 + 1模式S2B2C商城小程序的沉浸式体验营销研究
人工智能·小程序·开源·零售
.似水11 分钟前
Python requests
开发语言·python
亚马逊云开发者13 分钟前
飞书多维表格利用 Amazon Bedrock AI 能力赋能业务
人工智能
不会飞的鲨鱼13 分钟前
FastMoss 国际电商Tiktok数据分析 JS 逆向 | MD5加密
javascript·python·数据挖掘·数据分析
阿部多瑞 ABU20 分钟前
# 主流大语言模型安全性测试(二):英文越狱提示词下的表现与分析
人工智能·安全·ai·语言模型·自然语言处理·安全性测试
tanyyinyu31 分钟前
Python列表:高效灵活的数据存储与操作指南
开发语言·windows·python
加油搞钱加油搞钱43 分钟前
鹰盾加密器“一机一码”技术全维度剖析:从底层实现到生态防护体系
开发语言·网络·python
小天才才1 小时前
【大模型】解耦大语言模型中的记忆与推理能力
人工智能·深度学习·语言模型·自然语言处理
站大爷IP1 小时前
Python项目文件组织与PyCharm实践:打造高效开发环境
python
AI大模型学习教程1 小时前
前端学AI之LangChain.js入门教程:实现智能对话机器人
人工智能·langchain