|
|
| 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
|
|
|
