EasyDict可以方便地应用 .
来访问dict的值。
麻烦
d = {'foo':3, 'bar':{'x':1, 'y':2}}
print(d['foo']) # 如何想要访问字典的元素需要这么写
print(d['bar']['y']) # 如果想要继续访问字典中字典的元素需要使用二维数组
# print(d.foo) 这样写会出错哦!
3
2
简单
from easydict import EasyDict as edict
easy = edict(d = {'foo':3, 'bar':{'x':1, 'y':2}}) # 将普通的字典传入到edict()
print(easy['foo']) # 这是传统的方法
print(easy.foo) # 这是我们使用easydict输出二者结果是一样的,但是可以更为方便的使用字典了
print(easy.bar.x) # 我们也是可以很方便的使用字典中字典的元素了
3
3
1
设置属性
from easydict import EasyDict as edict
d = edict() # 这个是输出{}
d.foo = 3 # 我们可以直接赋值语句对字典元素进行创建
d.bar = {'prob':'value'} # 另外我们也可以创建字典中的字典
d.bar.prob = 'newer' # 另外我们也可以很方便的修改字典中元素的值
print(d)
{'foo': 3, 'bar': {'prob': 'newer'}}
利用 Easydict 建立全局变量
在深度学习中往往利用easydict建立一个全局的变量
from easydict import EasyDict as edict
config = edict() # 创建一个空白的 dict
config.TRAIN = edict() # Train是key,值是{}
config.Test = edict() # Test是key,值是{}
# config.TRAIN = {} # 这个和上面的那句话是等价的,相当于创建一个字典的key
config.TRAIN.batch_size = 25 # 然后在里面写值,表示Train里面的value也是一个字典
config.TRAIN.early_stopping_num = 10
config.TRAIN.lr = 0.0001
print(config)
{'TRAIN': {'batch_size': 25, 'early_stopping_num': 10, 'lr': 0.0001}, 'Test': {}}
利用 EasyDict 读取配置文件
同样,Easydict 还可以配合 yaml 配置文件一起使用。下面是一个简单的例子。
import yaml
from easydict import EasyDict
def setup_config():
with open(os.path.join('test_config.yaml')) as f:
cfg = yaml.safe_load(f) # 读取配置文件
cfg = EasyDict(cfg) # 存成 Easydict 的格式
return cfg
参考(感谢)
https://blog.youkuaiyun.com/alxe_made/article/details/80507415