detectron2/modeling/backbone/__init__.py

# Copyright (c) Facebook, Inc. and its affiliates.
from .build import build_backbone, build_text_backbone, BACKBONE_REGISTRY  # noqa F401 isort:skip

from .backbone import Backbone
from .fpn import FPN, LastLevelMaxPool
from .regnet import RegNet
from .resnet import (
    BasicStem,
    ResNet,
    ResNetBlockBase,
    build_resnet_backbone,
    make_stage,
    BottleneckBlock,
)
from .clip_backbone import ModifiedResNet, build_resnet_clip, build_clip_resnet_backbone, build_clip_language_encoder
from .clip_swin import build_clip_swin, build_clip_swin_backbone

__all__ = [k for k in globals().keys() if not k.startswith("_")]
# TODO can expose more resnet blocks after careful consideration