Python 当标签也是图像时，如何在Keras中加载图像？

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

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

测试集目录类似。所有输入图像都在一个文件夹中，因此我无法使用来自目录的flow_。当使用来自数据帧的flow_时，我不能将标签目录作为y_col参数

---

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

我认为您必须在Keras中编写自定义数据生成器，以读取目录并提供X和y对的批次-在您的情况下，y也是一个图像。（查找“keras自定义生成器”）这就是我需要的。谢谢问题解决了。