CUDA 算子

MMCV 提供了检测、分割等任务中常用的 CUDA 算子

• BallQuery

• BBoxOverlaps

• CARAFE

• CrissCrossAttention

• ContextBlock

• CornerPool

• Deformable Convolution v1/v2

• Deformable RoIPool

• GeneralizedAttention

• MaskedConv

• NMS

• PSAMask

• RoIPool

• RoIAlign

• SimpleRoIAlign

• SigmoidFocalLoss

• SoftmaxFocalLoss

• SoftNMS

• Synchronized BatchNorm

• Weight standardization

• Correlation