CUDA 算子¶
MMCV 提供了检测、分割等任务中常用的 CUDA 算子
ActiveRotatedFilter
AssignScoreWithK
BallQuery
BBoxOverlaps
CARAFE
CrissCrossAttention
ContextBlock
ConvexIoU
CornerPool
ChamferDistance
Deformable Convolution v1/v2
Deformable RoIPool
DiffIoURotated
DynamicScatter
GatherPoints
FurthestPointSample
FurthestPointSampleWithDist
GeneralizedAttention
KNN
MaskedConv
MinAreaPolygon
NMS
NMS3D
PointsInPolygons
PSAMask
RotatedFeatureAlign
RoIPointPool3d
RoIPool
RiRoIAlignRotated
RoIAlign
RoIAwarePool3d
SimpleRoIAlign
SigmoidFocalLoss
SoftmaxFocalLoss
SoftNMS
Sparse Convolution
Synchronized BatchNorm
Voxelization
ThreeInterpolate
ThreeNN
Weight standardization
Correlation