Pytorch显示图片

最新推荐文章于 2024-10-09 22:31:47 发布

赛博炼丹师

最新推荐文章于 2024-10-09 22:31:47 发布

阅读量557

点赞数

分类专栏： Pytorch 文章标签： pytorch python 深度学习人工智能

本文链接：https://blog.csdn.net/Victor_Li_/article/details/134021333

版权

Pytorch 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

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()