标签也是图像时,如何在Keras中加载图像?

时间:2019-11-16 06:20:55

标签: python keras deep-learning

我想训练一个用于图像抠像的模型。训练图像和测试集图像大小相同。我想从Keras中的目录加载图像。我的训练集目录如下:

├───image
│       0000_1.png
│       0000_2.png
│       ...
├───label
│       0000_1.png
│       0000_2.png
│       ...

测试集目录与此类似。所有输入图像都在一个文件夹中,所以我不能使用flow_from_directory。而且当使用flow_from_dataframe时,我无法将标签目录作为y_col参数给出。

如何将输入图像和相应的输出图像作为标签加载,以在Keras中训练我的cnn?

0 个答案:

没有答案