Keras ImageDataGenerator方法flow_from_directory

Question

我有一个关于ImageDataGenerator的flow_from_directory方法的问题。我的标签是2D图像，因此我需要为图像和蒙版生成相同变换的增强数据。 阅读文档https://keras.io/preprocessing/image/看起来像flow_from_directory只接受几种格式的图像（PNG，JPG或BMP）。那么.mhd图像或直接numpy数组呢？我可以将它们送到这个模块吗？ 感谢

Answer 1

flow_from_directory将使用PIL（see here）读取图像，因此Keras可以读取PIL支持的所有格式。

但是，我没有看到任何同时读取掩码和图像的方法，并执行相同的转换。我担心你必须创建自己的ImageDataGenerator。如果从当前代码开始，这非常简单。