如何将图像数据集拆分为训练集和测试集？

Question

我正在研究加州理工学院的256_ObjectCategories数据集。他们在不同的文件夹中组织了256个类别的所有图像。我使用Keras的ImageDataGenerator来加载数据集，但是我无法将其拆分为训练和使用相同的测试。如何在没有移动图像或更改目录的终端中执行此操作？任何帮助表示赞赏。谢谢。 :)

Answer 1

现在似乎没有ImageDataGenerator的开箱即用。请参阅此主题：https://github.com/fchollet/keras/issues/5862

用户AloshkaD建议您使用glob：rasterList = glob.glob(os.path.join(path_of_your_image_directory, '*.jpg'))创建索引列表，以编程方式拆分并将该列表的验证部分提供给fit_generator（）的validation_data参数。