pytorch的ImageNet类

官方文档：IMAGENET
源码：SOURCE CODE FOR TORCHVISION.DATASETS.IMAGENET

这个类可以帮助我们更方便使用ImageNet数据集，只需要下载好，然后提供数据集的根目录即可。
根目录下面应该包含这些文件：

ILSVRC2012_img_train.tar或者train文件夹
ILSVRC2012_img_val.tar或者val文件夹
ILSVRC2012_devkit_t12.tar.gz或者meta.bin

没有的话会报错。

源码解读

继承了ImageFolder类，ImageFolder继承了DatasetFolder，用来处理同种类的图片放在同一个文件夹的数据集。有一个class_to_idx字典和classes列表，classes就是root下面所有文件夹的名字，class_to_idx就是从class到label的映射，每个class对应的label就是将所有class按照字母顺序排序之后该class所处的位置或者说index（从0开始）。