Keras ImageDataGenerator方法flow_from_directory

Question

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

Answer 1

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

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