使用带有csv标签的图像在Keras中进行回归

Question

我有一个道路图像数据集和一个csv，用于带有要尝试预测的连续值（转向角）的标签。目录如下所示：

  /project1
        /images
              /img1.jpg
              /img2.jpg
              ...
        /labels.csv

labels.csv看起来像

img   angle

img1   0.018293
img2  -1.265023
...

我在Keras上的所有经验都是关于分类问题的，我只在ImageDataGenerator类上使用flow_from_directory

例如

train_datagen = ImageDataGenerator(...)
train_generator = train_datagen.flow_from_directory(...)

我意识到对于回归问题这是不可能的，因为我们不能仅仅将图像放入与它们各自类别相关的特定文件夹中。

为像这样的回归问题创建生成器的最有效方法是什么？我是否不知道某种Keras方法？

Answer 1

如today所述，您想将Sequence子类化以提供给model.fit_generator()。碰巧的是，我写了一个小小的工作示例，说明如何使用.csv文件中的标签进行操作。您可以找到它here。