mirror of https://github.com/open-mmlab/mmcv.git
63 lines
4.1 KiB
Markdown
63 lines
4.1 KiB
Markdown
## 算子
|
|
|
|
MMCV 提供了检测、分割等任务中常用的算子
|
|
|
|
| Device | CPU | CUDA | MLU | MPS | Ascend |
|
|
| ---------------------------- | --- | ---- | --- | --- | ------ |
|
|
| ActiveRotatedFilter | √ | √ | | | |
|
|
| AssignScoreWithK | | √ | | | |
|
|
| BallQuery | | √ | | | |
|
|
| BBoxOverlaps | | √ | √ | √ | |
|
|
| BorderAlign | | √ | | | |
|
|
| BoxIouRotated | √ | √ | | | |
|
|
| BoxIouQuadri | √ | √ | | | |
|
|
| 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 | √ | √ | | | |
|
|
| NMSQuadri | √ | √ | | | |
|
|
| PixelGroup | √ | | | | |
|
|
| PointsInBoxes | √ | √ | | | |
|
|
| PointsInPolygons | | √ | | | |
|
|
| PSAMask | √ | √ | √ | | √ |
|
|
| RotatedFeatureAlign | √ | √ | | | |
|
|
| RoIPointPool3d | | √ | √ | | |
|
|
| RoIPool | | √ | √ | | √ |
|
|
| RoIAlignRotated | √ | √ | √ | | |
|
|
| RiRoIAlignRotated | | √ | | | |
|
|
| RoIAlign | √ | √ | √ | | |
|
|
| RoIAwarePool3d | | √ | √ | | |
|
|
| SAConv2d | | √ | | | |
|
|
| SigmoidFocalLoss | | √ | √ | | √ |
|
|
| SoftmaxFocalLoss | | √ | | | √ |
|
|
| SoftNMS | | √ | | | |
|
|
| Sparse Convolution | | √ | | | |
|
|
| Synchronized BatchNorm | | √ | | | |
|
|
| ThreeInterpolate | | √ | | | |
|
|
| ThreeNN | | √ | √ | | |
|
|
| TINShift | | √ | √ | | |
|
|
| UpFirDn2d | | √ | | | |
|
|
| Voxelization | √ | √ | √ | | |
|
|
| PrRoIPool | | √ | | | |
|