caffe基础-06softmax层的配置

最新推荐文章于 2019-03-07 20:25:50 发布

跬步千里OvO

最新推荐文章于 2019-03-07 20:25:50 发布

阅读量484

点赞数

CC 4.0 BY-SA版权

文章标签： softmax层

本文链接：https://blog.youkuaiyun.com/Terrenceyuu/article/details/76228187

caffe 专栏收录该内容

14 篇文章

订阅专栏

本文介绍如何在Caffe中配置Softmax层以进行分类任务。主要涉及两种配置方式：使用SoftmaxWithLoss层输出损失值，适用于训练阶段；使用Softmax层输出概率值，适用于测试阶段。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

caffe基础-06softmax层的配置

#softmax-loss layer：输出loss值 具体参见 #http://blog.youkuaiyun.com/supercally/article/details/54234115
layer {
  name: "loss"
  type: "SoftmaxWithLoss"
  bottom: "ip1"
  bottom: "label"
  top: "loss"
}

#softmax layer: 输出似然值
layers {
  bottom: "cls3_fc"
  top: "prob"
  name: "prob"
  type: “Softmax"
}