Python keras ImageDataGenerator.flow与颜色_模式

我将python与keras一起使用，希望使用keras图像预处理，我的一些图像是rgb，而一些是灰度，我需要读取所有图像，并将灰度图像的尺寸更改为x、x、3或进行处理，并且希望将其作为.flow函数的一部分，就像我可以使用颜色模式和来自目录的.flow一样，可以设置为rgb并将所有图像作为rgb读取，即使它们是灰度图像，

---

这可能吗？

您不能将其作为

.flow

的一部分，因为这假设您已经将准备好的图像加载到4D张量中。您可以使用

.flow\u from\u directory

使用的函数，该函数实际使用

color\u mode

参数：

img = load_img(os.path.join(self.directory, fname),
               color_mode=self.color_mode,
               target_size=self.target_size,
               interpolation=self.interpolation)

---

这是来自代码。您可以使用此功能加载图像，然后调用

.flow

是否加载正确表示为rgb 3d阵列的灰度图像？是的，应该加载。