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()
相关推荐
love530love6 分钟前
教程:PyCharm 中搭建多级隔离的 Poetry 环境(从 Anaconda 到项目专属.venv)
开发语言·ide·人工智能·windows·python·pycharm
聚客AI7 分钟前
PyTorch进阶:从自定义损失函数到生产部署全栈指南
人工智能·pytorch·深度学习
Apache RocketMQ9 分钟前
RocketMQ 客户端负载均衡机制详解及最佳实践
人工智能
Ian在掘金12 分钟前
bat+python实现easy connect自动连接
前端·python
midsummer_woo13 分钟前
【2025】pycharm 安装
ide·python·pycharm
可爱美少女19 分钟前
Kaggle-Predicting Optimal Fertilizers-(多分类+xgboost+同一特征值多样性)
人工智能·分类·数据挖掘
AщYΘ23 分钟前
6.6 day38
python
老胖闲聊33 分钟前
Python ROS2【机器人中间件框架】 简介
python·中间件·机器人
社会零时工44 分钟前
【OpenCV】相机标定之利用棋盘格信息标定
人工智能·数码相机·opencv
像素工坊可视化1 小时前
监控升级:可视化如何让每一个细节 “说话”
运维·人工智能·安全