mmdetection中Faster RCNN默认使用4层FPN

最新推荐文章于 2024-01-09 21:26:28 发布

Tahy

最新推荐文章于 2024-01-09 21:26:28 发布

阅读量1k

点赞数

分类专栏： mmdetection

本文链接：https://blog.csdn.net/a404NotFound/article/details/108805159

版权

mmdetection 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

mmdetection中Faster RCNN默认使用4层FPN，即P2~P5。根据配置文件，还会通过最大池化生成P6，但是并不使用P6层。
依据：mmdetection/mmdet/models/roi_heads/standard_roi_head.py文件中_bbox_forward函数bbox_feats = self.bbox_roi_extractor(x[:self.bbox_roi_extractor.num_inputs], rois)
其中的self.bbox_roi_extractor.num_inputs指的是：
被mmdetection/mmdet/models/roi_heads/roi_extractors/single_level_roi_extractor.py中SingleRoIExtractor类继承的BaseRoIExtractor(同目录下base_roi_extractor.py)类中的num_inputs函数，即len(self.featmap_strides)。
对应到配置文件，则根据model-roi_head-bbox_roi_extractor-featmap_strides=[4, 8, 16, 32]来确定。