Spaces:

Realcat
/

image-matching-webui

Running on Zero

image-matching-webui / third_party /TopicFM /src /__init__.py

fix: roma

358ab8f over 3 years ago

318 Bytes

from yacs.config import CfgNode
from .config.default import _CN


def lower_config(yacs_cfg):
    if not isinstance(yacs_cfg, CfgNode):
        return yacs_cfg
    return {k.lower(): lower_config(v) for k, v in yacs_cfg.items()}


def get_model_cfg():
    cfg = lower_config(lower_config(_CN))
    return cfg["model"]

	from yacs.config import CfgNode
	from .config.default import _CN


	def lower_config(yacs_cfg):
	if not isinstance(yacs_cfg, CfgNode):
	return yacs_cfg
	return {k.lower(): lower_config(v) for k, v in yacs_cfg.items()}


	def get_model_cfg():
	cfg = lower_config(lower_config(_CN))
	return cfg["model"]