mirror of
https://github.com/FoundationVision/GLEE.git
synced 2025-06-03 14:51:17 +08:00
18 lines
475 B
Python
18 lines
475 B
Python
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
||
|
|
from .build import build_backbone, BACKBONE_REGISTRY # noqa F401 isort:skip
|
||
|
|
|
||
|
|
from .backbone import Backbone
|
||
|
|
from .fpn import FPN
|
||
|
|
from .regnet import RegNet
|
||
|
|
from .resnet import (
|
||
|
|
BasicStem,
|
||
|
|
ResNet,
|
||
|
|
ResNetBlockBase,
|
||
|
|
build_resnet_backbone,
|
||
|
|
make_stage,
|
||
|
|
BottleneckBlock,
|
||
|
|
)
|
||
|
|
|
||
|
|
__all__ = [k for k in globals().keys() if not k.startswith("_")]
|
||
|
|
# TODO can expose more resnet blocks after careful consideration
|