如何在微调FC层中选择感知器的数量?

时间:2018-06-19 07:17:06

标签: python tensorflow

我使用VGG-16预训练模型并微调最后3个FC层。 但在我的情况下,我只使用3个类作为我的分类。 我想问如何选择FC图层的感知器我应该可视化Conv5_3图层,然后做出决定吗? BTW,VGG-16官方型号为FC层的4096,4096,1000感知器。

1 个答案:

答案 0 :(得分:0)

调整网络的最简单方法是在VGG之上添加另一个FC层(权重内核大小为1000x3)。

或者,将最后一个FC层(大小为4096x1000)替换为大小为4096x3的FC层。

不要忘记正确初始化新添加的图层。