from . import registry, detection, segmentation
from .backbone import BACKBONE_CLASS_DICT, BACKBONE_FUNC_DICT
from .detection.registry import DETECTION_MODEL_CLASS_DICT, DETECTION_MODEL_FUNC_DICT
from .segmentation.registry import SEGMENTATION_MODEL_CLASS_DICT, SEGMENTATION_MODEL_FUNC_DICT
from .wrapper import WRAPPER_CLASS_DICT

MODEL_DICT = dict()
MODEL_DICT.update(BACKBONE_CLASS_DICT)
MODEL_DICT.update(BACKBONE_FUNC_DICT)
MODEL_DICT.update(DETECTION_MODEL_CLASS_DICT)
MODEL_DICT.update(DETECTION_MODEL_FUNC_DICT)
MODEL_DICT.update(SEGMENTATION_MODEL_CLASS_DICT)
MODEL_DICT.update(SEGMENTATION_MODEL_FUNC_DICT)
MODEL_DICT.update(WRAPPER_CLASS_DICT)
