coreml-community/ControlNet-v1-1-Annotators-cpu

1# Copyright (c) Facebook, Inc. and its affiliates.2from annotator.oneformer.detectron2.layers import ShapeSpec3 4from .anchor_generator import build_anchor_generator, ANCHOR_GENERATOR_REGISTRY5from .backbone import (6    BACKBONE_REGISTRY,7    FPN,8    Backbone,9    ResNet,10    ResNetBlockBase,11    build_backbone,12    build_resnet_backbone,13    make_stage,14    ViT,15    SimpleFeaturePyramid,16    get_vit_lr_decay_rate,17    MViT,18    SwinTransformer,19)20from .meta_arch import (21    META_ARCH_REGISTRY,22    SEM_SEG_HEADS_REGISTRY,23    GeneralizedRCNN,24    PanopticFPN,25    ProposalNetwork,26    RetinaNet,27    SemanticSegmentor,28    build_model,29    build_sem_seg_head,30    FCOS,31)32from .postprocessing import detector_postprocess33from .proposal_generator import (34    PROPOSAL_GENERATOR_REGISTRY,35    build_proposal_generator,36    RPN_HEAD_REGISTRY,37    build_rpn_head,38)39from .roi_heads import (40    ROI_BOX_HEAD_REGISTRY,41    ROI_HEADS_REGISTRY,42    ROI_KEYPOINT_HEAD_REGISTRY,43    ROI_MASK_HEAD_REGISTRY,44    ROIHeads,45    StandardROIHeads,46    BaseMaskRCNNHead,47    BaseKeypointRCNNHead,48    FastRCNNOutputLayers,49    build_box_head,50    build_keypoint_head,51    build_mask_head,52    build_roi_heads,53)54from .test_time_augmentation import DatasetMapperTTA, GeneralizedRCNNWithTTA55from .mmdet_wrapper import MMDetBackbone, MMDetDetector56 57_EXCLUDE = {"ShapeSpec"}58__all__ = [k for k in globals().keys() if k not in _EXCLUDE and not k.startswith("_")]59 60 61from annotator.oneformer.detectron2.utils.env import fixup_module_metadata62 63fixup_module_metadata(__name__, globals(), __all__)64del fixup_module_metadata65