mirror of https://github.com/open-mmlab/mmcv.git
24 lines
395 B
Markdown
24 lines
395 B
Markdown
## CUDA ops
|
|
|
|
We implement common CUDA ops used in detection, segmentation, etc.
|
|
|
|
- BBoxOverlaps
|
|
- CARAFE
|
|
- CrissCrossAttention
|
|
- ContextBlock
|
|
- CornerPool
|
|
- Deformable Convolution v1/v2
|
|
- Deformable RoIPool
|
|
- GeneralizedAttention
|
|
- MaskedConv
|
|
- NMS
|
|
- PSAMask
|
|
- RoIPool
|
|
- RoIAlign
|
|
- SimpleRoIAlign
|
|
- SigmoidFocalLoss
|
|
- SoftmaxFocalLoss
|
|
- SoftNMS
|
|
- Synchronized BatchNorm
|
|
- Weight standardization
|