首页 文章

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

提问于
浏览
0

我有一个道路图像数据集和一个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方法?

1 回答

  • 0

    正如 today 已经提到的那样,你想要将 Sequence 子类提供给 model.fit_generator() . 碰巧的是,我写了一个小的工作示例,说明如何使用.csv文件中的标签 . 你可以找到它here .

相关问题