Faster R-CNN是如何添加ROIPoolingLayer和SmoothL1LossLayer的?

  1. ROIPoolingLayer和SmoothL1LossLayer类的声明可以在caffe-fast-rcnn/include/caffe下的fast_rcnn_layers.hpp文件中找到。

  2. ROIPoolingLayer和SmoothL1LossLayer的实现在caffe-fast-rcnn/src/caffe/layers下的4个文件中

    • roi_pooling_layer.cpp
    • roi_pooling_layer.cu
    • smooth_L1_loss_layer.cpp
    • smooth_L1_loss_layer.cu
  3. ROIPoolingLayer和SmoothL1LossLayer的测试在caffe-fast-rcnn/src/caffe/test下的2个文件中

    • test_roi_pooling_layer.cpp
    • test_smooth_L1_loss_layer.cpp
  4. 在caffe-fast-rcnn/src/caffe/proto/caffe.proto中添加ROIPoolingParameter和SmoothL1LossParameter,让ProtoBuf支持ROIPoolingLayer和SmoothL1LossLayer的识别。

  5. 不需要在在layer_factory.cpp添加代码。


参考阅读:
how to develop new layers, https://github.com/BVLC/caffe/issues/684

评论 19
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值