Faster R-CNN中ZF Net更换成CaffeNet

  1. 对于ZF中的train.prototxt开头的input-data、结尾的loss_ls和loss_box不能改变,这个是Faster RCNN代码框架决定的。

    layer {
    name: 'input-data'
    type: 'Python'
    top: 'data'
    top: 'im_info'
    top: 'gt_boxes'
    python_param {
      module: 'roi_data_layer.layer'
      layer: 'RoIDataLayer'
      param_str: "'num_classes': 21"
      }
    }
    
    layer {
    name: "loss_cls"
    type: "SoftmaxWithLoss"
    bottom: "cls_score"
    bottom: "labels"
    propagate_down: 1
    propagate_down: 0
    top: "cls_loss"
    loss_weight: 1
    loss_param {
    ignore_label: -1
    normalize: true
    }
    }
    layer {
    name: "loss_bbox"
    type: "SmoothL1Loss"
    bottom: "bbox_pred"
    bottom: "bbox_targets"
    bottom: 'bbox_inside_weights'
    bottom: 'bbox_outside_weights'
    top: "bbox_loss"
    loss_weight: 1
    }
  2. 将CaffeNet的train.prototxt中的Conv1-Conv5,RCNN对应的部分直接复制。

  3. train.prototxt参考第2步即可。

若有疑惑可以给我留言。

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值