2022-06-15 10:57:29 +08:00
|
|
|
## 算子
|
2021-07-07 13:10:04 +08:00
|
|
|
|
2022-06-15 10:57:29 +08:00
|
|
|
MMCV 提供了检测、分割等任务中常用的算子
|
2021-09-21 13:54:37 +08:00
|
|
|
|
2022-07-22 19:30:01 +08:00
|
|
|
| Device | CPU | CUDA | MLU | MPS |
|
|
|
|
|
| ---------------------------- | --- | ---- | --- | --- |
|
|
|
|
|
| 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 | √ | √ | | |
|
