将train_val.prototxt 转换成deploy.prototxt

1.删除输入数据(如:type:data...inckude{phase: TRAIN}),然后添加一个数据维度描述。

input: "data" 
input_dim: 1 
input_dim: 3 
input_dim: 224 
input_dim: 224
force_backward: true

 

2.移除最后的“loss” 和“accuracy” 层,加入“prob”层。

layers {
  name: "prob"
  type: SOFTMAX
  bottom: "fc8"
  top: "prob"
}

如果train_val文件中还有其他的预处理层,就稍微复杂点。如下,在'data'层,在‘data’层和‘conv1’层 (with  bottom:”data” / top:”conv1″ ). 插入一个层来计算输入数据的均值。

layer {
name: “mean”
type: “Convolution”
<strong>bottom: “data”
top: “data”</strong>
param {
lr_mult: 0
decay_mult: 0
}

…}

在deploy.prototxt文件中,“mean” 层必须保留,只是容器改变,相应的‘conv1’也要改变 ( bottom:”mean”/ top:”conv1″ )

layer {
name: “mean”
type: “Convolution”
<strong>bottom: “data”
top: “mean“</strong>
param {
lr_mult: 0
decay_mult: 0
}

…}


  • 6
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值