mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmcv/docs/en/understand_mmcv/ops.md

39 lines
644 B
Markdown
Raw Normal View History

add ext ops, support parrots (#310) * 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>
2020-06-28 23:15:47 +08:00
## CUDA ops
We implement common CUDA ops used in detection, segmentation, etc.
[Feature] Add assign_score_withk op for paconv from mmdet3d (#1357) * add ops (assign_score_withk) in mmdet3d * add ops (assign_score_withk) in mmdet3d * refactor code * fix typo * refactor code * fix typo * polish docstring Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-10-15 13:18:31 +08:00
- AssignScoreWithK
[Feature] Add ballquery op from mmdet3d (#1332)
2021-09-23 21:08:00 +08:00
- BallQuery
add ext ops, support parrots (#310) * 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>
2020-06-28 23:15:47 +08:00
- BBoxOverlaps
- CARAFE
- CrissCrossAttention
- ContextBlock
- CornerPool
- Deformable Convolution v1/v2
- Deformable RoIPool
[Feature] Add voxel ops from mmdet3d (#1381) * add ops (voxel) in mmdet3d * add ops (voxel) in mmdet3d * add ops (voxel) in mmdet3d * refactor code * update test * update test * update test * refactor code * refactor code * refactor code * fix typo * fix typo
2021-10-21 11:08:11 +08:00
- DynamicScatter
[Feature] Add gather points op from mmdet3d (#1338) * add ops (gather points) in mmdet3d * add ops (gather points) in mmdet3d * refactor code * refactor code * fix typo
2021-10-14 16:58:12 +08:00
- GatherPoints
[Feature] Add fps op from mmdet3d (#1337) * 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
2021-10-13 20:39:31 +08:00
- FurthestPointSample
- FurthestPointSampleWithDist
add ext ops, support parrots (#310) * 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>
2020-06-28 23:15:47 +08:00
- GeneralizedAttention
[Feature] Add group points ops from mmdet3d (#1415) * add op (group points) and its related ops (ball query and knn) in mmdet3d * refactor code * fix typo * refactor code * fix typo * refactor code * make input contiguous Co-authored-by: zhouzaida <zhouzaida@163.com>
2021-10-23 14:01:31 +08:00
- GroupPoints
[Feature] Add knn op from mmdet3d (#1354) * add ops (knn) in mmdet3d * refactor code * refactor code * fix typo * fix typo * fix typo * refactor code * refactor code * spell typo * fix spell typo
2021-10-14 13:33:13 +08:00
- KNN
add ext ops, support parrots (#310) * 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>
2020-06-28 23:15:47 +08:00
- MaskedConv
- NMS
- PSAMask
[Feature] Add feature refine CUDA op for rotated detection. (#1603) * re PR * replace all the feature_refine with rotated_feature_align * replace all the FR with RotatedFeatureAlign * Update mmcv/ops/rotated_feature_align.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/rotated_feature_align.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * fix lint * Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/csrc/pytorch/cuda/cudabind.cpp Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/csrc/pytorch/cuda/rotated_feature_align_cuda.cu Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * fix lint * Update cudabind.cpp * Update cudabind.cpp * fix bug in test. & add backward test * fix lint Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-12-23 17:38:58 +08:00
- RotatedFeatureAlign
[Feature] Add roipoint_pool3d op from mmdet3d (#1358) * add ops (roipoint_pool3d) in mmdet3d * refactor code * fix typo * add unit test * refactor code * refactor code * refactor code * fix typo
2021-10-15 17:15:53 +08:00
- RoIPointPool3d
add ext ops, support parrots (#310) * 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>
2020-06-28 23:15:47 +08:00
- RoIPool
- RoIAlign
[Feature] Add roiaware pool3d ops from mmdet3d (#1382) * add ops (roiaware pool3d) in mmdet3d * refactor code * fix typo Co-authored-by: zhouzaida <zhouzaida@163.com>
2021-10-21 20:00:04 +08:00
- RoIAwarePool3d
add ext ops, support parrots (#310) * 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>
2020-06-28 23:15:47 +08:00
- SimpleRoIAlign
- SigmoidFocalLoss
- SoftmaxFocalLoss
- SoftNMS
- Synchronized BatchNorm
[Feature] Add voxel ops from mmdet3d (#1381) * add ops (voxel) in mmdet3d * add ops (voxel) in mmdet3d * add ops (voxel) in mmdet3d * refactor code * update test * update test * update test * refactor code * refactor code * refactor code * fix typo * fix typo
2021-10-21 11:08:11 +08:00
- Voxelization
[Feature] Add interpolate ops from mmdet3d (#1355) * add ops (interpolate) in mmdet3d * refactor code * fix typo * fix typo * fix typo * refactor code
2021-10-14 20:51:02 +08:00
- ThreeInterpolate
- ThreeNN
add ext ops, support parrots (#310) * 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>
2020-06-28 23:15:47 +08:00
- Weight standardization
[Feature] Add Correlation CUDA op (#1361)
2021-09-23 21:08:41 +08:00
- Correlation