gitextract_pswz5bz1/ ├── .dev_scripts/ │ └── check_installation.py ├── .dockerignore ├── .github/ │ ├── ISSUE_TEMPLATE/ │ │ ├── 1-bug-report.yml │ │ ├── 2-feature_request.yml │ │ ├── 3-documentation.yml │ │ └── config.yml │ ├── pull_request_template.md │ └── workflows/ │ ├── build_macos_wheel.yml │ ├── lint.yml │ ├── merge_stage_test.yml │ ├── pr_stage_test.yml │ └── publish-to-pypi.yml ├── .gitignore ├── .pre-commit-config-zh-cn.yaml ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── CITATION.cff ├── CONTRIBUTING.md ├── CONTRIBUTING_zh-CN.md ├── LICENSE ├── LICENSES.md ├── MANIFEST.in ├── README.md ├── README_zh-CN.md ├── TERMINOLOGY.md ├── docker/ │ ├── README.md │ ├── dev/ │ │ └── Dockerfile │ └── release/ │ └── Dockerfile ├── docs/ │ ├── en/ │ │ ├── Makefile │ │ ├── _static/ │ │ │ ├── css/ │ │ │ │ └── readthedocs.css │ │ │ └── version.json │ │ ├── _templates/ │ │ │ └── classtemplate.rst │ │ ├── api/ │ │ │ ├── arraymisc.rst │ │ │ ├── cnn.rst │ │ │ ├── image.rst │ │ │ ├── ops.rst │ │ │ ├── transforms.rst │ │ │ ├── utils.rst │ │ │ ├── video.rst │ │ │ └── visualization.rst │ │ ├── community/ │ │ │ ├── contributing.md │ │ │ └── pr.md │ │ ├── compatibility.md │ │ ├── conf.py │ │ ├── deployment/ │ │ │ └── mmcv_ops_definition.md │ │ ├── docutils.conf │ │ ├── faq.md │ │ ├── get_started/ │ │ │ ├── api_reference.md │ │ │ ├── build.md │ │ │ ├── installation.md │ │ │ ├── introduction.md │ │ │ └── previous_versions.md │ │ ├── index.rst │ │ ├── make.bat │ │ ├── switch_language.md │ │ └── understand_mmcv/ │ │ ├── cnn.md │ │ ├── data_process.md │ │ ├── data_transform.md │ │ ├── ops.md │ │ └── visualization.md │ └── zh_cn/ │ ├── Makefile │ ├── _static/ │ │ ├── css/ │ │ │ └── readthedocs.css │ │ └── version.json │ ├── _templates/ │ │ └── classtemplate.rst │ ├── api/ │ │ ├── arraymisc.rst │ │ ├── cnn.rst │ │ ├── image.rst │ │ ├── ops.rst │ │ ├── transforms.rst │ │ ├── utils.rst │ │ ├── video.rst │ │ └── visualization.rst │ ├── community/ │ │ ├── code_style.md │ │ ├── contributing.md │ │ └── pr.md │ ├── compatibility.md │ ├── conf.py │ ├── docutils.conf │ ├── faq.md │ ├── get_started/ │ │ ├── api_reference.md │ │ ├── article.md │ │ ├── build.md │ │ ├── installation.md │ │ ├── introduction.md │ │ └── previous_versions.md │ ├── index.rst │ ├── make.bat │ ├── switch_language.md │ └── understand_mmcv/ │ ├── cnn.md │ ├── data_process.md │ ├── data_transform.md │ ├── ops.md │ └── visualization.md ├── mmcv/ │ ├── __init__.py │ ├── arraymisc/ │ │ ├── __init__.py │ │ └── quantization.py │ ├── cnn/ │ │ ├── __init__.py │ │ ├── alexnet.py │ │ ├── bricks/ │ │ │ ├── __init__.py │ │ │ ├── activation.py │ │ │ ├── context_block.py │ │ │ ├── conv.py │ │ │ ├── conv2d_adaptive_padding.py │ │ │ ├── conv_module.py │ │ │ ├── conv_ws.py │ │ │ ├── depthwise_separable_conv_module.py │ │ │ ├── drop.py │ │ │ ├── generalized_attention.py │ │ │ ├── hsigmoid.py │ │ │ ├── hswish.py │ │ │ ├── non_local.py │ │ │ ├── norm.py │ │ │ ├── padding.py │ │ │ ├── plugin.py │ │ │ ├── scale.py │ │ │ ├── swish.py │ │ │ ├── transformer.py │ │ │ ├── upsample.py │ │ │ └── wrappers.py │ │ ├── resnet.py │ │ ├── rfsearch/ │ │ │ ├── __init__.py │ │ │ ├── operator.py │ │ │ ├── search.py │ │ │ └── utils.py │ │ ├── utils/ │ │ │ ├── __init__.py │ │ │ ├── flops_counter.py │ │ │ └── fuse_conv_bn.py │ │ └── vgg.py │ ├── image/ │ │ ├── __init__.py │ │ ├── colorspace.py │ │ ├── geometric.py │ │ ├── io.py │ │ ├── misc.py │ │ └── photometric.py │ ├── ops/ │ │ ├── __init__.py │ │ ├── active_rotated_filter.py │ │ ├── assign_score_withk.py │ │ ├── ball_query.py │ │ ├── bbox.py │ │ ├── bezier_align.py │ │ ├── bias_act.py │ │ ├── border_align.py │ │ ├── box_iou_quadri.py │ │ ├── box_iou_rotated.py │ │ ├── carafe.py │ │ ├── cc_attention.py │ │ ├── chamfer_distance.py │ │ ├── contour_expand.py │ │ ├── conv2d_gradfix.py │ │ ├── convex_iou.py │ │ ├── corner_pool.py │ │ ├── correlation.py │ │ ├── csrc/ │ │ │ ├── README.md │ │ │ ├── common/ │ │ │ │ ├── box_iou_rotated_utils.hpp │ │ │ │ ├── cuda/ │ │ │ │ │ ├── active_rotated_filter_cuda_kernel.cuh │ │ │ │ │ ├── assign_score_withk_cuda_kernel.cuh │ │ │ │ │ ├── ball_query_cuda_kernel.cuh │ │ │ │ │ ├── bbox_overlaps_cuda_kernel.cuh │ │ │ │ │ ├── bezier_align_cuda_kernel.cuh │ │ │ │ │ ├── border_align_cuda_kernel.cuh │ │ │ │ │ ├── box_iou_quadri_cuda.cuh │ │ │ │ │ ├── box_iou_rotated_cuda.cuh │ │ │ │ │ ├── carafe_cuda_kernel.cuh │ │ │ │ │ ├── carafe_naive_cuda_kernel.cuh │ │ │ │ │ ├── chamfer_distance_cuda_kernel.cuh │ │ │ │ │ ├── common_cuda_helper.hpp │ │ │ │ │ ├── convex_iou_cuda_kernel.cuh │ │ │ │ │ ├── correlation_cuda.cuh │ │ │ │ │ ├── deform_conv_cuda_kernel.cuh │ │ │ │ │ ├── deform_roi_pool_cuda_kernel.cuh │ │ │ │ │ ├── diff_iou_rotated_cuda_kernel.cuh │ │ │ │ │ ├── furthest_point_sample_cuda_kernel.cuh │ │ │ │ │ ├── gather_points_cuda_kernel.cuh │ │ │ │ │ ├── group_points_cuda_kernel.cuh │ │ │ │ │ ├── iou3d_cuda_kernel.cuh │ │ │ │ │ ├── knn_cuda_kernel.cuh │ │ │ │ │ ├── masked_conv2d_cuda_kernel.cuh │ │ │ │ │ ├── min_area_polygons_cuda.cuh │ │ │ │ │ ├── modulated_deform_conv_cuda_kernel.cuh │ │ │ │ │ ├── ms_deform_attn_cuda_kernel.cuh │ │ │ │ │ ├── nms_cuda_kernel.cuh │ │ │ │ │ ├── nms_quadri_cuda.cuh │ │ │ │ │ ├── nms_rotated_cuda.cuh │ │ │ │ │ ├── parrots_cudawarpfunction.cuh │ │ │ │ │ ├── points_in_boxes_cuda_kernel.cuh │ │ │ │ │ ├── points_in_polygons_cuda_kernel.cuh │ │ │ │ │ ├── prroi_pool_cuda_kernel.cuh │ │ │ │ │ ├── psamask_cuda_kernel.cuh │ │ │ │ │ ├── riroi_align_rotated_cuda_kernel.cuh │ │ │ │ │ ├── roi_align_cuda_kernel.cuh │ │ │ │ │ ├── roi_align_rotated_cuda_kernel.cuh │ │ │ │ │ ├── roi_pool_cuda_kernel.cuh │ │ │ │ │ ├── roiaware_pool3d_cuda_kernel.cuh │ │ │ │ │ ├── roipoint_pool3d_cuda_kernel.cuh │ │ │ │ │ ├── rotated_feature_align_cuda_kernel.cuh │ │ │ │ │ ├── scatter_points_cuda_kernel.cuh │ │ │ │ │ ├── sigmoid_focal_loss_cuda_kernel.cuh │ │ │ │ │ ├── softmax_focal_loss_cuda_kernel.cuh │ │ │ │ │ ├── spconv/ │ │ │ │ │ │ ├── indice.cuh │ │ │ │ │ │ └── reordering.cuh │ │ │ │ │ ├── stack_ball_query_cuda_kernel.cuh │ │ │ │ │ ├── stack_group_points_cuda_kernel.cuh │ │ │ │ │ ├── sync_bn_cuda_kernel.cuh │ │ │ │ │ ├── three_interpolate_cuda_kernel.cuh │ │ │ │ │ ├── three_nn_cuda_kernel.cuh │ │ │ │ │ ├── tin_shift_cuda_kernel.cuh │ │ │ │ │ └── voxelization_cuda_kernel.cuh │ │ │ │ ├── mlu/ │ │ │ │ │ ├── common_mlu_helper.hpp │ │ │ │ │ ├── masked_conv2d_mlu_kernel.mlu │ │ │ │ │ └── roi_pool_mlu_kernel.mlu │ │ │ │ ├── mps/ │ │ │ │ │ ├── MPSDevice.h │ │ │ │ │ ├── MPSLibrary.h │ │ │ │ │ ├── MPSLibrary.mm │ │ │ │ │ ├── MPSStream.h │ │ │ │ │ └── MPSUtils.h │ │ │ │ ├── musa/ │ │ │ │ │ ├── active_rotated_filter_musa_kernel.muh │ │ │ │ │ ├── assign_score_withk_musa_kernel.muh │ │ │ │ │ ├── ball_query_musa_kernel.muh │ │ │ │ │ ├── bbox_overlaps_musa_kernel.muh │ │ │ │ │ ├── bezier_align_musa_kernel.muh │ │ │ │ │ ├── border_align_musa_kernel.muh │ │ │ │ │ ├── box_iou_quadri_musa.muh │ │ │ │ │ ├── box_iou_rotated_musa.muh │ │ │ │ │ ├── carafe_musa_kernel.muh │ │ │ │ │ ├── carafe_naive_musa_kernel.muh │ │ │ │ │ ├── chamfer_distance_musa_kernel.muh │ │ │ │ │ ├── common_musa_helper.hpp │ │ │ │ │ ├── convex_iou_musa_kernel.muh │ │ │ │ │ ├── correlation_musa.muh │ │ │ │ │ ├── deform_conv_musa_kernel.muh │ │ │ │ │ ├── deform_roi_pool_musa_kernel.muh │ │ │ │ │ ├── diff_iou_rotated_musa_kernel.muh │ │ │ │ │ ├── furthest_point_sample_musa_kernel.muh │ │ │ │ │ ├── gather_points_musa_kernel.muh │ │ │ │ │ ├── group_points_musa_kernel.muh │ │ │ │ │ ├── iou3d_musa_kernel.muh │ │ │ │ │ ├── knn_musa_kernel.muh │ │ │ │ │ ├── masked_conv2d_musa_kernel.muh │ │ │ │ │ ├── min_area_polygons_musa.muh │ │ │ │ │ ├── modulated_deform_conv_musa_kernel.muh │ │ │ │ │ ├── ms_deform_attn_musa_kernel.muh │ │ │ │ │ ├── nms_musa_kernel.muh │ │ │ │ │ ├── nms_quadri_musa.muh │ │ │ │ │ ├── nms_rotated_musa.muh │ │ │ │ │ ├── points_in_boxes_musa_kernel.muh │ │ │ │ │ ├── points_in_polygons_musa_kernel.muh │ │ │ │ │ ├── prroi_pool_musa_kernel.muh │ │ │ │ │ ├── psamask_musa_kernel.muh │ │ │ │ │ ├── riroi_align_rotated_musa_kernel.muh │ │ │ │ │ ├── roi_align_musa_kernel.muh │ │ │ │ │ ├── roi_align_rotated_musa_kernel.muh │ │ │ │ │ ├── roi_pool_musa_kernel.muh │ │ │ │ │ ├── roiaware_pool3d_musa_kernel.muh │ │ │ │ │ ├── roipoint_pool3d_musa_kernel.muh │ │ │ │ │ ├── rotated_feature_align_musa_kernel.muh │ │ │ │ │ ├── scatter_points_musa_kernel.muh │ │ │ │ │ ├── sigmoid_focal_loss_musa_kernel.muh │ │ │ │ │ ├── softmax_focal_loss_musa_kernel.muh │ │ │ │ │ ├── spconv/ │ │ │ │ │ │ ├── indice.muh │ │ │ │ │ │ └── reordering.muh │ │ │ │ │ ├── stack_ball_query_musa_kernel.muh │ │ │ │ │ ├── stack_group_points_musa_kernel.muh │ │ │ │ │ ├── sync_bn_musa_kernel.muh │ │ │ │ │ ├── three_interpolate_musa_kernel.muh │ │ │ │ │ ├── three_nn_musa_kernel.muh │ │ │ │ │ ├── tin_shift_musa_kernel.muh │ │ │ │ │ └── voxelization_musa_kernel.muh │ │ │ │ ├── parrots_cpp_helper.hpp │ │ │ │ ├── parrots_cuda_helper.hpp │ │ │ │ ├── pytorch_cpp_helper.hpp │ │ │ │ ├── pytorch_cuda_helper.hpp │ │ │ │ ├── pytorch_device_registry.hpp │ │ │ │ ├── pytorch_mlu_helper.hpp │ │ │ │ ├── pytorch_musa_helper.hpp │ │ │ │ ├── pytorch_npu_helper.hpp │ │ │ │ ├── pytorch_npu_util.hpp │ │ │ │ └── utils/ │ │ │ │ └── spconv/ │ │ │ │ ├── paramsgrid.h │ │ │ │ ├── prettyprint.h │ │ │ │ ├── pybind11_utils.h │ │ │ │ ├── spconv/ │ │ │ │ │ ├── geometry.h │ │ │ │ │ ├── indice.h │ │ │ │ │ ├── maxpool.h │ │ │ │ │ ├── mp_helper.h │ │ │ │ │ ├── point2voxel.h │ │ │ │ │ └── reordering.h │ │ │ │ └── tensorview/ │ │ │ │ ├── helper_kernel.cuh │ │ │ │ ├── helper_kernel.muh │ │ │ │ ├── helper_launch.h │ │ │ │ └── tensorview.h │ │ │ ├── parrots/ │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ ├── active_rotated_filter_parrots.cpp │ │ │ │ ├── active_rotated_filter_pytorch.h │ │ │ │ ├── assign_score_withk.cpp │ │ │ │ ├── assign_score_withk_parrots.cpp │ │ │ │ ├── assign_score_withk_pytorch.h │ │ │ │ ├── ball_query._parrots.cpp │ │ │ │ ├── ball_query.cpp │ │ │ │ ├── ball_query_pytorch.h │ │ │ │ ├── bbox_overlaps.cpp │ │ │ │ ├── bbox_overlaps_parrots.cpp │ │ │ │ ├── bbox_overlaps_pytorch.h │ │ │ │ ├── border_align.cpp │ │ │ │ ├── border_align_parrots.cpp │ │ │ │ ├── border_align_pytorch.h │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ ├── box_iou_rotated_parrots.cpp │ │ │ │ ├── box_iou_rotated_pytorch.h │ │ │ │ ├── carafe.cpp │ │ │ │ ├── carafe_naive.cpp │ │ │ │ ├── carafe_naive_parrots.cpp │ │ │ │ ├── carafe_naive_pytorch.h │ │ │ │ ├── carafe_parrots.cpp │ │ │ │ ├── carafe_pytorch.h │ │ │ │ ├── chamfer_distance.cpp │ │ │ │ ├── chamfer_distance_parrots.cpp │ │ │ │ ├── chamfer_distance_pytorch.h │ │ │ │ ├── contour_expand.cpp │ │ │ │ ├── contour_expand_parrots.cpp │ │ │ │ ├── contour_expand_pytorch.h │ │ │ │ ├── convex_iou.cpp │ │ │ │ ├── convex_iou_parrots.cpp │ │ │ │ ├── convex_iou_pytorch.h │ │ │ │ ├── correlation.cpp │ │ │ │ ├── correlation_parrots.cpp │ │ │ │ ├── correlation_pytorch.h │ │ │ │ ├── cudabind.cpp │ │ │ │ ├── deform_conv.cpp │ │ │ │ ├── deform_conv_parrots.cpp │ │ │ │ ├── deform_conv_pytorch.h │ │ │ │ ├── deform_roi_pool.cpp │ │ │ │ ├── deform_roi_pool_parrots.cpp │ │ │ │ ├── deform_roi_pool_pytorch.h │ │ │ │ ├── diff_iou_rotated.cpp │ │ │ │ ├── diff_iou_rotated_parrots.cpp │ │ │ │ ├── diff_iou_rotated_pytorch.h │ │ │ │ ├── focal_loss.cpp │ │ │ │ ├── focal_loss_parrots.cpp │ │ │ │ ├── focal_loss_pytorch.h │ │ │ │ ├── furthest_point_sample.cpp │ │ │ │ ├── furthest_point_sample_parrots.cpp │ │ │ │ ├── furthest_point_sample_pytorch.h │ │ │ │ ├── fused_bias_leakyrelu.cpp │ │ │ │ ├── fused_bias_parrots.cpp │ │ │ │ ├── gather_points.cpp │ │ │ │ ├── gather_points_parrots.cpp │ │ │ │ ├── gather_points_pytorch.h │ │ │ │ ├── group_points.cpp │ │ │ │ ├── group_points_parrots.cpp │ │ │ │ ├── group_points_pytorch.h │ │ │ │ ├── info.cpp │ │ │ │ ├── iou3d.cpp │ │ │ │ ├── iou3d_parrots.cpp │ │ │ │ ├── iou3d_pytorch.h │ │ │ │ ├── knn.cpp │ │ │ │ ├── knn_parrots.cpp │ │ │ │ ├── knn_pytorch.h │ │ │ │ ├── masked_conv2d.cpp │ │ │ │ ├── masked_conv2d_parrots.cpp │ │ │ │ ├── masked_conv2d_pytorch.h │ │ │ │ ├── min_area_polygons.cpp │ │ │ │ ├── min_area_polygons_parrots.cpp │ │ │ │ ├── min_area_polygons_pytorch.h │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ ├── modulated_deform_conv_parrots.cpp │ │ │ │ ├── modulated_deform_conv_pytorch.h │ │ │ │ ├── ms_deform_attn.cpp │ │ │ │ ├── ms_deform_attn_parrots.cpp │ │ │ │ ├── nms.cpp │ │ │ │ ├── nms_parrots.cpp │ │ │ │ ├── nms_pytorch.h │ │ │ │ ├── nms_rotated.cpp │ │ │ │ ├── pixel_group.cpp │ │ │ │ ├── pixel_group_parrots.cpp │ │ │ │ ├── pixel_group_pytorch.h │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ ├── points_in_boxes_parrots.cpp │ │ │ │ ├── points_in_boxes_pytorch.h │ │ │ │ ├── points_in_polygons.cpp │ │ │ │ ├── points_in_polygons_parrots.cpp │ │ │ │ ├── points_in_polygons_pytorch.h │ │ │ │ ├── prroi_pool.cpp │ │ │ │ ├── prroi_pool_parrots.cpp │ │ │ │ ├── prroi_pool_pytorch.h │ │ │ │ ├── psamask.cpp │ │ │ │ ├── psamask_parrots.cpp │ │ │ │ ├── psamask_pytorch.h │ │ │ │ ├── riroi_align_rotated.cpp │ │ │ │ ├── riroi_align_rotated_parrots.cpp │ │ │ │ ├── riroi_align_rotated_pytorch.h │ │ │ │ ├── roi_align.cpp │ │ │ │ ├── roi_align_parrots.cpp │ │ │ │ ├── roi_align_pytorch.h │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ ├── roi_align_rotated_parrots.cpp │ │ │ │ ├── roi_align_rotated_pytorch.h │ │ │ │ ├── roi_pool.cpp │ │ │ │ ├── roi_pool_parrots.cpp │ │ │ │ ├── roi_pool_pytorch.h │ │ │ │ ├── roiaware_pool3d.cpp │ │ │ │ ├── roiaware_pool3d_parrots.cpp │ │ │ │ ├── roiaware_pool3d_pytorch.h │ │ │ │ ├── roipoint_pool3d.cpp │ │ │ │ ├── roipoint_pool3d_parrots.cpp │ │ │ │ ├── roipoint_pool3d_pytorch.h │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ ├── rotated_feature_align_parrots.cpp │ │ │ │ ├── rotated_feature_align_pytorch.h │ │ │ │ ├── sync_bn.cpp │ │ │ │ ├── sync_bn_parrots.cpp │ │ │ │ ├── sync_bn_pytorch.h │ │ │ │ ├── three_interpolate.cpp │ │ │ │ ├── three_interpolate_parrots.cpp │ │ │ │ ├── three_interpolate_pytorch.h │ │ │ │ ├── three_nn.cpp │ │ │ │ ├── three_nn_parrots.cpp │ │ │ │ ├── three_nn_pytorch.h │ │ │ │ ├── tin_shift.cpp │ │ │ │ ├── tin_shift_parrots.cpp │ │ │ │ ├── tin_shift_pytorch.h │ │ │ │ ├── upfirdn2d.cpp │ │ │ │ ├── upfirdn2d_parrots.cpp │ │ │ │ ├── voxelization.cpp │ │ │ │ ├── voxelization_parrots.cpp │ │ │ │ └── voxelization_pytorch.h │ │ │ └── pytorch/ │ │ │ ├── active_rotated_filter.cpp │ │ │ ├── assign_score_withk.cpp │ │ │ ├── ball_query.cpp │ │ │ ├── bbox_overlaps.cpp │ │ │ ├── bezier_align.cpp │ │ │ ├── bias_act.cpp │ │ │ ├── border_align.cpp │ │ │ ├── box_iou_quadri.cpp │ │ │ ├── box_iou_rotated.cpp │ │ │ ├── carafe.cpp │ │ │ ├── carafe_naive.cpp │ │ │ ├── chamfer_distance.cpp │ │ │ ├── contour_expand.cpp │ │ │ ├── convex_iou.cpp │ │ │ ├── correlation.cpp │ │ │ ├── cpu/ │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ ├── bbox_overlaps_cpu.cpp │ │ │ │ ├── bezier_align.cpp │ │ │ │ ├── box_iou_quadri.cpp │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ ├── deform_conv.cpp │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ ├── nms.cpp │ │ │ │ ├── nms_quadri.cpp │ │ │ │ ├── nms_rotated.cpp │ │ │ │ ├── pixel_group.cpp │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ ├── psamask.cpp │ │ │ │ ├── roi_align.cpp │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ ├── sparse_indice.cpp │ │ │ │ ├── sparse_maxpool.cpp │ │ │ │ ├── sparse_reordering.cpp │ │ │ │ └── voxelization.cpp │ │ │ ├── cuda/ │ │ │ │ ├── active_rotated_filter_cuda.cu │ │ │ │ ├── assign_score_withk_cuda.cu │ │ │ │ ├── ball_query_cuda.cu │ │ │ │ ├── bbox_overlaps_cuda.cu │ │ │ │ ├── bezier_align_cuda.cu │ │ │ │ ├── bias_act_cuda.cu │ │ │ │ ├── border_align_cuda.cu │ │ │ │ ├── box_iou_quadri_cuda.cu │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ ├── carafe_cuda.cu │ │ │ │ ├── carafe_naive_cuda.cu │ │ │ │ ├── chamfer_distance_cuda.cu │ │ │ │ ├── convex_iou.cu │ │ │ │ ├── correlation_cuda.cu │ │ │ │ ├── cudabind.cpp │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ ├── deform_roi_pool_cuda.cu │ │ │ │ ├── diff_iou_rotated_cuda.cu │ │ │ │ ├── filtered_lrelu.cu │ │ │ │ ├── focal_loss_cuda.cu │ │ │ │ ├── furthest_point_sample_cuda.cu │ │ │ │ ├── fused_bias_leakyrelu_cuda.cu │ │ │ │ ├── fused_spconv_ops_cuda.cu │ │ │ │ ├── gather_points_cuda.cu │ │ │ │ ├── group_points_cuda.cu │ │ │ │ ├── iou3d_cuda.cu │ │ │ │ ├── knn_cuda.cu │ │ │ │ ├── masked_conv2d_cuda.cu │ │ │ │ ├── min_area_polygons.cu │ │ │ │ ├── modulated_deform_conv_cuda.cu │ │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ │ ├── nms_cuda.cu │ │ │ │ ├── nms_quadri_cuda.cu │ │ │ │ ├── nms_rotated_cuda.cu │ │ │ │ ├── points_in_boxes_cuda.cu │ │ │ │ ├── points_in_polygons_cuda.cu │ │ │ │ ├── prroi_pool_cuda.cu │ │ │ │ ├── psamask_cuda.cu │ │ │ │ ├── riroi_align_rotated_cuda.cu │ │ │ │ ├── roi_align_cuda.cu │ │ │ │ ├── roi_align_rotated_cuda.cu │ │ │ │ ├── roi_pool_cuda.cu │ │ │ │ ├── roiaware_pool3d_cuda.cu │ │ │ │ ├── roipoint_pool3d_cuda.cu │ │ │ │ ├── rotated_feature_align_cuda.cu │ │ │ │ ├── scatter_points_cuda.cu │ │ │ │ ├── sparse_indice.cu │ │ │ │ ├── sparse_maxpool.cu │ │ │ │ ├── sparse_pool_ops_cuda.cu │ │ │ │ ├── sparse_reordering.cu │ │ │ │ ├── spconv_ops_cuda.cu │ │ │ │ ├── stack_ball_query_cuda.cu │ │ │ │ ├── stack_group_points_cuda.cu │ │ │ │ ├── sync_bn_cuda.cu │ │ │ │ ├── three_interpolate_cuda.cu │ │ │ │ ├── three_nn_cuda.cu │ │ │ │ ├── tin_shift_cuda.cu │ │ │ │ ├── upfirdn2d_kernel.cu │ │ │ │ └── voxelization_cuda.cu │ │ │ ├── deform_conv.cpp │ │ │ ├── deform_roi_pool.cpp │ │ │ ├── diff_iou_rotated.cpp │ │ │ ├── filtered_lrelu.cpp │ │ │ ├── focal_loss.cpp │ │ │ ├── furthest_point_sample.cpp │ │ │ ├── fused_bias_leakyrelu.cpp │ │ │ ├── fused_spconv_ops.cpp │ │ │ ├── gather_points.cpp │ │ │ ├── group_points.cpp │ │ │ ├── info.cpp │ │ │ ├── iou3d.cpp │ │ │ ├── knn.cpp │ │ │ ├── masked_conv2d.cpp │ │ │ ├── min_area_polygons.cpp │ │ │ ├── mlu/ │ │ │ │ ├── ball_query_mlu.cpp │ │ │ │ ├── bbox_overlaps_mlu.cpp │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ ├── carafe_mlu.cpp │ │ │ │ ├── deform_roi_pool_mlu.cpp │ │ │ │ ├── diff_iou_rotated_mlu.cpp │ │ │ │ ├── focal_loss_sigmoid_mlu.cpp │ │ │ │ ├── iou3d_mlu.cpp │ │ │ │ ├── masked_conv2d_mlu.cpp │ │ │ │ ├── mlu_common_helper.cpp │ │ │ │ ├── mlu_common_helper.h │ │ │ │ ├── ms_deform_attn_mlu.cpp │ │ │ │ ├── nms_mlu.cpp │ │ │ │ ├── nms_rotated_mlu.cpp │ │ │ │ ├── psamask_mlu.cpp │ │ │ │ ├── roi_align_mlu.cpp │ │ │ │ ├── roi_align_rotated_mlu.cpp │ │ │ │ ├── roi_pool_mlu.cpp │ │ │ │ ├── roiaware_pool3d_mlu.cpp │ │ │ │ ├── roipoint_pool3d_mlu.cpp │ │ │ │ ├── rotated_feature_align_mlu.cpp │ │ │ │ ├── scatter_points_mlu.cpp │ │ │ │ ├── sparse_conv_mlu.cpp │ │ │ │ ├── three_nn_mlu.cpp │ │ │ │ ├── tin_shift_mlu.cpp │ │ │ │ └── voxelization_mlu.cpp │ │ │ ├── modulated_deform_conv.cpp │ │ │ ├── mps/ │ │ │ │ └── bbox_overlaps_mps.mm │ │ │ ├── ms_deform_attn.cpp │ │ │ ├── musa/ │ │ │ │ ├── active_rotated_filter_musa.mu │ │ │ │ ├── assign_score_withk_musa.mu │ │ │ │ ├── ball_query_musa.mu │ │ │ │ ├── bbox_overlaps_musa.mu │ │ │ │ ├── bezier_align_musa.mu │ │ │ │ ├── bias_act_musa.mu │ │ │ │ ├── border_align_musa.mu │ │ │ │ ├── box_iou_quadri_musa.mu │ │ │ │ ├── box_iou_rotated_musa.mu │ │ │ │ ├── carafe_musa.mu │ │ │ │ ├── carafe_naive_musa.mu │ │ │ │ ├── chamfer_distance_musa.mu │ │ │ │ ├── convex_iou.mu │ │ │ │ ├── correlation_musa.mu │ │ │ │ ├── deform_conv_musa.mu │ │ │ │ ├── deform_roi_pool_musa.mu │ │ │ │ ├── diff_iou_rotated_musa.mu │ │ │ │ ├── filtered_lrelu.mu │ │ │ │ ├── focal_loss_musa.mu │ │ │ │ ├── furthest_point_sample_musa.mu │ │ │ │ ├── fused_bias_leakyrelu_musa.mu │ │ │ │ ├── fused_spconv_ops_musa.mu │ │ │ │ ├── gather_points_musa.mu │ │ │ │ ├── group_points_musa.mu │ │ │ │ ├── iou3d_musa.mu │ │ │ │ ├── knn_musa.mu │ │ │ │ ├── masked_conv2d_musa.mu │ │ │ │ ├── min_area_polygons.mu │ │ │ │ ├── modulated_deform_conv_musa.mu │ │ │ │ ├── ms_deform_attn_musa.mu │ │ │ │ ├── musabind.cpp │ │ │ │ ├── nms_musa.mu │ │ │ │ ├── nms_quadri_musa.mu │ │ │ │ ├── nms_rotated_musa.mu │ │ │ │ ├── points_in_boxes_musa.mu │ │ │ │ ├── points_in_polygons_musa.mu │ │ │ │ ├── prroi_pool_musa.mu │ │ │ │ ├── psamask_musa.mu │ │ │ │ ├── riroi_align_rotated_musa.mu │ │ │ │ ├── roi_align_musa.mu │ │ │ │ ├── roi_align_rotated_musa.mu │ │ │ │ ├── roi_pool_musa.mu │ │ │ │ ├── roiaware_pool3d_musa.mu │ │ │ │ ├── roipoint_pool3d_musa.mu │ │ │ │ ├── rotated_feature_align_musa.mu │ │ │ │ ├── scatter_points_musa.mu │ │ │ │ ├── sparse_indice.mu │ │ │ │ ├── sparse_maxpool.mu │ │ │ │ ├── sparse_pool_ops_musa.mu │ │ │ │ ├── sparse_reordering.mu │ │ │ │ ├── spconv_ops_musa.mu │ │ │ │ ├── stack_ball_query_musa.mu │ │ │ │ ├── stack_group_points_musa.mu │ │ │ │ ├── sync_bn_musa.mu │ │ │ │ ├── three_interpolate_musa.mu │ │ │ │ ├── three_nn_musa.mu │ │ │ │ ├── tin_shift_musa.mu │ │ │ │ ├── upfirdn2d_kernel.mu │ │ │ │ └── voxelization_musa.mu │ │ │ ├── nms.cpp │ │ │ ├── nms_quadri.cpp │ │ │ ├── nms_rotated.cpp │ │ │ ├── npu/ │ │ │ │ ├── active_rotated_filter_npu.cpp │ │ │ │ ├── assign_score_withk_npu.cpp │ │ │ │ ├── ball_query_npu.cpp │ │ │ │ ├── bbox_overlaps_npu.cpp │ │ │ │ ├── border_align_npu.cpp │ │ │ │ ├── box_iou_quadri_npu.cpp │ │ │ │ ├── box_iou_rotated_npu.cpp │ │ │ │ ├── boxes_overlap_bev_npu.cpp │ │ │ │ ├── chamfer_distance_npu.cpp │ │ │ │ ├── common_util.h │ │ │ │ ├── deform_roi_pool.cpp │ │ │ │ ├── diff_iou_rotated_npu.cpp │ │ │ │ ├── focal_loss_npu.cpp │ │ │ │ ├── furthest_point_sample_npu.cpp │ │ │ │ ├── furthest_point_sampling_with_dist_npu.cpp │ │ │ │ ├── fused_bias_leakyrelu_npu.cpp │ │ │ │ ├── gather_points_npu.cpp │ │ │ │ ├── group_points_npu.cpp │ │ │ │ ├── knn_npu.cpp │ │ │ │ ├── ms_deform_attn_npu.cpp │ │ │ │ ├── nms3d_normal_npu.cpp │ │ │ │ ├── nms3d_npu.cpp │ │ │ │ ├── nms_npu.cpp │ │ │ │ ├── nms_rotated_npu.cpp │ │ │ │ ├── points_in_box_npu.cpp │ │ │ │ ├── points_in_box_npu_all.cpp │ │ │ │ ├── points_in_polygons_npu.cpp │ │ │ │ ├── psa_mask_npu.cpp │ │ │ │ ├── roi_align_npu.cpp │ │ │ │ ├── roi_align_rotated_npu.cpp │ │ │ │ ├── roi_pool_npu.cpp │ │ │ │ ├── roiaware_pool3d_npu.cpp │ │ │ │ ├── roipoint_pool3d_forward.cpp │ │ │ │ ├── rotated_feature_align_npu.cpp │ │ │ │ ├── stack_ball_query_npu.cpp │ │ │ │ ├── stack_group_points_npu.cpp │ │ │ │ ├── three_interpolate_npu.cpp │ │ │ │ ├── three_nn_npu.cpp │ │ │ │ └── voxelization_npu.cpp │ │ │ ├── pixel_group.cpp │ │ │ ├── points_in_boxes.cpp │ │ │ ├── points_in_polygons.cpp │ │ │ ├── prroi_pool.cpp │ │ │ ├── psamask.cpp │ │ │ ├── pybind.cpp │ │ │ ├── riroi_align_rotated.cpp │ │ │ ├── roi_align.cpp │ │ │ ├── roi_align_rotated.cpp │ │ │ ├── roi_pool.cpp │ │ │ ├── roiaware_pool3d.cpp │ │ │ ├── roipoint_pool3d.cpp │ │ │ ├── rotated_feature_align.cpp │ │ │ ├── scatter_points.cpp │ │ │ ├── sparse_pool_ops.cpp │ │ │ ├── spconv_ops.cpp │ │ │ ├── spconv_utils.h │ │ │ ├── sync_bn.cpp │ │ │ ├── three_interpolate.cpp │ │ │ ├── three_nn.cpp │ │ │ ├── tin_shift.cpp │ │ │ ├── upfirdn2d.cpp │ │ │ └── voxelization.cpp │ │ ├── deform_conv.py │ │ ├── deform_roi_pool.py │ │ ├── deprecated_wrappers.py │ │ ├── diff_iou_rotated.py │ │ ├── filtered_lrelu.py │ │ ├── focal_loss.py │ │ ├── furthest_point_sample.py │ │ ├── fused_bias_leakyrelu.py │ │ ├── gather_points.py │ │ ├── group_points.py │ │ ├── info.py │ │ ├── iou3d.py │ │ ├── knn.py │ │ ├── masked_conv.py │ │ ├── merge_cells.py │ │ ├── min_area_polygons.py │ │ ├── modulated_deform_conv.py │ │ ├── multi_scale_deform_attn.py │ │ ├── nms.py │ │ ├── pixel_group.py │ │ ├── point_sample.py │ │ ├── points_in_boxes.py │ │ ├── points_in_polygons.py │ │ ├── points_sampler.py │ │ ├── prroi_pool.py │ │ ├── psa_mask.py │ │ ├── riroi_align_rotated.py │ │ ├── roi_align.py │ │ ├── roi_align_rotated.py │ │ ├── roi_pool.py │ │ ├── roiaware_pool3d.py │ │ ├── roipoint_pool3d.py │ │ ├── rotated_feature_align.py │ │ ├── saconv.py │ │ ├── scatter_points.py │ │ ├── sparse_conv.py │ │ ├── sparse_functional.py │ │ ├── sparse_modules.py │ │ ├── sparse_ops.py │ │ ├── sparse_pool.py │ │ ├── sparse_structure.py │ │ ├── sync_bn.py │ │ ├── three_interpolate.py │ │ ├── three_nn.py │ │ ├── tin_shift.py │ │ ├── upfirdn2d.py │ │ └── voxelize.py │ ├── transforms/ │ │ ├── __init__.py │ │ ├── base.py │ │ ├── builder.py │ │ ├── formatting.py │ │ ├── loading.py │ │ ├── processing.py │ │ ├── utils.py │ │ └── wrappers.py │ ├── utils/ │ │ ├── __init__.py │ │ ├── device_type.py │ │ ├── env.py │ │ ├── ext_loader.py │ │ └── parrots_jit.py │ ├── version.py │ ├── video/ │ │ ├── __init__.py │ │ ├── io.py │ │ ├── optflow.py │ │ └── processing.py │ └── visualization/ │ ├── __init__.py │ ├── color.py │ ├── image.py │ └── optflow.py ├── requirements/ │ ├── build.txt │ ├── docs.txt │ ├── optional.txt │ ├── runtime.txt │ └── test.txt ├── requirements.txt ├── setup.cfg ├── setup.py └── tests/ ├── test_arraymisc.py ├── test_cnn/ │ ├── test_build_layers.py │ ├── test_context_block.py │ ├── test_conv2d_adaptive_padding.py │ ├── test_conv_module.py │ ├── test_depthwise_seperable_conv_module.py │ ├── test_flops_counter.py │ ├── test_fuse_conv_bn.py │ ├── test_generalized_attention.py │ ├── test_hsigmoid.py │ ├── test_hswish.py │ ├── test_non_local.py │ ├── test_rfsearch/ │ │ ├── test_operator.py │ │ └── test_search.py │ ├── test_scale.py │ ├── test_silu.py │ ├── test_swish.py │ ├── test_transformer.py │ └── test_wrappers.py ├── test_image/ │ ├── test_colorspace.py │ ├── test_geometric.py │ ├── test_image_misc.py │ ├── test_io.py │ └── test_photometric.py ├── test_ops/ │ ├── output.pkl │ ├── test_active_rotated_filter.py │ ├── test_assign_score_withk.py │ ├── test_ball_query.py │ ├── test_bbox.py │ ├── test_bezier_align.py │ ├── test_bias_act.py │ ├── test_bilinear_grid_sample.py │ ├── test_border_align.py │ ├── test_box_iou_quadri.py │ ├── test_box_iou_rotated.py │ ├── test_carafe.py │ ├── test_cc_attention.py │ ├── test_chamfer_distance.py │ ├── test_contour_expand.py │ ├── test_conv_gradfix.py │ ├── test_convex_iou.py │ ├── test_corner_pool.py │ ├── test_correlation.py │ ├── test_deform_conv.py │ ├── test_deform_roi_pool.py │ ├── test_diff_iou_rotated.py │ ├── test_filtered_lrelu.py │ ├── test_focal_loss.py │ ├── test_furthest_point_sample.py │ ├── test_fused_bias_leakyrelu.py │ ├── test_gather_points.py │ ├── test_group_points.py │ ├── test_info.py │ ├── test_iou3d.py │ ├── test_knn.py │ ├── test_masked_conv2d.py │ ├── test_merge_cells.py │ ├── test_min_area_polygons.py │ ├── test_modulated_deform_conv.py │ ├── test_ms_deformable_attn.py │ ├── test_nms.py │ ├── test_nms_quadri.py │ ├── test_nms_rotated.py │ ├── test_onnx.py │ ├── test_pixel_group.py │ ├── test_points_in_polygons.py │ ├── test_prroi_pool.py │ ├── test_psa_mask.py │ ├── test_riroi_align_rotated.py │ ├── test_roi_align.py │ ├── test_roi_align_rotated.py │ ├── test_roi_pool.py │ ├── test_roiaware_pool3d.py │ ├── test_roipoint_pool3d.py │ ├── test_rotated_feature_align.py │ ├── test_saconv.py │ ├── test_scatter_points.py │ ├── test_spconv.py │ ├── test_syncbn.py │ ├── test_three_interpolate.py │ ├── test_three_nn.py │ ├── test_tin_shift.py │ ├── test_upfirdn2d.py │ └── test_voxelization.py ├── test_transforms/ │ ├── test_transforms_formatting.py │ ├── test_transforms_loading.py │ ├── test_transforms_processing.py │ └── test_transforms_wrapper.py ├── test_utils/ │ ├── test_env.py │ └── test_parrots_jit.py ├── test_video/ │ ├── test_optflow.py │ ├── test_processing.py │ └── test_reader.py └── test_visualization.py