Spaces:
Running
on
L40S
Running
on
L40S
# Copyright (c) Facebook, Inc. and its affiliates. | |
from .box_head import ROI_BOX_HEAD_REGISTRY, build_box_head, FastRCNNConvFCHead | |
from .keypoint_head import ( | |
ROI_KEYPOINT_HEAD_REGISTRY, | |
build_keypoint_head, | |
BaseKeypointRCNNHead, | |
KRCNNConvDeconvUpsampleHead, | |
) | |
from .mask_head import ( | |
ROI_MASK_HEAD_REGISTRY, | |
build_mask_head, | |
BaseMaskRCNNHead, | |
MaskRCNNConvUpsampleHead, | |
) | |
from .roi_heads import ( | |
ROI_HEADS_REGISTRY, | |
ROIHeads, | |
Res5ROIHeads, | |
StandardROIHeads, | |
build_roi_heads, | |
select_foreground_proposals, | |
) | |
from .cascade_rcnn import CascadeROIHeads | |
from .rotated_fast_rcnn import RROIHeads | |
from .fast_rcnn import FastRCNNOutputLayers | |
from . import cascade_rcnn # isort:skip | |
__all__ = list(globals().keys()) | |