安装
在Python中,可以使用PyYAML库来读取YAML文件。首先,确保已经安装了PyYAML库。如果没有安装,可以使用以下命令进行安装:
pip install pyyaml
Yaml文件示例
learning_rate: 0.001
num_epochs: 50
batch_size: 24
use_resnet: True
num_iter: 0
file_root: '/VOC2012/allimgs/'
读取代码
import yaml
with open('conf/train.yaml','r') as f:
data = yaml.safe_load(f)
print(data)
可以看到输出的是一个字典,遍历字典即可。
遍历字典
for key,value in data.items():
print('key:{%s}--value:{%s}'%(key,value))