| # Copy from: https://github.com/facebookresearch/detectron2/blob/2bd05b42983468c50a1c80d5e7dc3952980e1cd4/detectron2/layers | |
| # Copyright (c) Facebook, Inc. and its affiliates. | |
| # from .batch_norm import FrozenBatchNorm2d, get_norm, NaiveSyncBatchNorm, CycleBatchNormList | |
| # from .deform_conv import DeformConv, ModulatedDeformConv | |
| # from .mask_ops import paste_masks_in_image | |
| # from .nms import batched_nms, batched_nms_rotated, nms, nms_rotated | |
| from .roi_align import ROIAlign, roi_align | |
| from .roi_align_rotated import ROIAlignRotated, roi_align_rotated | |
| # from .shape_spec import ShapeSpec | |
| from .wrappers import ( | |
| BatchNorm2d, | |
| Conv2d, | |
| ConvTranspose2d, | |
| cat, | |
| interpolate, | |
| Linear, | |
| nonzero_tuple, | |
| cross_entropy, | |
| shapes_to_tensor, | |
| move_device_like, | |
| ) | |
| # from .blocks import CNNBlockBase, DepthwiseSeparableConv2d | |
| # from .aspp import ASPP | |
| # from .losses import ciou_loss, diou_loss | |
| __all__ = [k for k in globals().keys() if not k.startswith("_")] | |