mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
63a6cbe913
* add ops (furthest point sample) in mmdet3d * refactor code * refactor code * fix typo * fix typo * fix typo * refactor code * fix typo * define DIVUP in common_cuda_helper.hpp
28 lines
473 B
Markdown
28 lines
473 B
Markdown
## CUDA ops
|
|
|
|
We implement common CUDA ops used in detection, segmentation, etc.
|
|
|
|
- BallQuery
|
|
- BBoxOverlaps
|
|
- CARAFE
|
|
- CrissCrossAttention
|
|
- ContextBlock
|
|
- CornerPool
|
|
- Deformable Convolution v1/v2
|
|
- Deformable RoIPool
|
|
- FurthestPointSample
|
|
- FurthestPointSampleWithDist
|
|
- GeneralizedAttention
|
|
- MaskedConv
|
|
- NMS
|
|
- PSAMask
|
|
- RoIPool
|
|
- RoIAlign
|
|
- SimpleRoIAlign
|
|
- SigmoidFocalLoss
|
|
- SoftmaxFocalLoss
|
|
- SoftNMS
|
|
- Synchronized BatchNorm
|
|
- Weight standardization
|
|
- Correlation
|