当前位置: 代码迷 >> python >> 具有color_mode的keras ImageDataGenerator.flow
  详细解决方案

具有color_mode的keras ImageDataGenerator.flow

热度:83   发布时间:2023-07-14 08:43:33.0

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

您不能将其作为.flow一部分进行.flow因为这假定您已经将准备好的图像加载到4D张量中。 您可以使用其功能.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

  相关解决方案