使用 csv（标签）为 CNN 加载图像？答案

【问题标题】：Load image using csv (labels) for CNN?使用 csv（标签）为 CNN 加载图像？
【发布时间】：2020-10-14 06:43:05
【问题描述】：

我想知道如何使用 pandas 和 Keras Image 数据流加载图像。目前，我的 csv 文件包含

    ID               Category

0   1.png               1
1   2.png               2
2   3.png               1
3   4.png               2
4   5.png               2

我的文件目录是这样的

 C:.
|   dataset.csv
|
+---test
|       10.png
|
\---train
    \---train
        +---01
        |       1.png
        |       3.png
        |
        \---02
                2.png
                4.png
                5.png

但是，从我发现的不同 StackOverflow 解决方案中，例如 this，他们的火车数据集没有分类到他们的相应类别文件夹中，因此能够flow_from_dataframe。

但就我而言，我的图片被进一步分类为不同的与我不确定的 csv 'Category' 对应的文件夹能够使用 csv 加载图像。

有解决办法吗？谢谢。

【问题讨论】：

你在 csv 输出中说 asddfasdsas.jpg, 123123wqlwd.jpg。但在第二张图中，你说1.png、2.png。对吗？
@thushv89 我已经更新了

标签： python pandas keras

【解决方案1】：

使用您的文件目录结构，您不需要 CSV 文件来识别您的图像标签，实际上这是 TensorFlow（和大多数神经网络框架）期望您拥有的文件夹。 flow_from_dataframe 方法 is designed to resolve the problem without an expected directory structure。要加载数据集，只需使用 flow_from_directory 方法。您可以阅读TensorFlow Official Guide 和TensorFlow Document 了解更多示例。

【讨论】：