一文读懂Fashion-MNIST：从图像存储到标签体系的完整解析

2026-02-04 05:19:15作者：郜逊炳

你是否在机器学习入门时遇到过MNIST数据集？这个包含手写数字的经典数据集虽然易用，但在现代计算机视觉任务中已经显得过于简单。Fashion-MNIST作为MNIST的替代品，提供了更贴近实际应用场景的时尚产品图像数据，成为算法 benchmark 的理想选择。读完本文后，你将能够：

理解Fashion-MNIST数据集的文件结构与存储格式
掌握图像数据的二进制解析方法
熟悉10类时尚产品的标签体系
学会如何加载并可视化数据集

数据集整体结构

Fashion-MNIST数据集采用与MNIST完全兼容的文件格式，这意味着你可以无缝替换现有MNIST的加载代码。项目的data/fashion/目录下包含四类核心文件：

train-images-idx3-ubyte.gz  # 训练集图像
train-labels-idx1-ubyte.gz  # 训练集标签
t10k-images-idx3-ubyte.gz   # 测试集图像
t10k-labels-idx1-ubyte.gz   # 测试集标签

这种设计使研究人员能够在不修改代码的情况下，直接对比算法在数字识别和时尚产品识别上的性能差异。数据集包含60,000个训练样本和10,000个测试样本，每个样本都是28×28像素的灰度图像，对应10类不同的时尚产品。

图像文件格式解析

图像文件采用IDX（Index）格式存储，这是一种简单的二进制文件格式，用于存储向量和多维矩阵数据。以训练集图像文件train-images-idx3-ubyte.gz为例，其结构包含以下几个部分：

文件头结构

魔数（Magic Number）：32位整数，用于标识文件类型和数据格式
样本数量：32位整数，表示文件中包含的图像数量
图像高度：32位整数，每个图像的高度（像素）
图像宽度：32位整数，每个图像的宽度（像素）

像素数据

紧随文件头之后的是像素数据，每个像素用一个8位无符号整数表示（0-255），对应灰度值。所有图像按顺序存储，每个图像的像素以行为主序（row-major order）排列，形成784个字节的扁平向量（28×28）。

项目中的utils/mnist_reader.py文件提供了完整的Python加载实现，核心代码如下：

def load_mnist(path, kind='train'):
    import os
    import gzip
    import numpy as np

    labels_path = os.path.join(path,
                              '%s-labels-idx1-ubyte.gz'
                              % kind)
    images_path = os.path.join(path,
                              '%s-images-idx3-ubyte.gz'
                              % kind)

    with gzip.open(labels_path, 'rb') as lbpath:
        labels = np.frombuffer(lbpath.read(), dtype=np.uint8,
                              offset=8)

    with gzip.open(images_path, 'rb') as imgpath:
        images = np.frombuffer(imgpath.read(), dtype=np.uint8,
                              offset=16).reshape(len(labels), 784)

    return images, labels

这段代码通过numpy.frombuffer直接从二进制数据中读取像素值，并通过reshape方法将扁平向量恢复为28×28的图像尺寸。注意文件头的偏移量：标签文件跳过前8字节，图像文件跳过前16字节。

标签体系详解

Fashion-MNIST包含10类时尚产品，每类对应一个0-9的整数标签。不同于MNIST的数字标签，Fashion-MNIST的标签代表具有实际意义的产品类别：

标签	类别名称	英文名称
0	T恤/上衣	T-shirt/top
1	裤子	Trouser
2	套头衫	Pullover
3	连衣裙	Dress
4	外套	Coat
5	凉鞋	Sandal
6	衬衫	Shirt
7	运动鞋	Sneaker
8	包	Bag
9	短靴	Ankle boot

这种标签设计涵盖了上衣、下装、鞋履和配饰等基本时尚品类，图像包含不同角度、颜色和款式的产品，为算法提供了适度的识别挑战。

上图展示了使用UMAP算法对Fashion-MNIST数据集进行的降维可视化，不同颜色代表不同类别的产品。可以看到，某些类别（如凉鞋和短靴）在特征空间中形成了较为集中的簇，而另一些类别（如衬衫和T恤）则有较多重叠，这反映了实际分类难度的差异。

数据加载与可视化实践

要在你的项目中使用Fashion-MNIST，只需调用utils/mnist_reader.py中的load_mnist函数，并指定数据路径和数据集类型：

from utils.mnist_reader import load_mnist

X_train, y_train = load_mnist('data/fashion', kind='train')
X_test, y_test = load_mnist('data/fashion', kind='t10k')

print(f"训练集规模: {X_train.shape}")  # (60000, 784)
print(f"测试集规模: {X_test.shape}")    # (10000, 784)

加载后的数据可以通过matplotlib进行可视化：

import matplotlib.pyplot as plt

# 显示前9个训练样本
fig, ax = plt.subplots(3, 3, figsize=(10, 10))
for i, axi in enumerate(ax.flat):
    axi.imshow(X_train[i].reshape(28, 28), cmap='gray')
    axi.set_title(f"Label: {y_train[i]}")
    axi.axis('off')
plt.show()