yacs直接读取yaml文档(python)

yacs在我理解是一种读写配置文件的python包。在机器学习领域，很多模型需要设置超参数，当超参数过多时，不方便管理，于是出现了很多类似yaml，yacs的包。

关于yacs的使用，网上有一些文档，类似这篇。这些方法都需要先加载定义了一些CfgNode的文档，然后通过merge_from_file函数加载yaml. 本文要介绍的是如何直接加载yaml文件。

如有文件test.yaml，内容如下：

DATASETS:
TRAIN: ["voc_2007_train", "coco_2017_train",]
TEST: ["coco_2017_test",]
DATALOADER:
NUM_WORKERS: 1
SOLVER:
BASE_LR: 0.01
WEIGHT_DECAY: 0.0001
STEPS: [120000, 160000]
MAX_ITER: 180000
IMS_PER_BATCH: 1

通过以下代码可以加载test.yaml，且引用配置文件里变量的方法不是字典方式，而是通过符号“.”级联

from yacs.config import CfgNode as CN
yaml_name='./test.yaml'
fcfg = open(yaml_name)
cfg = CN.load_cfg(fcfg)
cfg.freeze()
print(cfg.DATASETS.TEST)#输出：["coco_2017_test"]

最后

以上就是鲤鱼手套为你收集整理的yacs直接读取yaml文档(python)的全部内容，希望文章能够帮你解决yacs直接读取yaml文档(python)所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。