mmcv.ops¶

`BorderAlign`	Border align pooling layer.
`CARAFE`	CARAFE: Content-Aware ReAssembly of FEatures
`CARAFENaive`
`CARAFEPack`	A unified package of CARAFE upsampler that contains: 1) channel compressor 2) content encoder 3) CARAFE op.
`Conv2d`	alias of `mmcv.ops.deprecated_wrappers.Conv2d_deprecated`
`ConvTranspose2d`	alias of `mmcv.ops.deprecated_wrappers.ConvTranspose2d_deprecated`
`CornerPool`	Corner Pooling.
`Correlation`	Correlation operator
`CrissCrossAttention`	Criss-Cross Attention Module.
`DeformConv2d`	Deformable 2D convolution.
`DeformConv2dPack`	A Deformable Conv Encapsulation that acts as normal Conv layers.
`DeformRoIPool`
`DeformRoIPoolPack`
`DynamicScatter`	Scatters points into voxels, used in the voxel encoder with dynamic voxelization.
`FusedBiasLeakyReLU`	Fused bias leaky ReLU.
`GroupAll`	Group xyz with feature.
`Linear`	alias of `mmcv.ops.deprecated_wrappers.Linear_deprecated`
`MaskedConv2d`	A MaskedConv2d which inherits the official Conv2d.
`MaxPool2d`	alias of `mmcv.ops.deprecated_wrappers.MaxPool2d_deprecated`
`ModulatedDeformConv2d`
`ModulatedDeformConv2dPack`	A ModulatedDeformable Conv Encapsulation that acts as normal Conv layers.
`ModulatedDeformRoIPoolPack`
`MultiScaleDeformableAttention`	An attention module used in Deformable-Detr.
`PSAMask`
`PointsSampler`	Points sampling.
`PrRoIPool`	The operation of precision RoI pooling.
`QueryAndGroup`	Groups points with a ball query of radius.
`RiRoIAlignRotated`	Rotation-invariant RoI align pooling layer for rotated proposals.
`RoIAlign`	RoI align pooling layer.
`RoIAlignRotated`	RoI align pooling layer for rotated proposals.
`RoIAwarePool3d`	Encode the geometry-specific features of each 3D proposal.
`RoIPointPool3d`	Encode the geometry-specific features of each 3D proposal.
`RoIPool`
`SAConv2d`	SAC (Switchable Atrous Convolution)
`SigmoidFocalLoss`
`SimpleRoIAlign`
`SoftmaxFocalLoss`
`SparseConv2d`
`SparseConv3d`
`SparseConvTensor`
`SparseConvTranspose2d`
`SparseConvTranspose3d`
`SparseInverseConv2d`
`SparseInverseConv3d`
`SparseMaxPool2d`
`SparseMaxPool3d`
`SparseModule`	place holder, All module subclass from this will take sptensor in SparseSequential.
`SparseSequential`	A sequential container.
`SubMConv2d`
`SubMConv3d`
`SyncBatchNorm`	Synchronized Batch Normalization.
`TINShift`	Temporal Interlace Shift.
`Voxelization`	Convert kitti points(N, >=3) to voxels.

`active_rotated_filter`
`assign_score_withk`
`ball_query`
`batched_nms`	Performs non-maximum suppression in a batched fashion.
`bbox_overlaps`	Calculate overlap between two set of bboxes.
`border_align`
`box_iou_rotated`	Return intersection-over-union (Jaccard index) of boxes.
`boxes_iou3d`	Calculate boxes 3D IoU.
`boxes_iou_bev`	Calculate boxes IoU in the Bird’s Eye View.
`boxes_overlap_bev`	Calculate boxes BEV overlap.
`carafe`
`carafe_naive`
`chamfer_distance`
`contour_expand`	Expand kernel contours so that foreground pixels are assigned into instances.
`convex_giou`	Return generalized intersection-over-union (Jaccard index) between point sets and polygons.
`convex_iou`	Return intersection-over-union (Jaccard index) between point sets and polygons.
`deform_conv2d`
`deform_roi_pool`
`diff_iou_rotated_2d`	Calculate differentiable iou of rotated 2d boxes.
`diff_iou_rotated_3d`	Calculate differentiable iou of rotated 3d boxes.
`dynamic_scatter`
`furthest_point_sample`
`furthest_point_sample_with_dist`
`fused_bias_leakyrelu`	Fused bias leaky ReLU function.
`gather_points`
`grouping_operation`
`knn`
`masked_conv2d`
`min_area_polygons`	Find the smallest polygons that surrounds all points in the point sets.
`modulated_deform_conv2d`
`nms`	Dispatch to either CPU or GPU NMS implementations.
`nms3d`	3D NMS function GPU implementation (for BEV boxes).
`nms3d_normal`	Normal 3D NMS function GPU implementation.
`nms_bev`	NMS function GPU implementation (for BEV boxes).
`nms_match`	Matched dets into different groups by NMS.
`nms_normal_bev`	Normal NMS function GPU implementation (for BEV boxes).
`nms_rotated`	Performs non-maximum suppression (NMS) on the rotated boxes according to their intersection-over-union (IoU).
`pixel_group`	Group pixels into text instances, which is widely used text detection methods.
`point_sample`	A wrapper around `grid_sample()` to support 3D point_coords tensors Unlike `torch.nn.functional.grid_sample()` it assumes point_coords to lie inside `[0, 1] x [0, 1]` square.
`points_in_boxes_all`	Find all boxes in which each point is (CUDA).
`points_in_boxes_cpu`	Find all boxes in which each point is (CPU).
`points_in_boxes_part`	Find the box in which each point is (CUDA).
`points_in_polygons`	Judging whether points are inside polygons, which is used in the ATSS assignment for the rotated boxes.
`prroi_pool`
`rel_roi_point_to_rel_img_point`	Convert roi based relative point coordinates to image based absolute point coordinates.
`riroi_align_rotated`
`roi_align`
`roi_align_rotated`
`roi_pool`
`rotated_feature_align`
`scatter_nd`	pytorch edition of tensorflow scatter_nd.
`sigmoid_focal_loss`
`soft_nms`	Dispatch to only CPU Soft NMS implementations.
`softmax_focal_loss`
`three_interpolate`
`three_nn`
`tin_shift`
`upfirdn2d`	Pad, upsample, filter, and downsample a batch of 2D images.
`voxelization`