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

时间:2016-04-22 07:39:03

标签: deep-learning convolution vgg-net

VGG模型接受3通道RGB图像作为输入,但我的数据是单灰度图像,有关如何在VGG模型的第一个转换层中使用权重的任何建议吗?

1 个答案:

答案 0 :(得分:1)

因此,第一层中的过滤器大小为wxhx 3(对于VGG,w = h = 3),我建议在最后一个维度上取平均值并将它们用作wxhx 1.当你&#39可能已经想通了,你不需要改变任何其他权重。

这源于人们已经应用于在流图像上初始化VGG Imagenet模型的类似方法。

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