Shortcuts

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

Read the Docs v: v1.3.18
Versions
latest
stable
v1.3.18
v1.3.17
v1.3.16
v1.3.15
v1.3.14
v1.3.13
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.