【caffe c++】网络配置文件说明之全连接层、softmax层

最新推荐文章于 2021-05-31 21:25:17 发布

黎曼不会算积分

最新推荐文章于 2021-05-31 21:25:17 发布

阅读量280

点赞数

分类专栏： Caffe 文章标签： c++ caffe

本文链接：https://blog.csdn.net/Feeryman_Lee/article/details/105233552

版权

Caffe 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

下面代码中没注释的部分看这篇：

【caffe c++】网络配置文件说明之卷积层、池化层

#全连接层，输出的是一个简单向量  参数跟卷积层一样
layer {
  name:"ip1"
  type:"InnerProduct"
  bottom:"pool2"
  top:"ip1"
  param{		#学习率
    lr_mult:1
    }
  param {
    lr_mult:2
    }
  inner_product_param {
    num_output:500		#输出向量的大小
    weight_filler{
        type:"xavier"
       }
    bias_filler{
        type:"constant"
        }
    }
 }

###测试的时候输入准确率
layer{
  name:"accuracy"
  type:"Accuracy"
  bottom:"ip2"
  bottom:"label"
  top:"accuracy"
  include{
    phase:TEST
    }  
}

#softmax-loss layer:输出loss值
layer{
  name:"loss"
  type:"SoftmaxWithLoss"
  bottom:"ip1"
  bottom:"label"
  top:"loss"
}

#softmax layer:输出似然值（准确率）
layer{
  bottom:"cls3_fc"
  top:"prob"
  name:"prob"
  type:"Softmax"
}