sherie
43ee50e948
Support fps and points_in_box ops for Ascend device ( #3031 )
2024-04-23 17:30:16 +08:00
DaGaiBa
e5562f8a45
Add multi_scale_deform_attn op adapter for NPU ( #3034 )
2024-03-01 15:02:58 +08:00
sherie
5494299ba2
[Feature] torch_npu support aclnn and add op ( #2997 )
2024-02-01 14:14:25 +08:00
MrShadowY
8fbacc446d
[Feature] Add the support of BoxIouRotated op for ascend device ( #2854 )
2023-07-19 11:32:45 +08:00
Danielmic
407c2bd062
[Feature] Add the implementation of dynamic_scatter with mlu-ops ( #2853 )
2023-06-30 16:59:32 +08:00
Danielmic
83f1b9e0ab
[Feature] Add the implementation of diff_iou_rotated with mlu-ops ( #2852 )
2023-06-30 15:38:02 +08:00
tudejiang79
fd3fbfe197
[Refactor] Repalce the implementation of rotated_feature_align with mlu_ops ( #2659 )
2023-06-01 00:58:29 +08:00
liuhw
e197effeca
[Feature] Add the support of arf op for ascend device ( #2789 )
2023-05-11 15:47:08 +08:00
Zhang
595f8fdff3
[Feature] Add the MLU support for box_iou_rotated op ( #2703 )
2023-04-20 19:31:08 +08:00
liuyuan1-v
0d1b224fb1
[Feature] Support NmsRotated with cambricon MLU backend ( #2643 )
...
* [Feature] Support NmsRotated with cambricon MLU backend
* [Feature] remove foolproofs in nms_rotated_mlu.cpp
* [Feature] fix lint in test_nms_rotated.py
* [Feature] fix kMLU not found in nms_rotated.cpp
* [Feature] modify mlu support in nms.py
* [Feature] modify nms_rotated support in ops.md
* [Feature] modify ops/nms.py
2023-03-23 11:36:32 +08:00
duzekun
06fa32853b
[Feature] Add MLU support for Sparse Convolution op ( #2589 )
...
* [Feature] Add sparse convolution MLU API
* [Feature] update cpp code style
* end-of-file
* delete libext.a
* code style
* update ops.md
---------
Co-authored-by: budefei <budefei@cambricon.com>
2023-03-20 19:28:13 +08:00
sherie
01a0f53ea4
[Feature] Add Ascend support for RoIAlign op ( #2638 )
...
* add roi_algin op
* add unit test
* update ops.md
* fix typo
---------
Co-authored-by: momo609 <963372609.qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-20 19:10:42 +08:00
liuhw
745985937a
[Feature] Add the support of voxelization op for ascend device ( #2613 )
...
* update
* update
2023-03-03 20:07:36 +08:00
liuhw
5a61e534ea
[Feature] Add Ascend support for bbox_overlaps ( #2580 )
...
* Add support for Ascend devices with bbox_overlaps
* ...
* ...
2023-02-06 18:33:16 +08:00
ckirchhoff
54ed0ed869
[Feature] Add support for Ascend devices with nms_rotated ( #2550 )
...
* [Feature]: add nms_rotated npu adaptater code
* [BugFix]: modify param in nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated.cpp
* [Doc]: add nms_rotated op in supported op list at ops.md
* [Test]: add nms_rotated unit_test
* [Bug]: remove device parameter in test_batched_nms function
2023-01-28 21:08:22 +08:00
liuduanhui
dfb03806a1
[Feature] Support ball_query with cambricon MLU backend and mlu-ops library. ( #2520 )
...
* [Feature] Support ball_query with cambricon MLU backend and mlu-ops library.
* [Fix] update operator data layout setting.
* [Fix] add cxx compile option to avoid symbol conflict.
* [Fix] fix lint errors.
* [Fix] update ops.md with info of ball_query support by MLU backend.
* [Feature] Fix typo.
* [Fix] Remove print.
* [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env.
* [Fix] update MMCV_MLU_OPS_PATH check logic.
* [Fix] update error info when failed to download mlu-ops.
* [Fix] check mlu-ops version matching info in mmcv.
* [Fix] revise wrong filename.
* [Fix] remove f.close and re.
2023-01-18 14:41:04 +08:00
zcc-zjut
84f60c178c
[Feature] Add support for Ascend devices with gather_points ( #2555 )
2023-01-18 10:33:56 +08:00
mengpenghui
c310d28c8f
[Feature] Add MLU support for DCN ( #2540 )
2023-01-13 17:46:19 +08:00
sherie
2810718a99
[Feature] Add Ascend support for RoIPool op ( #2483 )
...
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
2023-01-12 11:52:28 +08:00
ZShaopeng
48ea88ab9f
[Feature] Support Voxelization with cambricon MLU device ( #2500 )
...
* [Feature] Support hard_voxelize with cambricon MLU backend
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
2023-01-10 19:43:45 +08:00
DaGaiBa
73856344f5
[Feature] Support PSAMask op for Ascend device ( #2487 )
2023-01-06 20:15:50 +08:00
Ryan Wang
f76de9077b
[Feature] Support DCNv1 on Ascend device ( #2480 )
...
* update lately npu modification--DCNv1
update lately npu modification--DCNv1
* update lately npu modification--DCNv1
* update lately npu modification--DCNv1
* update lately npu modification--DCNv1
* update lately npu modification--DCNv1
* update lately npu modification--DCNv1
* check code
* Add ops to EN/ZH documents
2023-01-06 14:17:29 +08:00
jayggh
46eb9ec5d0
[Feature] Add NPU adapter for Deformable RoIPool ( #2481 )
...
* add npu adapter for deformroipool
* cleancode
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
2022-12-14 15:09:46 +08:00
jayggh
4c51afce2a
[Feature] Add FusedBiasLeakyRelu npu adapter ( #2474 )
...
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* clean code
* fix autocast bugs on npu (#2273 )
fix autocast bugs on npu (#2273 )
* code format
* code format
* code format
* bug fix
* pytorch_npu_helper.hpp clean code
* Npu dev (#2306 )
* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather
* raise ImportError when compile with npu
* add npu test case (#2307 )
* add npu test case
* Update focal_loss.py
* add comment
* clean lint
* update dtype assert
* update DDP forward and comment
* fix bug
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* sigmoidfocalloss npu adapter bug fix
* BugFix: modify softmaxFocalLoss adapter
* BugFix: remove equal sign in the code
* add npu install information in README
* add modulatedDeformConv npu adapter
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* add modulatedDeformConv npu adapter
* merge master branch 20221103
* Add masked_ Conv2d operator in NPU
* add nms_npu
* fix bug
* fix code check
* fix code check
* fix code check
* Masked_conv2d NPU
* Masked_conv2d NPU
* Masked_conv2d NPU
* remove npu-install-info in README.md
* annotate the clang-format in pre-commit-config-zh-ch.yaml
* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv
* Create fused_bias_leakyrelu_npu.cpp
Add NPU adapter for fused_bias_leaky_relu operator
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update ops.md
* Update ops.md
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update ops.md
* amend for CI
* bugfix
* amend ops.md
* Update test_fused_bias_leakyrelu.py
* clean code
* bugfix
* clean code
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-12-13 11:13:46 +08:00
wangjiangben-hw
37aa6dd04c
[Docs] Replace NPU with Ascend in installation documentation ( #2469 )
...
* update NPU --> ASCEND
* update docs
* update
* update docs
* Update docs/en/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/zh_cn/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/zh_cn/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/zh_cn/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/zh_cn/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/en/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/en/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/en/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update docs/en/get_started/build.md
* Update docs/zh_cn/get_started/build.md
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-07 15:07:08 +08:00
mengpenghui
49d891754b
[Feature] Support modulated_deform_conv with cambricon MLU backend ( #2411 )
...
* [Feature] Support modulated_deform_conv with cambricon MLU backend
* fix error of torch_mlu
* modify with commit suggestion
* Update modulated_deform_conv.py
* Update mmcv/ops/modulated_deform_conv.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-04 17:46:44 +08:00
ckirchhoff
944a53f9d9
[Docs] Add currently npu supported ops in ops.md ( #2420 )
2022-11-17 14:16:36 +08:00
ZShaopeng
12442667ff
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend ( #2396 )
...
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support msdeformattn_1104 with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* Revert "[Feature] Support MsDeformAttn with cambricon MLU backend"
This reverts commit 27963cccc86d240852a40a2c1510147a3e9f269f.
* [Feature] Support MsdeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsdeformAttn with cambricon MLU backend
Co-authored-by: zhangshaopeng <wicky-zheng@outlook.com>
Co-authored-by: wicky-zheng <root@notebook-mmcv-290m8-torch1-6-y0yv66-notebook-0.notebook-mmcv-290m8-torch1-6-y0yv66.ns-ad8b689a0ecd41fcb4469c803dcd539d.svc.cluster.local>
2022-11-16 14:08:04 +08:00
Zhang
b622fb2e29
[Feature] Support RoIAwarePool3d with cambricon MLU backend ( #2359 )
...
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-03 11:13:34 +08:00
Yue Zhou
a4f304a5f5
[Feature] Add box_iou_quadri & nms_quadri ( #2277 )
...
* update
* update
* fix link
* fix bug
* update nms_quadri
* fix lint
* Update test_nms_quadri.py
* Update box_iou_quadri.py
* fix bug
* Update test_nms_quadri.py
* Update box_iou_rotated_utils.hpp
* Update box_iou_quadri.py
* Update mmcv/ops/nms.py
2022-10-28 16:53:33 +08:00
cathyzhang222
7fd7058a9c
[Feature] Support Iou3d with cambricon MLU backend ( #2339 )
...
* [Feature] Support Iou3d with cambricon MLU backend
* [Feature] add double line
* [Feature] add line
2022-10-24 15:58:58 +08:00
ZShaopeng
83bedf5a2b
[Feature] Support RoipointPool3d with cambricon MLU backend ( #2247 )
...
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
2022-09-08 17:19:12 +08:00
bdf
e843d73fc7
[Feature] Support DeformRoiPool with cambricon MLU backend ( #2137 )
...
* [Feature] Support DeformRoiPool with cambricon MLU backend
* [Fix] Remove use of std library
* [Fix] Correct the error information
* [Refactor] Refactor test deform_roi_pool code
* [Fix] Fix judgment error
* [Fix] Modify the large tensor check
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 22:37:28 +08:00
liuduanhui
a08517790d
[Feature] Support ThreeNN with cambricon MLU backend ( #2215 )
2022-08-23 15:18:47 +08:00
bdf
e663670a74
[Feature] Support MaskedConv2d with cambricon MLU backend ( #2202 )
...
* [Feature] Support MaskedConv2d with cambricon MLU backend
* [Refactor] Refactor test masked_conv2d code
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-21 23:21:54 +08:00
bdf
832b845257
[Feature] Add carafe op for MLU ( #2212 )
...
* [Feature] Support CARAFE with Cambricon MLU backend
* [Docs] Add comments for common funtions
* [Test] Add allclose test for carafe
* Remove print
Co-authored-by: zcyKTH <zcy19950525@gmail.com>
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-19 10:36:26 +08:00
Jingwei Zhang
35affc35bf
[Feature] Support PrRoIPool operation ( #2113 )
...
* support prroipooling
* fix build bug
* fix bug of precision
* fix lint
* add ut
* add license and citation
* polish op and add type hint
* refactor ut
* add docstring for prroipool
* polish code
* format cu and cuh code
* fix format
* fix format
* fix typo
Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-07-23 23:08:32 +08:00
q.yao
22fadceecd
[Feature] Add MPS bbox overlap ( #2123 )
...
* add mps bbox overlap
* format
* update document and manifest
* update readme
2022-07-22 19:30:01 +08:00
zhouchenyang
2708fac6c5
[Feature] Support RoiPool with Cambricon MLU backend ( #2073 )
...
* [Feature] Support RoiPool with cambricon MLU backend
* [Docs] Update ops.md
2022-06-29 11:24:00 +08:00
Alex Yang
9110df94fc
[Docs] Reformat op docs ( #2052 )
...
* [Docs]:update ops.md
* fix missing ops
2022-06-15 10:57:29 +08:00
Yue Zhou
834d5978df
[Feature] Add ChamferDistance op in gpu ( #1933 )
...
* update
* fix lint
* fix lint
* Update mmcv/ops/chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update chamfer_distance.py
* Update test_chamfer_distance.py
* fix
* Update chamfer_distance_cuda_kernel.cuh
* Update chamfer_distance_cuda_kernel.cuh
* fix
* Update mmcv/ops/chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-08 18:12:57 +08:00
Danila Rukhovich
e9f48a4f8e
[Enhance] Replace BEV IoU with 3D IoU ( #1902 )
...
* add iou3d
* revert deprecated python function
* fix lint
* replace 3d iou/nms calls for bev iou/nms
2022-05-25 17:05:50 +08:00
Danila Rukhovich
aee596d523
[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
Wenhao Wu
c1de4c9bbf
[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
Yue Zhou
594ff3c0d3
[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
Yue Zhou
9acc892a44
[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
Yue Zhou
51b40c332a
[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
zhouyue
304efbb650
[Feature] Add points_in_polygons CUDA op for rotated detection. ( #1600 )
2021-12-24 10:56:48 +08:00
zhouyue
0bcbeadb53
[Feature] Add RiRoIAlignRotated CUDA op for rotated detection. ( #1599 )
2021-12-24 00:30:34 +08:00
zhouyue
2475dc3452
[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