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

来源:互联网 发布:四知文言文阅读答案 编辑:程序博客网 时间:2024/06/07 16:15
  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

3 0
原创粉丝点击