首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何将RGB图像转换为灰度图像,扩展灰度图像的维数以在InceptionV3中使用?

如何将RGB图像转换为灰度图像,扩展灰度图像的维数以在InceptionV3中使用?
EN

Stack Overflow用户
提问于 2020-12-07 10:17:06
回答 2查看 6K关注 0票数 1

我正在训练一个Keras模型,我有RGB格式的训练图像。我想训练我的模型,但在InceptionV3上的灰度图像,但它以RGB图像作为输入。我的问题是:如何才能首先将RGB图像转换为灰度,然后在三维中复制?我正在使用Keras的TrainDataGenerator.flow_from_directory方法。

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2020-12-07 13:43:19

ImageDataGenerator中,您可以传递一个预处理函数。使用函数tf.image.rgb_to_grayscaletf.image.grayscale_to_rgb执行转换:

代码语言:javascript
复制
def to_grayscale_then_rgb(image):
    image = tf.image.rgb_to_grayscale(image)
    image = tf.image.grayscale_to_rgb(image)
    return image
代码语言:javascript
复制
tf.keras.preprocessing.image.ImageDataGenerator(
    rescale=1/255,
    preprocessing_function=to_grayscale_then_rgb
)
票数 1
EN

Stack Overflow用户

发布于 2021-05-07 12:45:10

我认为最简单的方法是使用color_mode ="grayscale“参数的TrainDataGenerator.flow_from_directory方法。

下面是指定如何执行此https://keras.io/api/preprocessing/image/的keras文档

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/65179822

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档