您现在的位置是：首页 >技术交流 >FashionMNIST数据集加载与可视化：PyTorch编程实践网站首页技术交流

# -*- coding: utf-8 -*-
"""
加载并显示FashionMNIST数据集

PyTorch Programming & Deep Learning
@author: Mike Yuan, Copyright 2020~2021
"""

import torchvision
import torchvision.transforms as transforms
import matplotlib.pyplot as plt

rows, cols = 10, 10  # 定义显示图像的网格布局为10行10列

def main():
    """ 主函数 """
    # 加载数据集
    # root: 数据集存储路径
    # train=True: 加载训练集；False: 加载测试集
    # download=False: 如果数据集不存在则不自动下载（需确保数据集已存在）
    # transform=transforms.ToTensor(): 将PIL图像转换为张量，并自动归一化像素值到[0,1]
    mnist_train = torchvision.datasets.FashionMNIST(
        root='../datasets/FashionMNIST', 
        train=True, 
        download=False,
        transform=transforms.ToTensor()
    )
    mnist_test = torchvision.datasets.FashionMNIST(
        root='../datasets/FashionMNIST', 
        train=