我想训练一个用于图像抠像的模型。训练图像和测试集图像大小相同。我想从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?