maskrcnn_benchmark/modeling/rpn/__init__.py

# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
# from .rpn import build_rpn
from .rpn import RPNModule
from .retina import RetinaNetModule
from .fcos import FCOSModule
from .atss import ATSSModule
from .dyhead import DyHeadModule
from .vldyhead import VLDyHeadModule

_RPN_META_ARCHITECTURES = {"RPN": RPNModule,
                           "RETINA": RetinaNetModule,
                           "FCOS": FCOSModule,
                           "ATSS": ATSSModule,
                           "DYHEAD": DyHeadModule,
                           "VLDYHEAD": VLDyHeadModule
                           }


def build_rpn(cfg):
    """
    This gives the gist of it. Not super important because it doesn't change as much
    """
    rpn_arch = _RPN_META_ARCHITECTURES[cfg.MODEL.RPN_ARCHITECTURE]
    return rpn_arch(cfg)