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

46 lines
772 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 active rotated filter op for rotated detection. (#1598) * add active_rotated_filter * fix lint * fix lint * renaming nRotation and nOrientation * Update active_rotated_filter_cuda_kernel.cuh * Update active_rotated_filter_cuda.cu * fix bug * fix lint * Update test_active_rotated_filter.py * fix lint * Update active_rotated_filter_cuda_kernel.cuh * renaming * Update mmcv/ops/active_rotated_filter.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/active_rotated_filter.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/active_rotated_filter.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/active_rotated_filter.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/active_rotated_filter.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/active_rotated_filter.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * fix lint * Resolving conflicts * fix lint. * Update __init__.py * Update mmcv/ops/csrc/pytorch/cuda/active_rotated_filter_cuda.cu Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update active_rotated_filter.cpp * fix lint * Update mmcv/ops/csrc/pytorch/cpu/active_rotated_filter.cpp Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/csrc/pytorch/cpu/active_rotated_filter.cpp Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update active_rotated_filter.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 17:35:20 +08:00
- ActiveRotatedFilter
[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
[Feature] Add convex IoU CUDA op for rotated detection. (#1610) * add convex iou * fix lint * add convex_iou * fix convex_iou * add convex_giou * fix bug * fix lint * fix bug * Update * update * add kernel loop * fix bug. * fix polygen typo * simplify reverse * Update convex_iou_cuda_kernel.cuh * Update mmcv/ops/convex_iou.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/ops/convex_iou.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add AT_DISPATCH_FLOATING_TYPES_AND_HALF * fix lint * fix lint * Resolving conflicts Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 20:43:12 +08:00
- ConvexIoU
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
- CornerPool
- Deformable Convolution v1/v2
- Deformable RoIPool
[Enhance] Differentiable rotated IoU (#1854) * diff_iou_rotated is working * add test; fix lint * fix lint for test * disable cpu build * refactor files structure * fix comments * remove extra .repeat() * add comment * fix j-1 bug; update doc * fix clang lint * update docstrings * fix comments * fix comments
2022-04-15 15:40:07 +08:00
- DiffIoURotated
[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
[Feature] Add min_area_polygons CUDA op for rotated detection. (#1611) * init * Update pybind.cpp * Update min_area_polygons_cuda.cuh * Update cudabind.cpp * fix bug * Create test_min_area_polygons.py * add test * update * Update min_area_polygons_cuda.cuh * fix bugs. * Update min_area_polygons_cuda.cuh * Update min_area_polygons.py * Update min_area_polygons_cuda.cuh * merge these 4 nested loops * add AT_DISPATCH_FLOATING_TYPES_AND_HALF * fix lint * Resolving conflicts
2022-01-10 11:00:50 +08:00
- MinAreaPolygon
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
- NMS
[Feature] Add points_in_polygons CUDA op for rotated detection. (#1600)
2021-12-24 10:56:48 +08:00
- PointsInPolygons
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
- PSAMask
[Feature] Add RiRoIAlignRotated CUDA op for rotated detection. (#1599)
2021-12-24 00:30:34 +08:00
- RiRoIAlignRotated
[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
[Feature] Add spconv ops from mmdet3d (#1581) * add ops (spconv) of mmdet3d * fix typo * refactor code * resolve comments in #1452 * fix compile error * fix bugs * fix bug * transform from 'types.h' to 'extension.h' * fix bug * transform from 'types.h' to 'extension.h' in parrots * add extension.h in pybind.cpp * add unittest * Recover code * (1) Remove prettyprint.h (2) Switch `T` to `scalar_t` (3) Remove useless lines (4) Refine example in docstring of sparse_modules.py * (1) rename from `cu.h` to `cuh` (2) remove useless files (3) move cpu files to `pytorch/cpu` * reorganize files * Add docstring for sparse_functional.py * use dispatcher * remove template * use dispatch in cuda ops * resolve Segmentation fault * remove useless files * fix lint * fix lint * fix lint * fix unittest in test_build_layers.py * add tensorview into include_dirs when compiling * recover all deleted files * fix lint and comments * recover setup.py * replace tv::GPU as tv::TorchGPU & support device guard * fix lint Co-authored-by: hdc <hudingchang.vendor@sensetime.com> Co-authored-by: grimoire <yaoqian@sensetime.com>
2022-02-18 15:53:10 +08:00
- Sparse Convolution
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
- 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