如何从Hydra配置文件中传递字典参数以实例化torchio的RemapLabels类
你可以直接在Hydra配置的_target_字段下方,按照YAML字典语法声明remapping参数即可,正确的config.yaml写法如下:
_target_: torchio.transforms.RemapLabels remapping: 2: 1 4: 3 6: 5 8: 7
YAML原生支持键值对字典结构,Hydra实例化对象时会自动把该字段解析为Python字典类型传入类的__init__方法,和你给出的TorchIO示例调用逻辑完全一致。如果需要传递RemapLabels的其他可选参数,也可以按照同样的键值对格式追加在配置项下方即可。
内容的提问来源于stack exchange,提问作者pete2213
相关产品推荐
相关产品推荐

