jwyang
first commit
4121bec
raw
history blame
697 Bytes
# 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