我有一个道路图像数据集和一个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 回答
正如
today
已经提到的那样,你想要将Sequence
子类提供给model.fit_generator()
. 碰巧的是,我写了一个小的工作示例,说明如何使用.csv文件中的标签 . 你可以找到它here .