Spaces:
Sleeping
Sleeping
| # Copyright (c) OpenMMLab. All rights reserved. | |
| from .assign_score_withk import assign_score_withk | |
| from .ball_query import ball_query | |
| 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 .correlation import Correlation | |
| 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 .furthest_point_sample import (furthest_point_sample, | |
| furthest_point_sample_with_dist) | |
| from .fused_bias_leakyrelu import FusedBiasLeakyReLU, fused_bias_leakyrelu | |
| from .gather_points import gather_points | |
| from .group_points import GroupAll, QueryAndGroup, grouping_operation | |
| from .info import (get_compiler_version, get_compiling_cuda_version, | |
| get_onnxruntime_op_path) | |
| from .iou3d import boxes_iou_bev, nms_bev, nms_normal_bev | |
| from .knn import knn | |
| 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 .points_in_boxes import (points_in_boxes_all, points_in_boxes_cpu, | |
| points_in_boxes_part) | |
| from .points_sampler import PointsSampler | |
| 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 .roiaware_pool3d import RoIAwarePool3d | |
| from .roipoint_pool3d import RoIPointPool3d | |
| from .saconv import SAConv2d | |
| from .scatter_points import DynamicScatter, dynamic_scatter | |
| from .sync_bn import SyncBatchNorm | |
| from .three_interpolate import three_interpolate | |
| from .three_nn import three_nn | |
| from .tin_shift import TINShift, tin_shift | |
| from .upfirdn2d import upfirdn2d | |
| from .voxelize import Voxelization, voxelization | |
| __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', 'assign_score_withk', | |
| 'box_iou_rotated', 'RoIPointPool3d', 'nms_rotated', 'knn', 'ball_query', | |
| 'upfirdn2d', 'FusedBiasLeakyReLU', 'fused_bias_leakyrelu', | |
| 'RoIAlignRotated', 'roi_align_rotated', 'pixel_group', 'QueryAndGroup', | |
| 'GroupAll', 'grouping_operation', 'contour_expand', 'three_nn', | |
| 'three_interpolate', 'MultiScaleDeformableAttention', 'BorderAlign', | |
| 'border_align', 'gather_points', 'furthest_point_sample', | |
| 'furthest_point_sample_with_dist', 'PointsSampler', 'Correlation', | |
| 'boxes_iou_bev', 'nms_bev', 'nms_normal_bev', 'Voxelization', | |
| 'voxelization', 'dynamic_scatter', 'DynamicScatter', 'RoIAwarePool3d', | |
| 'points_in_boxes_part', 'points_in_boxes_cpu', 'points_in_boxes_all' | |
| ] | |