|
|
|
from detectron2.layers import ShapeSpec |
|
|
|
from .anchor_generator import build_anchor_generator, ANCHOR_GENERATOR_REGISTRY |
|
from .backbone import ( |
|
BACKBONE_REGISTRY, |
|
FPN, |
|
Backbone, |
|
ResNet, |
|
ResNetBlockBase, |
|
build_backbone, |
|
build_resnet_backbone, |
|
make_stage, |
|
ViT, |
|
SimpleFeaturePyramid, |
|
get_vit_lr_decay_rate, |
|
MViT, |
|
SwinTransformer, |
|
) |
|
from .meta_arch import ( |
|
META_ARCH_REGISTRY, |
|
SEM_SEG_HEADS_REGISTRY, |
|
GeneralizedRCNN, |
|
PanopticFPN, |
|
ProposalNetwork, |
|
RetinaNet, |
|
SemanticSegmentor, |
|
build_model, |
|
build_sem_seg_head, |
|
FCOS, |
|
) |
|
from .postprocessing import detector_postprocess |
|
from .proposal_generator import ( |
|
PROPOSAL_GENERATOR_REGISTRY, |
|
build_proposal_generator, |
|
RPN_HEAD_REGISTRY, |
|
build_rpn_head, |
|
) |
|
from .roi_heads import ( |
|
ROI_BOX_HEAD_REGISTRY, |
|
ROI_HEADS_REGISTRY, |
|
ROI_KEYPOINT_HEAD_REGISTRY, |
|
ROI_MASK_HEAD_REGISTRY, |
|
ROIHeads, |
|
StandardROIHeads, |
|
BaseMaskRCNNHead, |
|
BaseKeypointRCNNHead, |
|
FastRCNNOutputLayers, |
|
build_box_head, |
|
build_keypoint_head, |
|
build_mask_head, |
|
build_roi_heads, |
|
) |
|
from .test_time_augmentation import DatasetMapperTTA, GeneralizedRCNNWithTTA |
|
from .mmdet_wrapper import MMDetBackbone, MMDetDetector |
|
|
|
_EXCLUDE = {"ShapeSpec"} |
|
__all__ = [k for k in globals().keys() if k not in _EXCLUDE and not k.startswith("_")] |
|
|
|
|
|
from detectron2.utils.env import fixup_module_metadata |
|
|
|
fixup_module_metadata(__name__, globals(), __all__) |
|
del fixup_module_metadata |
|
|
