代码之家 › 专栏 › 技术社区 › thebeancounter

凯拉斯ImageDataGenerator.flow文件带颜色模式

deep-learning keras image-processing image python

1

thebeancounter · 技术社区 · 7 年前

我正在使用python和keras,并且希望使用keras图像预处理,我的一些图像是rgb,而一些是灰度的,我需要读取所有这些图像,并将灰度图像的尺寸更改为x,x,3或进行处理,作为.flow函数的一部分,我可以使用color\u mode和目录中的.flow\u,将其设置为rgb,并以rgb格式读取所有图像,即使它们是灰度图像, 有可能吗?

1 回复 | 直到 7 年前

1

2

nuric 7 年前

你不能把它当作你的一部分 .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 代码。您可以使用此函数加载图像,然后调用 .流量