python - 带有 color_mode 的 keras ImageDataGenerator.flow

Question

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

Answer 1

您不能将其作为 .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。