卷积神经网络的分类层-softmax和sigmoid
1. softmax
在学习一些卷积网络的模型时,我们经常看到模型最后有一个softmax层,这个层的主要作用就是输出物体是各个类的概率,接下来我们就来看一下其计算公式:
如上图所示,在全连接层得到了两个结果(有两个类别)与,为了公式方便书写我们就把它们看成是和。我们想要得到的是各个类别的概率。这里需要注意在softmax层所有输出的概率之和为1。因此我们就要用到一个计算公式:
也就得到
2. sigmoid
sigmoid输出,每个节点之间互不相干。
计算公式:
sigmoid函数主要针对二分类问题,它可以不满足概率分布。当输入一张图片让你判断是否为猫的时候,那么结果只有两种,这是满足概率分布的;如果输入一张图片,让你判断是动物还是猫,那么就不满足概率分布了。