如何使用单通道输入微调 VGG 模型?

how to fine-tune VGG model with single channel input?

VGG 模型接受 3 通道 RGB 图像作为输入,但我的数据是单个灰度图像,对于如何利用 VGG 模型第一个 conv 层中的权重有什么建议吗?

所以你在第一层的过滤器大小为 w x h x 3(对于 VGG,w=h=3),我建议取最后一个维度的平均值并将它们用作 w x h x 1。正如你可能想通了,您不需要更改任何其他权重。

这受到人们应用于在流图像上初始化 VGG Imagenet 模型的类似方法的启发。

https://arxiv.org/pdf/1507.02159v1.pdf(第 2.2 节)