mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
c0f5492ee9
* add ext ops, support parrots * fix lint * fix lint * update op from mmdetection * support non-pytorch env * fix import bug * test not import mmcv.op * rename mmcv.op to mmcv.ops * fix compile warning * 1. fix syncbn warning in pytorch 1.5 2. support only cpu compile 3. add point_sample from mmdet * fix text bug * update docstrings * fix line endings * minor updates * remove non_local from ops * bug fix for nonlocal2d * rename ops_ext to _ext and _ext to _flow_warp_ext * update the doc * try clang-format github action * fix github action * add ops to api.rst * fix cpp format * fix clang format issues * remove .clang-format Co-authored-by: Kai Chen <chenkaidev@gmail.com>
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
|