是否可以用 2D 灰度图像提供预训练的 Inception model（tensorflow 2.0/Keras）？

Question

根据 Keras 2.0 文档，关于可以输入到预训练初始 model 的图像的输入形状：

input_shape：可选的形状元组，仅在 include_top 为 False 时指定（否则输入形状必须为 (299, 299, 3)（使用 'channels_last' 数据格式）或 (3, 299, 299)（使用 'channels_first'数据格式）。它应该正好有 3 个输入通道，并且宽度和高度应该不小于 75。例如 (150, 150, 3) 将是一个有效值。

但是，我正在处理 2D 的灰度图像。 我应该如何处理这种情况？

Answer 1

对于伪RGB图像，您可以将灰度图像复制3次

import numpy as np
# img=np.zeros((224,224))

如果您的图像的形状长度为 2，则只有宽度和高度，您首先需要添加一个额外的尺寸：

img = np.expand_dims(img,-1)

您将最后一个维度重复 3 次：

img = np.repeat(img,3,2)
print(img.shape)
# (224,224,3)