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

注册表是用来干什么的?

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',

你可能感兴趣的:(视觉检测图像分割干货,深度学习,目标检测,计算机视觉,mmdetection)