deep-learning - 如何使用单通道输入微调 VGG 模型？

Question

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

Answer 1

所以你在第一层的过滤器的大小是 wxhx 3（对于 VGG，w=h=3），我建议取最后一个维度的平均值并将它们用作 wxhx 1。你可能已经想通了，你不需要更改任何其他权重。

这是从人们在流图像上初始化 VGG Imagenet 模型所应用的类似方法中得到启发的。

https://arxiv.org/pdf/1507.02159v1.pdf（第 2.2 节）