我們可以使用RGB圖像訓練的模型的灰度權重來進行灰度圖像嗎？

Question

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

在此先感謝。

Answer 1

有可用的幾個選項：

1. 使用標準OpenCV函數轉換灰度圖像彩色圖像。
2. 在前面添加一個卷積層，其深度爲3作爲輸出。按照正常進行。
3. 從現有網絡的第一層中選擇一組權重。
4. 平均來自現有網絡第一層的權重。

每個選項分別有不同的取捨：

1. 這大概是什麼工作好，但你付出額外的計算的價格，這可能不是很多在這種情況下。
2. 您可能本質上是試圖映射到RGB的學習過濾器，所以這可能類似於1.
3. 這是一種黑客攻擊，可能無法像您期望的那樣工作。
4. 同3.