【mmdetection系列】mmdetection之neck讲解

目录

1.configs

2.具体实现

3.如何调用的？

3.1 注册

3.2 调用

---

类似于backbone部分。

配置部分在configs/_base_/models目录下，具体实现在mmdet/models/necks目录下。

1.configs

我们看下yolox的neck吧。https://github.com/open-mmlab/mmdetection/blob/master/configs/yolox/yolox_s_8x8_300e_coco.py

neck=dict(
        type='YOLOXPAFPN',
        in_channels=[128, 256, 512],
        out_channels=128,
        num_csp_blocks=1)

2.具体实现

neck作用：一般是收集到不同阶段的特征图的网络层，并将其传递到预测层。

我们可以找到YOLOXPAFPN类的具体实现：