python-pytorch获取FashionMNIST实际图片标签数据集

在查看pytorch官方文档的时候，在这里链接中https://pytorch.org/tutorials/beginner/basics/data_tutorial.html的Creating a Custom Dataset for your files章节，有提到要自定义数据集，需要用到实际的图片和标签。

在网上找了半天没找到，写了一个脚本将图片和标签文本下载到本地。

import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor

# Download training data from open datasets.
training_data = datasets.FashionMNIST(
    root="data",
    train=True,
    download=True,
    transform=ToTensor(),
)

# Download test data from open datasets.
test_data = datasets.FashionMNIST(
    root="data",
    train=False,
    download=True,
    transform=ToTensor(),
)

# 写入到本地
count=0
for index,x in test_data:
    print(index.size(),x)
    count=count+1
    

    classes = [
        "T-shirttop",
        "Trouser",
        "Pullover",
        "Dress",
        "Coat",
        "Sandal",
        "Shirt",
        "Sneaker",
        "Bag",
        "Ankleboot",
    ]



    import torch
    from torchvision.utils import save_image
    folder_path = './data/imageandlableTest'  # 替换为你的文件夹路径
    filename = '{}{}.jpg'.format(classes[x],count)  # 图片文件名

    # 确保文件夹存在
    import os
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)

    # 保存图片
    save_path = os.path.join(folder_path, filename)
    save_image(index, save_path)



    with open('./data/imageandlableTest/output.txt', 'a') as f:  
        f.write("{},{}\n".format(filename,x))
    print(count)