原作者:安逸轩
reference:http://andyjin.applinzi.com/?p=291
享受生活,感受阳光
CNN中哪一层的输出作为最后提取到的特征?
在使用CNN提取特征时,到底使用哪一层的输出作为最后的特征呢?
很多人可能会说:“当然是最后一个全连接层了!这有什么问题?”
我相信很多人跟我的想法一样:最后一个全连接层的输出连接最后的分类器,那么最后一个全连接层的输出当然是最后提取的特征了。
实则不是。正确的答案是倒数第二个全连接层的输出才是最后我们要提取的特征,也就是最后一个全连接层的输入才是我们需要的特征。
原因是:最后一个全连接层的输出维度,在设计时是和训练样本的类别数一致的。比如你的训练样本有2622类,那么在设计最后的分类器时要有2622个输入,则最后一个全连接层的输出也是2622维的。这样,最后一个全连接层的输出维度就和训练样本有了密切的关系,因此把它作为最后的特征显然不合适。