yacs库,用于为一个系统构建config文件
Install
$ pip install yacs
Import
from yacs.config import CfgNode as CN
Usage
create config node
需要创建CN()这个作为容器来装载我们的参数,这个容器可以嵌套
from yacs.config import CfgNode as CN
__C = CN()
__C.name = 'test'
__C.model = CN() # 嵌套使用
__C.model.backbone = 'resnet'
__C.model.depth = 18
print(__C)
‘’’
name: test
model:
backbone: resnet
depth: 18
‘’’
API reference
use __C as created config file
def get_cfg_defaults():
return __C.clone()
print(__C.clear()) # None
__C.merge_from_file("./test_config.yaml")
Addition:
你需要merge的yaml文件中,不能有default参数中不存在的参数,不然会报错,但是可以比default中设定的参数少,比如default文件中有name参数,这是不需要特定改动的,你可以在yaml中不设置name这个key。
from yacs.config import CfgNode as CN
__C = CN()
__C.name = 'test'
__C.model = CN()
__C.model.backbone = 'resnet'
__C.model.depth = 18
name: test
model:
backbone: resnet
depth: 18
model:
depth: 34
name: test
model:
backbone: resnet
depth: 29
batch_normalization: True
na_me: test
4. merge_from_list()
可以用list来传递参数
from yacs.config import CfgNode as CN
__C = CN()
__C.name = 'test'
__C.model = CN()
__C.model.backbone = 'resnet'
__C.model.depth = 18
print(__C)
‘’’
model:
backbone: resnet
depth: 18
name: test
‘’’
opts = ["name", 'test_name', "model.backbone", "vgg"]
__C.merge_from_list(opts)
print(__C)
‘’’
model:
backbone: vgg
depth: 18
name: test_name
‘’’
other details are the same as merge_from_file
merge_from_other_cfg()
the same as merge_from_file and merge_from_list, the only difference is that the merged file is also a CfgNode class
freeze()
freeze the configs, and you can not change the value after this operation
from yacs.config import CfgNode as CN
__C = CN()
__C.name = 'test'
__C.model = CN()
__C.model.backbone = 'resnet'
__C.model.depth = 18
__C.freeze()
__C.name = 'test2' # error
7. defrost()
reverse operation of freeze()
from yacs.config import CfgNode as CN
__C = CN()
__C.name = 'test'
__C.model = CN()
__C.model.backbone = 'resnet'
__C.model.depth = 18
__C.freeze()
__C.name = 'test2' # error
__C.defrost() # not freeze cfgs, after this operation you can change the value
__C.name = 'test2' # work