mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
35affc35bf
* support prroipooling * fix build bug * fix bug of precision * fix lint * add ut * add license and citation * polish op and add type hint * refactor ut * add docstring for prroipool * polish code * format cu and cuh code * fix format * fix format * fix typo Co-authored-by: zhouzaida <zhouzaida@163.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
3.4 KiB
3.4 KiB
ops
We implement common ops used in detection, segmentation, etc.
| 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 | √ | √ | ||
| PrRoIPool | √ |