当前位置: 代码迷 >> 综合 >> caffe测试--如何得到deploy.prototxt?
  详细解决方案

caffe测试--如何得到deploy.prototxt?

热度:89   发布时间:2023-12-14 15:22:01.0

 

步骤:

 

  1. 复制train_val.prototxt中的所有层。
  2. 删除没有“bottom”字段的前几层(用于数据输入的data层 train test),并设置新的data层(不包含train、test,不指定类型,指定数据输入的格式input_dim)。图1
  3. 删除所有必须专门学习的参数。blobs_lr,weight_decay,weight_filler,bias_filler 图2
  4. 修改softmax层的类型,在训练网络时使用的是SoftmaxWithLoss返回loss,现在需要获得预测的结果,所以使用prob  图3

图1

图2

图3

参考:https://github.com/BVLC/caffe/issues/550

  相关解决方案