时间戳

首页 > 解决方案 > 如何将 ImageFolder 与训练集和测试集的图像列表一起使用？

python-3.x - 如何将 ImageFolder 与训练集和测试集的图像列表一起使用？

问题描述

我正在研究斯坦福狗数据集，该数据集提供 120 类图像和 12000 个用于训练的图像名称列表和大约 8000 多个用于测试的名称。

如何使用 ImageFolder 根据图像名称加载训练集和测试集？下面是列表的样子。

动车组：

       [array(['n02085620-Chihuahua/n02085620_4441.jpg'], dtype='<U38')],
       [array(['n02085620-Chihuahua/n02085620_1502.jpg'], dtype='<U38')],
       ...,
       [array(['n02116738-African_hunting_dog/n02116738_6754.jpg'], dtype='<U48')],
       [array(['n02116738-African_hunting_dog/n02116738_9333.jpg'], dtype='<U48')],
       [array(['n02116738-African_hunting_dog/n02116738_2503.jpg'], dtype='<U48')]],
      dtype=object)

测试集：

array([[array(['n02085620-Chihuahua/n02085620_2650.jpg'], dtype='<U38')],
       [array(['n02085620-Chihuahua/n02085620_4919.jpg'], dtype='<U38')],
       [array(['n02085620-Chihuahua/n02085620_1765.jpg'], dtype='<U38')],
       ...,
       [array(['n02116738-African_hunting_dog/n02116738_3635.jpg'], dtype='<U48')],
       [array(['n02116738-African_hunting_dog/n02116738_2988.jpg'], dtype='<U48')],
       [array(['n02116738-African_hunting_dog/n02116738_6330.jpg'], dtype='<U48')]],
      dtype=object)

标签： python-3.xpytorchdataset

解决方案

推荐阅读