mmcv Registry类

通过 Registry 类,用户可以通过字符串方式实例化任何想要的模块。

在 MMCV 中所有的类实例化都是通过 build_from_cfg 函数实现,做的事情非常简单,就是给定 module_name,然后从 self._module_dict 提取即可。

CONVERTERS = Registry('converter')

@CONVERTERS.register_module()
class Converter1(object):
    def __init__(self, a, b):
        self.a = a
        self.b = b

converter_cfg = dict(type='Converter1', a=a_value, b=b_value)
converter = build_from_cfg(converter_cfg,CONVERTERS)

 

posted @ 2022-09-19 14:41  Tomorrow1126  阅读(114)  评论(0)    收藏  举报