MMDetection2.17-自定义组件时注册表(Registry)分析理解--以如何增加注意力模块为例

Posted 2022-01-17 会飞的毛毛虫_

注册表是用来干什么的？

MMDection的所有功能都是基于注册器来实现的。注册器的使用目的是为了在算法训练、调参中通过直接更改配置文件(超参数，如学习率等)，完成对整个网络的修改。注册器在这期间可以帮助我们完成字符串到类的映射，自动解析config中的内容。其实说白了注册器只是通过一个类完成了string类型到类名的映射。
mmdetection的组件大多数以类的形式定义，下面是在mmdet\\models\\builder.py中实例化的注册器:

# Copyright (c) OpenMMLab. All rights reserved.
import warnings
from mmcv.cnn import MODELS as MMCV_MODELS
from mmcv.utils import Registry
MODELS = Registry(\'models\',