Pytorch从本地获取数据集

本文详细介绍如何在Pytorch中从本地加载MNIST和Fashion MNIST数据集,避免网络下载,包括修改数据源路径和使用ToTensor转换。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Pytorch从本地获取数据集

  • 在学习pytorch的过程中需要从MNIST获取数据集,然而下载是让人头疼的事,从网上寻找数据资源比较便捷
  • 获取到的数据如何在pytorch中加载呢

1 下载数据集

2. 从本地进行数据加载

  • 获取测试集与训练集

    直接运行后,发现依旧是下载数据,那我本地的数据集怎么才能被加载

    mnist_train = torchvision.datasets.FashionMNIST(root='./MNIST', train=True, download=True,transform=transforms.ToTensor())
    mnist_test = torchvision.datasets.FashionMNIST(root='./MNIST', train=False, download=True,transform=transforms.ToTensor())
  
  • 查找数据源url

按下Ctrl,左键点击FashionMNIS,进入mnist.py,在resources下可以看到,这里是数据集的下载路径,需要将其修改为本地文件的路径

在这里插入图片描述

  • 查找本地数据源
    在这里插入图片描述

  • 将本地数据源替换之前的路径

特别要注意后面的None,这个是md5校验码,如果不填会报错

在这里插入图片描述

  • 运行程序,加载数据集

        mnist_train = torchvision.datasets.FashionMNIST(root='./MNIST', train=True, download=True,transform=transforms.ToTensor())
        mnist_test = torchvision.datasets.FashionMNIST(root='./MNIST', train=False, download=True,transform=transforms.ToTensor())
        print(type(mnist_train))
        print(len(mnist_train), len(mnist_test))
    

在这里插入图片描述
完成

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

wangxiaobei2017

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值