Skip to content

GitLab

Unverified Commit 9110df94 authored by Alex Yang's avatar Alex Yang Committed by GitHub
Browse files

[Docs] Reformat op docs (#2052) 

* [Docs]:update ops.md

* fix missing ops
parent 30f30581
Hide whitespace changes
Inline Side-by-side
Showing with 114 additions and 89 deletions
docs/en/understand_mmcv/ops.md
View file @ 9110df94
## CUDA ops ## ops
We implement common CUDA ops used in detection, segmentation, etc. We implement common ops used in detection, segmentation, etc.
- ActiveRotatedFilter | Device | CPU | CUDA | MLU |
- AssignScoreWithK | ---------------------------- | --- | ---- | --- |
- BallQuery | ActiveRotatedFilter | √ | √ | |
- BBoxOverlaps | AssignScoreWithK | | √ | |
- CARAFE | BallQuery | | √ | |
- CrissCrossAttention | BBoxOverlaps | | √ | √ |
- ContextBlock | BorderAlign | | √ | |
- ConvexIoU | BoxIouRotated | √ | √ | |
- CornerPool | CARAFE | | √ | |
- ChamferDistance | ChamferDistance | | √ | |
- Deformable Convolution v1/v2 | CrissCrossAttention | | √ | |
- Deformable RoIPool | ContourExpand | √ | | |
- DiffIoURotated | ConvexIoU | | √ | |
- DynamicScatter | CornerPool | | √ | |
- GatherPoints | Correlation | | √ | |
- FurthestPointSample | Deformable Convolution v1/v2 | √ | √ | |
- FurthestPointSampleWithDist | Deformable RoIPool | | √ | |
- GeneralizedAttention | DiffIoURotated | | √ | |
- GroupPoints | DynamicScatter | | √ | |
- KNN | FurthestPointSample | | √ | |
- MaskedConv | FurthestPointSampleWithDist | | √ | |
- MinAreaPolygon | FusedBiasLeakyrelu | | √ | |
- NMS | GatherPoints | | √ | |
- NMS3D | GroupPoints | | √ | |
- PointsInPolygons | Iou3d | | √ | |
- PSAMask | KNN | | √ | |
- RiRoIAlignRotated | MaskedConv | | √ | |
- RotatedFeatureAlign | MergeCells | | √ | |
- RoIPointPool3d | MinAreaPolygon | | √ | |
- RoIPool | ModulatedDeformConv2d | √ | √ | |
- RoIAlign | MultiScaleDeformableAttn | | √ | |
- RoIAwarePool3d | NMS | √ | √ | √ |
- SimpleRoIAlign | NMSRotated | √ | √ | |
- SigmoidFocalLoss | PixelGroup | √ | | |
- SoftmaxFocalLoss | PointsInBoxes | √ | √ | |
- SoftNMS | PointsInPolygons | | √ | |
- Sparse Convolution | PSAMask | √ | √ | √ |
- Synchronized BatchNorm | RotatedFeatureAlign | √ | √ | |
- Voxelization | RoIPointPool3d | | √ | |
- ThreeInterpolate | RoIPool | | √ | |
- ThreeNN | RoIAlignRotated | √ | √ | √ |
- Weight standardization | RiRoIAlignRotated | | √ | |
- Correlation | RoIAlign | √ | √ | √ |
| RoIAwarePool3d | | √ | |
| SAConv2d | | √ | |
| SigmoidFocalLoss | | √ | √ |
| SoftmaxFocalLoss | | √ | |
| SoftNMS | | √ | |
| Sparse Convolution | | √ | |
| Synchronized BatchNorm | | √ | |
| ThreeInterpolate | | √ | |
| ThreeNN | | √ | |
| TINShift | | √ | √ |
| UpFirDn2d | | √ | |
| Voxelization | √ | √ | |
docs/zh_cn/understand_mmcv/ops.md
View file @ 9110df94
## CUDA 算子 ## 算子
MMCV 提供了检测、分割等任务中常用的 CUDA 算子 MMCV 提供了检测、分割等任务中常用的算子
- ActiveRotatedFilter | Device | CPU | CUDA | MLU |
- AssignScoreWithK | ---------------------------- | --- | ---- | --- |
- BallQuery | ActiveRotatedFilter | √ | √ | |
- BBoxOverlaps | AssignScoreWithK | | √ | |
- CARAFE | BallQuery | | √ | |
- CrissCrossAttention | BBoxOverlaps | | √ | √ |
- ContextBlock | BorderAlign | | √ | |
- ConvexIoU | BoxIouRotated | √ | √ | |
- CornerPool | CARAFE | | √ | |
- ChamferDistance | ChamferDistance | | √ | |
- Deformable Convolution v1/v2 | CrissCrossAttention | | √ | |
- Deformable RoIPool | ContourExpand | √ | | |
- DiffIoURotated | ConvexIoU | | √ | |
- DynamicScatter | CornerPool | | √ | |
- GatherPoints | Correlation | | √ | |
- FurthestPointSample | Deformable Convolution v1/v2 | √ | √ | |
- FurthestPointSampleWithDist | Deformable RoIPool | | √ | |
- GeneralizedAttention | DiffIoURotated | | √ | |
- KNN | DynamicScatter | | √ | |
- MaskedConv | FurthestPointSample | | √ | |
- MinAreaPolygon | FurthestPointSampleWithDist | | √ | |
- NMS | FusedBiasLeakyrelu | | √ | |
- NMS3D | GatherPoints | | √ | |
- PointsInPolygons | GroupPoints | | √ | |
- PSAMask | Iou3d | | √ | |
- RotatedFeatureAlign | KNN | | √ | |
- RoIPointPool3d | MaskedConv | | √ | |
- RoIPool | MergeCells | | √ | |
- RiRoIAlignRotated | MinAreaPolygon | | √ | |
- RoIAlign | ModulatedDeformConv2d | √ | √ | |
- RoIAwarePool3d | MultiScaleDeformableAttn | | √ | |
- SimpleRoIAlign | NMS | √ | √ | √ |
- SigmoidFocalLoss | NMSRotated | √ | √ | |
- SoftmaxFocalLoss | PixelGroup | √ | | |
- SoftNMS | PointsInBoxes | √ | √ | |
- Sparse Convolution | PointsInPolygons | | √ | |
- Synchronized BatchNorm | PSAMask | √ | √ | √ |
- Voxelization | RotatedFeatureAlign | √ | √ | |
- ThreeInterpolate | RoIPointPool3d | | √ | |
- ThreeNN | RoIPool | | √ | |
- Weight standardization | RoIAlignRotated | √ | √ | √ |
- Correlation | RiRoIAlignRotated | | √ | |
| RoIAlign | √ | √ | √ |
| RoIAwarePool3d | | √ | |
| SAConv2d | | √ | |
| SigmoidFocalLoss | | √ | √ |
| SoftmaxFocalLoss | | √ | |
| SoftNMS | | √ | |
| Sparse Convolution | | √ | |
| Synchronized BatchNorm | | √ | |
| ThreeInterpolate | | √ | |
| ThreeNN | | √ | |
| TINShift | | √ | √ |
| UpFirDn2d | | √ | |
| Voxelization | √ | √ | |
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment