mmcv/mmcv/ops
pc 0230fc3bc0
[Feature] Add rotated_feature_align_cpu and update iod3d in parrots (#2027)
2022-06-07 16:20:06 +08:00
..
csrc [Feature] Add rotated_feature_align_cpu and update iod3d in parrots (#2027) 2022-06-07 16:20:06 +08:00
__init__.py [Enhance] Replace BEV IoU with 3D IoU (#1902) 2022-05-25 17:05:50 +08:00
active_rotated_filter.py Add type hints in mmcv/ops/active_rotated_filter.py (#2017) 2022-05-28 19:57:27 +08:00
assign_score_withk.py Add type hints in ops/assign_score_withk.py (#2023) 2022-05-29 20:55:50 +08:00
ball_query.py Add type hints for mmcv/ops (#2037) 2022-06-06 23:10:06 +08:00
bbox.py Add type hints in bbox.py and box_iou_rotated.py (#1981) 2022-05-27 10:21:48 +08:00
border_align.py Add type hints for mmcv/ops (#2037) 2022-06-06 23:10:06 +08:00
box_iou_rotated.py Add type hints in bbox.py and box_iou_rotated.py (#1981) 2022-05-27 10:21:48 +08:00
carafe.py Add type hints for mmcv/ops (#2032) 2022-06-01 21:02:03 +08:00
cc_attention.py Add type hints for mmcv/ops (#1987) 2022-05-27 12:45:47 +08:00
contour_expand.py Add type hints for mmcv/ops (#1987) 2022-05-27 12:45:47 +08:00
convex_iou.py Add type hints for mmcv/ops (#1987) 2022-05-27 12:45:47 +08:00
corner_pool.py Add type hints for mmcv/ops (#2032) 2022-06-01 21:02:03 +08:00
correlation.py Add type hints for mmcv/ops (#2037) 2022-06-06 23:10:06 +08:00
deform_conv.py Add type hints for mmcv/ops (#2037) 2022-06-06 23:10:06 +08:00
deform_roi_pool.py Add type hints for mmcv/ops (#2037) 2022-06-06 23:10:06 +08:00
deprecated_wrappers.py [Fix] Fix some warnings in unittest (#1522) 2021-12-22 10:57:10 +08:00
diff_iou_rotated.py Add type hints for mmcv/ops (#2032) 2022-06-01 21:02:03 +08:00
focal_loss.py [Enhancement] Add type hints in mmcv/ops/focal_loss.py. (#1994) 2022-06-02 23:32:01 +08:00
furthest_point_sample.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
fused_bias_leakyrelu.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
gather_points.py support fp16 training with gather_points op (#1758) 2022-03-02 17:55:13 +08:00
group_points.py [Docs] Fix the format of the docstring (#1573) 2021-12-09 22:15:52 +08:00
info.py [Docs] Add header for files (#1181) 2021-08-08 22:10:55 +08:00
iou3d.py Add type hint for the "boxes_iou_bev" function in iou3d.py (#2005) 2022-05-27 15:01:01 +08:00
knn.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
masked_conv.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
merge_cells.py [Enhancement] Lower the restrictions of _resize method in BaseMergeCell (#1959) 2022-05-20 15:13:13 +08:00
min_area_polygons.py [Feature] Add min_area_polygons CUDA op for rotated detection. (#1611) 2022-01-10 11:00:50 +08:00
modulated_deform_conv.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
multi_scale_deform_attn.py Fixed a typo in multi_scale_deform_attn.py (#1838) 2022-03-28 15:35:50 +08:00
nms.py [Fix] Fix parrots unit tests (#1913) 2022-05-09 16:59:32 +08:00
pixel_group.py Add type hints in mmcv/ops/pixel_group.py (#2016) 2022-05-31 10:06:05 +08:00
point_sample.py Add type hints in mmcv/ops/point_sample.py (#2019) 2022-05-29 09:25:27 +08:00
points_in_boxes.py Add type hint in mmcv/ops/points_in_boxes.py (#2007) 2022-05-28 11:57:34 +08:00
points_in_polygons.py Add type hint in mmcv/ops/points_in_polygons.py (#2013) 2022-05-28 11:57:08 +08:00
points_sampler.py Add type hints in mmcv/ops/points_sampler.py (#2015) 2022-05-28 15:35:17 +08:00
psa_mask.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
riroi_align_rotated.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
roi_align.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
roi_align_rotated.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
roi_pool.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
roiaware_pool3d.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
roipoint_pool3d.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
rotated_feature_align.py [Feature] Add rotated_feature_align cpu & onnxruntime implementation (#1878) 2022-04-29 10:02:33 +08:00
saconv.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
scatter_points.py [Fix] Fix some warnings in unittest (#1522) 2021-12-22 10:57:10 +08:00
sparse_conv.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
sparse_functional.py [Feature] Add spconv ops from mmdet3d (#1581) 2022-02-18 15:53:10 +08:00
sparse_modules.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
sparse_ops.py [Feature] Add spconv ops from mmdet3d (#1581) 2022-02-18 15:53:10 +08:00
sparse_pool.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
sparse_structure.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
sync_bn.py Add pyupgrade pre-commit hook (#1937) 2022-05-18 11:47:14 +08:00
three_interpolate.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
three_nn.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
tin_shift.py [Feature] Add several MLU ops (#1563) 2022-04-16 15:45:00 +08:00
upfirdn2d.py [Docs] Fix the format of return (#1462) 2021-12-09 10:54:38 +08:00
voxelize.py [Feature] Add nondeterministic voxelization op from mmdet3d (#1783) 2022-03-15 14:21:34 +08:00