caffe digits环境下finetune时prototxt文件修改方法

1、删除两个data层中的多余参数,剩余如下形式:

layer {
  name: "data"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    mirror: true
    crop_size: 227
  }
}
layer {
  name: "data"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  transform_param {
    mirror: false
    crop_size: 227
  }
}

2、将所有include {phase: TRAIN}替换成include {stage: "train"}
将所有include {phase: TEST}替换成include {stage: "val"}
在loss层添加exclude {stage: "deploy"}

3、将fc8层的名字替换成fc8-re(只要换一个名字即可,因为分类数量与pretrained model数量不同时会初始化错误),并删除fc8中的num_output

4、在文件最后添加softmax层

layer {
  name: "prob"
  type: "Softmax"
  bottom: "fc8"
  top: "prob"
  include{
      stage:"deploy"
      }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值