cutler/modeling/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.

from .roi_heads import (
    ROI_HEADS_REGISTRY,
    ROIHeads,
    CustomStandardROIHeads,
    FastRCNNOutputLayers,
    build_roi_heads,
)
from .roi_heads.custom_cascade_rcnn import CustomCascadeROIHeads
from .roi_heads.fast_rcnn import FastRCNNOutputLayers
from .meta_arch.rcnn import GeneralizedRCNN, ProposalNetwork
from .meta_arch.build import build_model

_EXCLUDE = {"ShapeSpec"}
__all__ = [k for k in globals().keys() if k not in _EXCLUDE and not k.startswith("_")]