yacs

Detectron2 配置 | 十

白昼怎懂夜的黑 提交于 2020-08-09 08:49:44
作者|facebookresearch 编译|Flin 来源|Github 使用配置 Detectron2的配置系统使用yaml和yacs( https://github.com/rbgirshick/yacs )。 除了访问和更新配置的基本操作外,我们还提供以下额外功能: 配置中可以包含" BASE :base.yaml"字段,该字段将首先加载基本配置。如果存在任何冲突,则基本配置中的值将在子配置中被覆盖。我们为标准模型架构提供了一些基本配置。 我们提供配置版本控制,以实现向后兼容。如果你的配置文件使用诸如"VERSION:2"之类的配置行进行了版本控制,即使我们将来重命​​名某些密钥,detectron2仍会识别它。 使用配置 CfgNode对象的一些基本用法如下所示: from detectron2.config import get_cfg cfg = get_cfg()#获取detectron2的默认配置 cfg.xxx = yyy#为你自己的自定义组件添加新配置 cfg.merge_from_file("my_cfg.yaml")#从文件加载值 cfg.merge_from_list(["MODEL.WEIGHTS","weights.pth"])#也可以从str列表中加载值 print(cfg.dump())#打印格式化的配置