具有color_mode的keras ImageDataGenerator.flow

Question

我正在将python与keras一起使用，并希望使用keras图像预处理，我的一些图像是rgb而有些是灰度的，我需要阅读所有图像并将灰度图像的尺寸更改为x，x，3或进行处理它们，并希望将其作为.flow函数的一部分，就像我可以将color_mode与.flow_from_directory一起使用一样，可以将其设置为rgb并将所有图像读取为rgb，即使它们是灰度的，这可能吗？

Answer 1

您不能将其作为.flow一部分进行.flow因为这假定您已经将准备好的图像加载到4D张量中。 您可以使用load_img其功能.flow_from_directory使用实际消耗color_mode参数：

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

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