pytorch 加載mnist數據集報錯not gzip file


利用pytorch加載mnist數據集的代碼如下

import torchvision
import torchvision.transforms as transforms
from torch.utils.data import DataLoader

train_data = torchvision.datasets.MNIST(
    root='./mnist/',
    train=True,                                     # this is training data
    transform=torchvision.transforms.ToTensor(),    # Converts a PIL.Image or numpy.ndarray to
                                                    # torch.FloatTensor of shape (C x H x W) and normalize in the range [0.0, 1.0]
    download=True,
)
test_data = torchvision.datasets.MNIST(
    root='./mnist/',
    train=False,                                     # this is training data
    transform=torchvision.transforms.ToTensor(),    # Converts a PIL.Image or numpy.ndarray to
                                                    # torch.FloatTensor of shape (C x H x W) and normalize in the range [0.0, 1.0]
    download=True,
)

train_data_loader = DataLoader(train_data, shuffle=True, batch_size = 100)
test_data_loader = DataLoader(test_data,shuffle=True, batch_size=100)

第一次使用mnist,需要下載,具體方法就是設置download=True,然而我運行的時候報錯了。

錯誤:not gzip file

可是明明是.gz文件啊,查了幾篇博客也沒有說清楚原因的,於是自行下載了四個文件(訓練集,測試集以及各自的標簽),放入./mnist/raw文件夾下,運行,報錯:找不到文件。

此時,系統需要找的是./mnist/process文件夾下的train.pt和test.pt文件,這應該是pytorch下載原文件后處理生成的,可是我無處下載,於是用另一台電腦下載,程序沒有報錯,我把生成的.pt文件拷貝過來,可以運行了


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM