用於 CT 圖像預訓練權重的 SegNet

Question

我正在嘗試訓練一個用於 ct 圖像分割任務的 SegNet（使用 Keras TF）。 我正在使用 VGG16 預訓練權重，但我在第一個卷積層遇到問題，因為我使用的是灰度圖像，但 VGG 是在 rgb 圖像上訓練的。 我使用第二種方法解決了這個問題（不能使用第一種方法，因為需要太多內存）。 然而它並沒有幫助我，價值觀真的很糟糕（訓練了 100 個時期）。

我應該從頭開始訓練第一個卷積層嗎？

Answer 1

您可以嘗試在 vgg 之前添加一個 Conv2D。 就像是：

> Your Input(shape=(height,width,1))

Conv2D(filters=3,kernel_size=1, padding='same',activation='relu')

> The VGG pretrained network (input = (height,width,3))

在您的情況下很有趣，因為通常使用 1x1 卷積來更改 object 的深度。