【Pytorch】--CIFAR10等数据集本地读取

最新推荐文章于 2025-04-20 18:20:22 发布

梦醒时分1218

最新推荐文章于 2025-04-20 18:20:22 发布

阅读量6.8k

点赞数 3

分类专栏： Pytorch

本文链接：https://blog.csdn.net/qq_35632833/article/details/104251683

版权

Pytorch 专栏收录该内容

4 篇文章

订阅专栏

文章目录

1.问题

$\quad$ 在使用Pytorch的时候，有时候需要在线下载数据集，因为在下载的过程中，封装好的代码，还要进行其他的操作(例如数据类型转换numpy->tensor)，但是有时候因为下载网站在国外，进度条一直显示0%,
$\quad$ 就像这样：
在这里插入图片描述

2.解决办法

step1.下载数据集到本地
step2. 将本地存放CIFAR数据集路径放在浏览器下，回车
step3. 修改class CIFAR10(VisionDataset)中的url

step4. 运行代码

import torch
import torchvision
LOAD_CIFAR = True 
DOWNLOAD_CIFAR = True

train_data = torchvision.datasets.CIFAR10(
    root='./cifar10/', 
    train=True, 
    transform=torchvision.transforms.ToTensor(), 
    download=DOWNLOAD_CIFAR,
)

一切ok
在这里插入图片描述

3.显示CIFAR10的图像

import torch
import torchvision
import matplotlib.pyplot as plt

EPOCH = 1
BATCH_SIZE = 50
LR = 0.001
DOWNLOAD_CIFAR = False

classes = ('plane', 'car', 'bird', 'cat',
           'deer', 'dog', 'frog', 'horse', 'ship', 'truck')

train_data = torchvision.datasets.CIFAR10(
    root='./cifar10/',  # 保存或者提取位置
    train=True,  # this is training data
    transform=torchvision.transforms.ToTensor(),  # 转换 PIL.Image or numpy.ndarray 成
    # torch.FloatTensor (C x H x W), 训练的时候 normalize 成 [0.0, 1.0] 区间
    download=DOWNLOAD_CIFAR,  # 没下载就下载, 下载了就不用再下了
)

# method 1 
# dataiter = iter(train_data)
# plt.show()
# for _ in range(len(train_data)):
#     images, labels = dataiter.__next__()
#     images = images.numpy().transpose(1, 2, 0)  # 把channel那一维放到最后
#     plt.title(str(classes[labels]))
#     plt.imshow(images)
#     plt.pause(1)

# as 2 list
# method 2
plt.show()
for images, labels in train_data:
    images = images.numpy().transpose(1, 2, 0)  # 把channel那一维放到最后
    plt.title(str(classes[labels]))
    plt.imshow(images)
    plt.pause(1)