mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
5c25ae1a19
* add ops (roipoint_pool3d) in mmdet3d * refactor code * fix typo * add unit test * refactor code * refactor code * refactor code * fix typo
34 lines
559 B
Markdown
34 lines
559 B
Markdown
## CUDA ops
|
|
|
|
We implement common CUDA ops used in detection, segmentation, etc.
|
|
|
|
- AssignScoreWithK
|
|
- BallQuery
|
|
- BBoxOverlaps
|
|
- CARAFE
|
|
- CrissCrossAttention
|
|
- ContextBlock
|
|
- CornerPool
|
|
- Deformable Convolution v1/v2
|
|
- Deformable RoIPool
|
|
- GatherPoints
|
|
- FurthestPointSample
|
|
- FurthestPointSampleWithDist
|
|
- GeneralizedAttention
|
|
- KNN
|
|
- MaskedConv
|
|
- NMS
|
|
- PSAMask
|
|
- RoIPointPool3d
|
|
- RoIPool
|
|
- RoIAlign
|
|
- SimpleRoIAlign
|
|
- SigmoidFocalLoss
|
|
- SoftmaxFocalLoss
|
|
- SoftNMS
|
|
- Synchronized BatchNorm
|
|
- ThreeInterpolate
|
|
- ThreeNN
|
|
- Weight standardization
|
|
- Correlation
|