是否可以用 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)