计算机视觉系列-MMCV 注册组件分析: Registry

计算机视觉系列-MMCV 注册组件分析: Registry

MMCV Registry

Registry 用法

在 OpenMMLab 中,通过 Registry 类,以字符串方式实例化模块, 管理构建不同的组件backbones、head 和 necks 等,Registry 类内部维护一个全局 key-value 对。
Registry 类的好处:

  • 解耦性强
  • 可扩展性强
  • 代码更易理解

例如 Faster R-CNN 的 backbone 模块实例化 配置:

   backbone=dict(
        type=

猜你喜欢

转载自blog.csdn.net/duan_zhihua/article/details/125044896