pytorch中如何简单的加载imagenet这种数据集

pytorch简单的加载imagenet这种数据集
图(1)是imagenet的的一级数据目录
pytorch简单的加载imagenet这种数据集
图(2)是二级目录

直接调用pytorch中的库方法加载代码如下:
pytorch简单的加载imagenet这种数据集
pytorch简单的加载imagenet这种数据集

它是安装你目录打的label,比如第一个二级目录下的图片
pytorch简单的加载imagenet这种数据集
图(3)是三级目录

pytorch简单的加载imagenet这种数据集
图(4):打开其中一张,全部都是????累
全部打为0,依次类推

加载完毕,所有的ImageFolder都可以用,不用再特地的自己打label了,非常简单快捷。

相关文章: