caffe网络配置train_val.prototxt转换成digits里的网络步骤

1.去掉Data层中的transform_param中的mean_file和backend,会自动读取之前数据集里生成的,mirror和crop_size不受影响,需要就填,不需要就不填
2.改  include {
    phase: TRAIN
  }
为 include { stage: "train" } ,同理TEST改成val,其实这个不改好像也可以,具体可以查看自定义网络旁边的问号帮助
3.修改最后一个全连接层的numoutput(这个其实不算吧,具体根据自己的填,只是提醒下别忘了)
4.最后一个全连接层后的网络替换为(ip2为最后一个全连接层),其实就是把之前的网络里的train_val和deploy给合并了
layer {
  name: "accuracy"
  type: "Accuracy"
  bottom: "ip2"
  bottom: "label"
  top: "accuracy"
  include { stage: "val" }
}
layer {
  name: "loss"
  type: "SoftmaxWithLoss"
  bottom: "ip2"
  bottom: "label"
  top: "loss"
  exclude { stage: "deploy" }
}
layer {
  name: "softmax"
  type: "Softmax"
  bottom: "ip2"
  top: "softmax"
  include { stage: "deploy" }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值