Shortcuts

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