我們可以將在RGB圖像上訓練的模型的權重用於灰度圖像嗎？

Question

實際上，我正在嘗試使用從Image Net上的預訓練初始模型到灰度數據集的轉移學習，所以我只想知道我可以使用相同的權重，還是應該在使用權重之前先進行一些更改。

提前致謝。

Answer 1

有一些可用的選項：

1. 使用標准的OpenCV功能將灰度圖像轉換為彩色圖像。
2. 在前面添加一個深度為3的卷積層作為輸出。 按照正常進行。
3. 從現有網絡的第一層中選擇一組權重。
4. 對現有網絡第一層的權重取平均值。

每個選項分別具有不同的權衡：

1. 這可能是行之有效的，但是您要付出額外計算的代價，在這種情況下，這可能不算什么。
2. 您可能實際上是在學習嘗試映射到RGB的濾鏡，因此這可能類似於1。
3. 這是一個hack，可能無法正常運行。
4. 與3相同