yaml文件查看模型的架构

本文主要是介绍yaml文件查看模型的架构，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

最近在看hrnet模型代码，想查看hrnet的模型架构，输出一下，但是模型参数需要cfg，我就想着怎么把yaml文件导进来然后打印模型呢，直接chat就可以了，下面解释一下每一部分，非常的好理解

yaml文件格式如下：

之后在hrnet文件夹下输入代码

首先加载文件路径

yaml_file_path = 'F:/code/DEKR-main/experiments/coco/w32/w32_4x_reg03_bs10_512_adam_lr1e-3_coco_x140.yaml'

 之后解析文件

yaml_file = Path(yaml_file_path)

下一步 判断文件是否存在如果存在就打开这个文件，并将这个文件定义为file，并用yaml加载这个文件，到这里会形成一个字典，你调用的话只能使用cfg[参数] 来进行访问，如果想要cfg.内容.内容的形式就需要使用OmegaConf创建这种访问方式。代码如下：

yaml_file_path = 'F:/code/DEKR-main/experiments/coco/w32/w32_4x_reg03_bs10_512_adam_lr1e-3_coco_x140.yaml'
yaml_file = Path(yaml_file_path)
if yaml_file.exists():with yaml_file.open('r') as file:cfg_dict = yaml.safe_load(file)cfg = OmegaConf.create(cfg_dict)print(cfg)pose = PoseHigherResolutionNet(cfg)print(pose)

完整代码如下：

import yaml
from omegaconf import OmegaConf
from pathlib import Path
yaml_file_path = 'F:/code/DEKR-main/experiments/coco/w32/w32_4x_reg03_bs10_512_adam_lr1e-3_coco_x140.yaml'
yaml_file = Path(yaml_file_path)
if yaml_file.exists():with yaml_file.open('r') as file:cfg_dict = yaml.safe_load(file)cfg = OmegaConf.create(cfg_dict)print(cfg)pose = PoseHigherResolutionNet(cfg)print(pose)
else:print(f"Error: The YAML file {yaml_file_path} does not exist.")

打印出来结果：

 

 

这篇关于yaml文件查看模型的架构的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---