| # Copyright 2023-2025 Marigold Team, ETH Zürich. All rights reserved. | |
| # | |
| # Licensed under the Apache License, Version 2.0 (the "License"); | |
| # you may not use this file except in compliance with the License. | |
| # You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, software | |
| # distributed under the License is distributed on an "AS IS" BASIS, | |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| # See the License for the specific language governing permissions and | |
| # limitations under the License. | |
| # -------------------------------------------------------------------------- | |
| # More information about Marigold: | |
| # https://marigoldmonodepth.github.io | |
| # https://marigoldcomputervision.github.io | |
| # Efficient inference pipelines are now part of diffusers: | |
| # https://huggingface.co/docs/diffusers/using-diffusers/marigold_usage | |
| # https://huggingface.co/docs/diffusers/api/pipelines/marigold | |
| # Examples of trained models and live demos: | |
| # https://huggingface.co/prs-eth | |
| # Related projects: | |
| # https://rollingdepth.github.io/ | |
| # https://marigolddepthcompletion.github.io/ | |
| # Citation (BibTeX): | |
| # https://github.com/prs-eth/Marigold#-citation | |
| # If you find Marigold useful, we kindly ask you to cite our papers. | |
| # -------------------------------------------------------------------------- | |
| import omegaconf | |
| from omegaconf import OmegaConf | |
| def recursive_load_config(config_path: str) -> OmegaConf: | |
| conf = OmegaConf.load(config_path) | |
| output_conf = OmegaConf.create({}) | |
| # Load base config. Later configs on the list will overwrite previous | |
| base_configs = conf.get("base_config", default_value=None) | |
| if base_configs is not None: | |
| assert isinstance(base_configs, omegaconf.listconfig.ListConfig) | |
| for _path in base_configs: | |
| assert ( | |
| _path != config_path | |
| ), "Circulate merging, base_config should not include itself." | |
| _base_conf = recursive_load_config(_path) | |
| output_conf = OmegaConf.merge(output_conf, _base_conf) | |
| # Merge configs and overwrite values | |
| output_conf = OmegaConf.merge(output_conf, conf) | |
| return output_conf | |
| def find_value_in_omegaconf(search_key, config): | |
| result_list = [] | |
| if isinstance(config, omegaconf.DictConfig): | |
| for key, value in config.items(): | |
| if key == search_key: | |
| result_list.append(value) | |
| elif isinstance(value, (omegaconf.DictConfig, omegaconf.ListConfig)): | |
| result_list.extend(find_value_in_omegaconf(search_key, value)) | |
| elif isinstance(config, omegaconf.ListConfig): | |
| for item in config: | |
| if isinstance(item, (omegaconf.DictConfig, omegaconf.ListConfig)): | |
| result_list.extend(find_value_in_omegaconf(search_key, item)) | |
| return result_list | |
| if "__main__" == __name__: | |
| conf = recursive_load_config("config/train_base.yaml") | |
| print(OmegaConf.to_yaml(conf)) | |