pytorch 1.7.0 读取mnist数据集加载并显示图像

亦凉 2022-12-19 11:49 148阅读 0赞

完整源代码：

import torch
    import torchvision
    import torchvision.transforms as transforms
    from matplotlib import pyplot
    import matplotlib.pyplot as plt
    import numpy as np
    
    batch_size = 64
    
    train_dataset = torchvision.datasets.MNIST(root='data', train=True, transform=transforms.ToTensor(), download=True)
    test_dataset = torchvision.datasets.MNIST(root='data', train=False, transform=transforms.ToTensor())
    
    train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True)
    test_loader = torch.utils.data.DataLoader(dataset=test_dataset, batch_size=batch_size, shuffle=True)
    
    
    def matplotlib_imshow(img, one_channel=False):
        if one_channel:
            img = img.mean(dim=0)
        # img = img / 2 + 0.5     # unnormalize
        npimg = img.numpy()
        if one_channel:
            pyplot.imshow(npimg, cmap="Greys")
            pyplot.show()
        else:
            plt.imshow(np.transpose(npimg, (1, 2, 0)))
    
    
    data_iter = iter(train_loader)
    images, labels = data_iter.next()
    img_grid = torchvision.utils.make_grid(images)
    matplotlib_imshow(img_grid, one_channel=True)
    
    print(torch.__version__)

效果如下：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0Jva21hbg_size_16_color_FFFFFF_t_70][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0Jva21hbg_size_16_color_FFFFFF_t_70]: /images/20221120/3b7f45a549dc4f51bdf9b68e92996b6a.png

发表评论取消回复

表情：

评论列表（有 0 条评论，148人围观）

还没有评论，来说两句吧...

相关阅读

相关【PyTorch】MNIST数据集图像可视化方法

问题怎么将MNIST数据集里的数字图像数据可视化？方法基于pytorch框架，torchvision.datasets.MNIST导入数据集，对图片数据进行

桃扇骨/ 2024年04月06日 12:45/ 0 赞/ 70 阅读

相关 pytorch怎么加载city数据集

PyTorch可以使用torchvision库中的torchvision.datasets模块加载Cityscapes数据集，它有一些类似的数据集，如COCO，VOC等。使用t

谁借莪１个温暖的怀抱￠/ 2024年03月25日 19:26/ 0 赞/ 30 阅读

相关使用pytorch解析mnist数据集

当解析MNIST数据集时，以下是代码的详细介绍： 1. \\导入必要的库\\： import torch import torchvision

小鱼儿/ 2024年02月20日 10:04/ 0 赞/ 76 阅读

相关 pytorch学习（三）—图像的加载/读取方式

import matplotlib.pyplot as plt import skimage.io as io import cv2 from

拼搏现实的明天。/ 2023年06月06日 12:01/ 0 赞/ 4 阅读

相关 PyTorch 07： PyTorch加载图像数据

到目前为止，我们一直使用的都是虚拟数据集。但在实际运用里，你会处理一些完整尺寸的图像，例如智能手机摄像头拍摄的图像。现在，我们将学习如何加载图像并使用它们训练神经网络。我们

╰+攻爆jí腚メ/ 2023年03月13日 15:29/ 0 赞/ 21 阅读

相关 pytorch 1.7.0 读取mnist数据集加载并显示图像

完整源代码： import torch import torchvision import torchvision.transforms as tra

亦凉/ 2022年12月19日 11:49/ 0 赞/ 149 阅读

相关 PyTorch—— 图像分类数据集（Fashion-MNIST）

PyTorch—— 图像分类数据集（Fashion-MNIST） 0、前言一、获取数据集二、读取小批量数据三、

古城微笑少年丶/ 2022年12月04日 01:03/ 0 赞/ 272 阅读

相关 Pytorch读取MNIST数据集并保存里面的图片

Pytorch读取MNIST数据集并保存里面的图片最近在学习Pytorch编程，在里面训练模型的时候需要用到比较常用的MNIST数据集，在此做出总结，以便后面不时之需。

绝地灬酷狼/ 2022年09月03日 05:21/ 0 赞/ 186 阅读

相关 pytorch加载pascal&&coco数据集

上一篇博客[https://blog.csdn.net/goodxin\_ie/article/details/84315458][https_blog.csdn.net_go

ゞ浴缸里的玫瑰/ 2022年04月15日 04:10/ 0 赞/ 454 阅读

相关 Pytorch 之 MNIST 数据集实现

目录 1. 数据集介绍 2. 代码 2. 读代码（个人喜欢的顺序） 2.1. 导入模块部分： 2.2. Main 函数：

逃离我推掉我的手/ 2021年12月13日 04:02/ 0 赞/ 448 阅读