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

亦凉 2022-12-19 11:49 138阅读 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 条评论，138人围观）

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

相关阅读

相关 pytorch怎么加载city数据集

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

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

相关使用pytorch解析mnist数据集

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

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

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

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

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

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

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

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

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

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

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

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

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

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

相关（超详细）读取mnist数据集并保存成图片

mnist数据集介绍、读取、保存成图片 1、mnist数据集介绍： MNIST数据集是一个手写体数据集，简单说就是一堆这样东西 ![图片][322

红太狼/ 2022年06月09日 23:35/ 0 赞/ 565 阅读

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

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

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

相关 MNIST数据集介绍及读取

转载自：《[MINIST数据集介绍及读取][MINIST]》 MNIST数据集是机器学习领域中非常经典的一个数据集，由60000个训练样本和10000个测试样本组成，每

冷不防/ 2022年01月20日 15:17/ 0 赞/ 451 阅读

相关 Pytorch 之 MNIST 数据集实现

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

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