DEV Community
·
PyTorch中的FashionMNIST
💡
原文英文，约400词，阅读约需2分钟。
📝
内容提要
本文介绍了如何使用Fashion-MNIST数据集，包括FashionMNIST()函数的参数设置，如数据路径、训练数据选择和转换函数。数据集包含60,000张训练图像和10,000张测试图像，用户可选择自动下载或手动下载。示例代码展示了如何加载和显示图像。
🎯
关键要点
- 本文介绍了如何使用Fashion-MNIST数据集。
- FashionMNIST()函数的第一个参数是root，表示数据路径，类型为str或pathlib.Path。
- 第二个参数是train，默认为True，表示使用训练数据（60,000张图像），如果为False则使用测试数据（10,000张图像）。
- 第三个参数是transform，默认为None，表示不进行数据转换。
- 第四个参数是target_transform，默认为None，表示不进行目标转换。
- 第五个参数是download，默认为False，表示不自动下载数据集。
- 如果download为True，数据集将从互联网下载并解压到指定路径。
- 用户可以手动下载数据集并将其放置在指定路径下。
- 示例代码展示了如何加载训练和测试数据集。
- 训练数据集包含60,000个数据点，测试数据集包含10,000个数据点。
- 数据集中包含10个类别的图像，如T恤、裤子、毛衣等。
- 提供了一个函数show_images用于显示训练和测试数据集中的图像。
❓
延伸问答
如何使用Fashion-MNIST数据集？
可以通过FashionMNIST()函数加载数据集，设置数据路径、选择训练或测试数据，并可选择是否进行数据转换。
FashionMNIST()函数的参数有哪些？
FashionMNIST()函数的参数包括root（数据路径）、train（是否使用训练数据）、transform（数据转换）、target_transform（目标转换）和download（是否自动下载数据集）。
Fashion-MNIST数据集中包含多少张图像？
数据集中包含60,000张训练图像和10,000张测试图像。
如何显示Fashion-MNIST数据集中的图像？
可以使用show_images函数来显示训练和测试数据集中的图像，函数会将图像以网格形式展示。
Fashion-MNIST数据集包含哪些类别的图像？
数据集包含10个类别的图像，如T恤、裤子、毛衣、裙子、外套、凉鞋、衬衫、运动鞋、包和踝靴。
如何手动下载Fashion-MNIST数据集？
用户可以手动下载数据集文件并将其放置在指定路径下，如data/FashionMNIST/raw/。
➡️
