mirror of https://github.com/open-mmlab/mmcv.git
57 lines
3.1 KiB
Python
57 lines
3.1 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .bbox import bbox_overlaps
|
|
from .border_align import BorderAlign, border_align
|
|
from .box_iou_rotated import box_iou_rotated
|
|
from .carafe import CARAFE, CARAFENaive, CARAFEPack, carafe, carafe_naive
|
|
from .cc_attention import CrissCrossAttention
|
|
from .contour_expand import contour_expand
|
|
from .corner_pool import CornerPool
|
|
from .deform_conv import DeformConv2d, DeformConv2dPack, deform_conv2d
|
|
from .deform_roi_pool import (DeformRoIPool, DeformRoIPoolPack,
|
|
ModulatedDeformRoIPoolPack, deform_roi_pool)
|
|
from .deprecated_wrappers import Conv2d_deprecated as Conv2d
|
|
from .deprecated_wrappers import ConvTranspose2d_deprecated as ConvTranspose2d
|
|
from .deprecated_wrappers import Linear_deprecated as Linear
|
|
from .deprecated_wrappers import MaxPool2d_deprecated as MaxPool2d
|
|
from .focal_loss import (SigmoidFocalLoss, SoftmaxFocalLoss,
|
|
sigmoid_focal_loss, softmax_focal_loss)
|
|
from .fused_bias_leakyrelu import FusedBiasLeakyReLU, fused_bias_leakyrelu
|
|
from .info import (get_compiler_version, get_compiling_cuda_version,
|
|
get_onnxruntime_op_path)
|
|
from .masked_conv import MaskedConv2d, masked_conv2d
|
|
from .modulated_deform_conv import (ModulatedDeformConv2d,
|
|
ModulatedDeformConv2dPack,
|
|
modulated_deform_conv2d)
|
|
from .multi_scale_deform_attn import MultiScaleDeformableAttention
|
|
from .nms import batched_nms, nms, nms_match, nms_rotated, soft_nms
|
|
from .pixel_group import pixel_group
|
|
from .point_sample import (SimpleRoIAlign, point_sample,
|
|
rel_roi_point_to_rel_img_point)
|
|
from .psa_mask import PSAMask
|
|
from .roi_align import RoIAlign, roi_align
|
|
from .roi_align_rotated import RoIAlignRotated, roi_align_rotated
|
|
from .roi_pool import RoIPool, roi_pool
|
|
from .saconv import SAConv2d
|
|
from .sync_bn import SyncBatchNorm
|
|
from .tin_shift import TINShift, tin_shift
|
|
from .upfirdn2d import upfirdn2d
|
|
|
|
__all__ = [
|
|
'bbox_overlaps', 'CARAFE', 'CARAFENaive', 'CARAFEPack', 'carafe',
|
|
'carafe_naive', 'CornerPool', 'DeformConv2d', 'DeformConv2dPack',
|
|
'deform_conv2d', 'DeformRoIPool', 'DeformRoIPoolPack',
|
|
'ModulatedDeformRoIPoolPack', 'deform_roi_pool', 'SigmoidFocalLoss',
|
|
'SoftmaxFocalLoss', 'sigmoid_focal_loss', 'softmax_focal_loss',
|
|
'get_compiler_version', 'get_compiling_cuda_version',
|
|
'get_onnxruntime_op_path', 'MaskedConv2d', 'masked_conv2d',
|
|
'ModulatedDeformConv2d', 'ModulatedDeformConv2dPack',
|
|
'modulated_deform_conv2d', 'batched_nms', 'nms', 'soft_nms', 'nms_match',
|
|
'RoIAlign', 'roi_align', 'RoIPool', 'roi_pool', 'SyncBatchNorm', 'Conv2d',
|
|
'ConvTranspose2d', 'Linear', 'MaxPool2d', 'CrissCrossAttention', 'PSAMask',
|
|
'point_sample', 'rel_roi_point_to_rel_img_point', 'SimpleRoIAlign',
|
|
'SAConv2d', 'TINShift', 'tin_shift', 'box_iou_rotated', 'nms_rotated',
|
|
'upfirdn2d', 'FusedBiasLeakyReLU', 'fused_bias_leakyrelu',
|
|
'RoIAlignRotated', 'roi_align_rotated', 'pixel_group', 'contour_expand',
|
|
'MultiScaleDeformableAttention', 'BorderAlign', 'border_align'
|
|
]
|