mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmcv/tests/test_ops
History
WilliamKyle b30755eef2
[Feature] Add rotated_feature_align cpu & onnxruntime implementation (#1878) 
* add rotated_feature_align cpu implementation

* add rotated_feature_align onnxruntime implementation

* Update code for advices from grimoire

Remove useless comment from mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp

Replace ambiguous function name atomicAdd in mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp

Simplify unit test with parameter in tests/test_ops/test_rotated_feature_align.py

Use fma in interpolate in mmcv/ops/csrc/onnxruntime/cpu/rotated_feature_align.cpp mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp

* Inline the function to reduce the overhead of the function call

Use fma in interpolate
 		2022-04-29 10:02:33 +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 Fix bbox unit test (#1914) 2022-04-26 16:49:28 +08:00
test_bilinear_grid_sample.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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 copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_carafe.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_cc_attention.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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 copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_deform_roi_pool.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_diff_iou_rotated.py [Enhance] Differentiable rotated IoU (#1854) 2022-04-15 15:40:07 +08:00
test_focal_loss.py [Refactor] Move is_mlu_available to mmcv/utils/device_type.py (#1884) 2022-04-17 00:15:35 +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 copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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 copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_iou3d.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_knn.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_masked_conv2d.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_merge_cells.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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 copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_ms_deformable_attn.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_nms.py [Refactor] Move is_mlu_available to mmcv/utils/device_type.py (#1884) 2022-04-17 00:15:35 +08:00
test_nms_rotated.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_onnx.py [Feature] Add rotated_feature_align cpu & onnxruntime implementation (#1878) 2022-04-29 10:02:33 +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_psa_mask.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_riroi_align_rotated.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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_align_rotated.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_roi_pool.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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 [Feature] Add rotated_feature_align cpu & onnxruntime implementation (#1878) 2022-04-29 10:02:33 +08:00
test_saconv.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_scatter_points.py Update unittest for mmdet3d ops (#1811) 2022-03-18 12:52:55 +08:00
test_spconv.py [Fix] Fix spconv unit test by releasing cache on GPU (#1872) 2022-04-14 22:35:29 +08:00
test_syncbn.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_tensorrt.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_tensorrt_preprocess.py Add copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +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 copyright pre-commit-hook (#1742) 2022-02-24 09:24:25 +08:00
test_voxelization.py [Feature] Add nondeterministic voxelization op from mmdet3d (#1783) 2022-03-15 14:21:34 +08:00