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