算子¶
MMCV 提供了检测、分割等任务中常用的算子
| Device | CPU | CUDA | MLU |
|---|---|---|---|
| ActiveRotatedFilter | √ | √ | |
| AssignScoreWithK | √ | ||
| BallQuery | √ | ||
| BBoxOverlaps | √ | √ | |
| BorderAlign | √ | ||
| BoxIouRotated | √ | √ | |
| CARAFE | √ | ||
| ChamferDistance | √ | ||
| CrissCrossAttention | √ | ||
| ContourExpand | √ | ||
| ConvexIoU | √ | ||
| CornerPool | √ | ||
| Correlation | √ | ||
| Deformable Convolution v1/v2 | √ | √ | |
| Deformable RoIPool | √ | ||
| DiffIoURotated | √ | ||
| DynamicScatter | √ | ||
| FurthestPointSample | √ | ||
| FurthestPointSampleWithDist | √ | ||
| FusedBiasLeakyrelu | √ | ||
| GatherPoints | √ | ||
| GroupPoints | √ | ||
| Iou3d | √ | ||
| KNN | √ | ||
| MaskedConv | √ | ||
| MergeCells | √ | ||
| MinAreaPolygon | √ | ||
| ModulatedDeformConv2d | √ | √ | |
| MultiScaleDeformableAttn | √ | ||
| NMS | √ | √ | √ |
| NMSRotated | √ | √ | |
| PixelGroup | √ | ||
| PointsInBoxes | √ | √ | |
| PointsInPolygons | √ | ||
| PSAMask | √ | √ | √ |
| RotatedFeatureAlign | √ | √ | |
| RoIPointPool3d | √ | ||
| RoIPool | √ | √ | |
| RoIAlignRotated | √ | √ | √ |
| RiRoIAlignRotated | √ | ||
| RoIAlign | √ | √ | √ |
| RoIAwarePool3d | √ | ||
| SAConv2d | √ | ||
| SigmoidFocalLoss | √ | √ | |
| SoftmaxFocalLoss | √ | ||
| SoftNMS | √ | ||
| Sparse Convolution | √ | ||
| Synchronized BatchNorm | √ | ||
| ThreeInterpolate | √ | ||
| ThreeNN | √ | ||
| TINShift | √ | √ | |
| UpFirDn2d | √ | ||
| Voxelization | √ | √ |