caffe中五种层的实现与参数配置(4)------softmax层

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/w_q_q2017/article/details/79183037

Softmax回归模型是logistic回归模型在多分类问题上的推广,在多分类问题中,待分类的类别数量大于2,且类别之间互斥。

Softmax公式:


通常情况下softmax会被用在网络中的最后一层,用来进行最后的分类和归一化。

通常有2种结构:

第一种

#可以计算给出每个样本对应的损失函数值
layer {
  name: "loss"
  type: "SoftmaxWithLoss"
  bottom: "ip2"
  bottom: "label"
  top: "loss"
}
第二种

#输出为每个类别的概率值
layers {
  name: "prob"
  type: “Softmax"
bottom: " ip2"
  top: "prob"
}



猜你喜欢

转载自blog.csdn.net/w_q_q2017/article/details/79183037