PyTorch中的ImageNet
💡
原文英文，约500词，阅读约需2分钟。
📝
内容提要
本文介绍了如何使用ImageNet数据集，包括ImageNet()函数的参数设置，如数据根目录、数据分割（训练或验证）、变换和加载器。用户需手动下载数据集并运行ImageNet()以提取数据。训练集包含1281167张图像，验证集有50000张，并提供了加载和显示图像的示例代码。
🎯
关键要点
- 本文介绍了如何使用ImageNet数据集。
- ImageNet()函数的第一个参数是根目录，类型为字符串或pathlib.Path。
- 第二个参数是数据分割，默认为'train'，可选值为'train'（1281167张图像）或'val'（50000张图像）。
- 用户需手动下载数据集并将其放入指定目录。
- transform和target_transform参数是可选的，必须使用。
- loader参数也是可选的，默认为torchvision.datasets.folder.default_loader。
- 训练集包含1281167张图像，验证集包含50000张图像。
- 提供了加载和显示图像的示例代码。
- 数据集中包含1000个类别的标签，分别对应训练和验证图像的索引。
➡️
