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 @   Tomorrow1126  阅读(94)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· SQL Server 2025 AI相关能力初探
· 单线程的Redis速度为什么快?
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
历史上的今天:
2020-09-19 c++科学计数法 、long long的范围
2020-09-19 最大序列和
2020-09-19 N阶上楼梯问题——动态规划(递推求解)
2020-09-19 动态规划题型总结
2020-09-19 hdu1232畅通工程——并查集
点击右上角即可分享
微信分享提示