如何解决图像数据生成器方法 flow_from_directory 中的颜色模式
我有一个关于 ImageDataGenerator 的 flow_from_directory 方法的问题。我的图像采用 RGB 形式,我正在使用转移学习,我想在灰度图像上训练模型,但预训练模型需要 3 个通道。 flow_from_directory 颜色模式中有一种方法,它采用灰度或 RBG 。我的问题是,如果我在这里写灰度,Keras 是自动将 RGB 图像转换为灰度还是别的什么?
IMAGE_SIZE=[224,224]
resnet = InceptionResNetV2(input_shape=IMAGE_SIZE + [3],weights='imagenet',include_top=False)
Flow_from_dir 方法:
test_set = test_datagen.flow_from_directory(valid_path,target_size = (224,224),batch_size = 32,class_mode = 'categorical',color_mode='grayscale')
解决方法
如果您看到文档,那就是:
color_mode:“灰度”、“rgb”、“rgba”之一。默认值:“RGB”。是否将图像转换为具有 1、3 或 4 个通道。
所以,是的,结果是灰度。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。