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

..

test_active_rotated_filter.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_assign_score_withk.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_ball_query.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_bbox.py

[Feature] Add MPS bbox overlap (#2123)

2022-07-22 19:30:01 +08:00

test_bilinear_grid_sample.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_border_align.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_box_iou_rotated.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_carafe.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_cc_attention.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_chamfer_distance.py

[Feature] Add ChamferDistance op in gpu (#1933)

2022-06-08 18:12:57 +08:00

test_contour_expand.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_convex_iou.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_corner_pool.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_correlation.py

[Fix] Fix the unit test of correlation op (#1659)

2022-01-13 14:27:50 +08:00

test_deform_conv.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_deform_roi_pool.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_diff_iou_rotated.py

[Enhance] Differentiable rotated IoU (#1854)

2022-04-15 15:40:07 +08:00

test_focal_loss.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_furthest_point_sample.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_fused_bias_leakyrelu.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_gather_points.py

Update unittest for mmdet3d ops (#1811)

2022-03-18 12:52:55 +08:00

test_group_points.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_info.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_iou3d.py

[Fix] make iou3d.boxes_iou3d actually calculate 3D IoU (#2018)

2022-06-09 21:08:32 +08:00

test_knn.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_masked_conv2d.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_merge_cells.py

[Enhancement] Lower the restrictions of _resize method in BaseMergeCell (#1959)

2022-05-20 15:13:13 +08:00

test_min_area_polygons.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_modulated_deform_conv.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_ms_deformable_attn.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_nms_rotated.py

[Fix] Fix batched_nms for rotated box and add type hints for nms.py (#2006)

2022-06-10 19:49:07 +08:00

test_nms.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_onnx.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_pixel_group.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_points_in_polygons.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_prroi_pool.py

[Feature] Support PrRoIPool operation (#2113)

2022-07-23 23:08:32 +08:00

test_psa_mask.py

[Feature] Support PSAMask with cambricon MLU backend (#2024)

2022-06-08 16:13:13 +08:00

test_riroi_align_rotated.py

[Fix] Fix parrots unit tests (#1913)

2022-05-09 16:59:32 +08:00

test_roi_align_rotated.py

[Feature] Support RoiAlignRotated with cambricon MLU backend (#2033)

2022-06-08 14:13:28 +08:00

test_roi_align.py

[Refactor] Move is_mlu_available to mmcv/utils/device_type.py (#1884)

2022-04-17 00:15:35 +08:00

test_roi_pool.py

[Feature] Support RoiPool with Cambricon MLU backend (#2073)

2022-06-29 11:24:00 +08:00

test_roiaware_pool3d.py

Update unittest for mmdet3d ops (#1811)

2022-03-18 12:52:55 +08:00

test_roipoint_pool3d.py

Update unittest for mmdet3d ops (#1811)

2022-03-18 12:52:55 +08:00

test_rotated_feature_align.py

[Fix] Skip CPU test in test_rotated_feature_align.py for parrots (#1968)

2022-05-20 17:04:58 +08:00

test_saconv.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_scatter_points.py

[Fix] Fix parrots unit tests (#1913)

2022-05-09 16:59:32 +08:00

test_spconv.py

[Fix] Fix parrots unit tests (#1913)

2022-05-09 16:59:32 +08:00

test_syncbn.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_tensorrt_preprocess.py

[Fix] Fix parrots unit tests (#1913)

2022-05-09 16:59:32 +08:00

test_tensorrt.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_three_interpolate.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_three_nn.py

Add copyright pre-commit-hook (#1742)

2022-02-24 09:24:25 +08:00

test_tin_shift.py

[Refactor] Move is_mlu_available to mmcv/utils/device_type.py (#1884)

2022-04-17 00:15:35 +08:00

test_upfirdn2d.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00

test_voxelization.py

Add pyupgrade pre-commit hook (#1937)

2022-05-18 11:47:14 +08:00