├── .gitignore ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── waymo-3d-3class-2sweep.py │ │ ├── waymo-3d-3class-3sweep-submission.py │ │ ├── waymo-3d-3class-3sweep.py │ │ └── waymo-3d-3class.py │ ├── default_runtime.py │ └── schedules │ │ └── cosine_2x.py └── flatformer │ ├── flatformer_waymo_D1_2x_3class.py │ ├── flatformer_waymo_D1_2x_3class_2f.py │ ├── flatformer_waymo_D1_2x_3class_3f.py │ ├── flatformer_waymo_D5_1x_3class.py │ └── flatformer_waymo_D5_1x_3class_3f.py ├── mmdet3d ├── __init__.py ├── apis │ ├── __init__.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── anchor │ │ ├── __init__.py │ │ └── anchor_3d_generator.py │ ├── bbox │ │ ├── __init__.py │ │ ├── assigners │ │ │ └── __init__.py │ │ ├── box_np_ops.py │ │ ├── coders │ │ │ ├── __init__.py │ │ │ ├── anchor_free_bbox_coder.py │ │ │ ├── centerpoint_bbox_coders.py │ │ │ ├── delta_xyzwhlr_bbox_coder.py │ │ │ ├── groupfree3d_bbox_coder.py │ │ │ └── partial_bin_based_bbox_coder.py │ │ ├── iou_calculators │ │ │ ├── __init__.py │ │ │ └── iou3d_calculator.py │ │ ├── samplers │ │ │ ├── __init__.py │ │ │ └── iou_neg_piecewise_sampler.py │ │ ├── structures │ │ │ ├── __init__.py │ │ │ ├── base_box3d.py │ │ │ ├── box_3d_mode.py │ │ │ ├── cam_box3d.py │ │ │ ├── coord_3d_mode.py │ │ │ ├── depth_box3d.py │ │ │ ├── lidar_box3d.py │ │ │ └── utils.py │ │ └── transforms.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── indoor_eval.py │ │ ├── kitti_utils │ │ │ ├── __init__.py │ │ │ ├── eval.py │ │ │ └── rotate_iou.py │ │ ├── seg_eval.py │ │ └── waymo_utils │ │ │ └── prediction_kitti_to_waymo.py │ ├── points │ │ ├── __init__.py │ │ ├── base_points.py │ │ ├── cam_points.py │ │ ├── depth_points.py │ │ └── lidar_points.py │ ├── post_processing │ │ ├── __init__.py │ │ ├── box3d_nms.py │ │ └── merge_augs.py │ ├── utils │ │ ├── __init__.py │ │ ├── gaussian.py │ │ └── visualize.py │ ├── visualizer │ │ ├── __init__.py │ │ ├── image_vis.py │ │ ├── open3d_vis.py │ │ └── show_result.py │ └── voxel │ │ ├── __init__.py │ │ ├── builder.py │ │ └── voxel_generator.py ├── datasets │ ├── __init__.py │ ├── builder.py │ ├── custom_3d.py │ ├── dataset_wrappers.py │ ├── kitti_dataset.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── data_augment_utils.py │ │ ├── dbsampler.py │ │ ├── formating.py │ │ ├── loading.py │ │ ├── test_time_aug.py │ │ └── transforms_3d.py │ ├── utils.py │ └── waymo_dataset.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ └── second.py │ ├── builder.py │ ├── dense_heads │ │ ├── __init__.py │ │ └── centerpoint_head.py │ ├── detectors │ │ ├── __init__.py │ │ ├── base.py │ │ ├── dynamic_voxelnet.py │ │ ├── single_stage.py │ │ └── voxelnet.py │ ├── losses │ │ └── __init__.py │ ├── middle_encoders │ │ ├── __init__.py │ │ └── flatformer.py │ ├── necks │ │ ├── __init__.py │ │ └── second_fpn.py │ ├── utils │ │ ├── __init__.py │ │ ├── clip_sigmoid.py │ │ └── mlp.py │ └── voxel_encoders │ │ ├── __init__.py │ │ ├── pillar_encoder.py │ │ ├── utils.py │ │ └── voxel_encoder.py ├── ops │ ├── __init__.py │ ├── ball_query │ │ ├── __init__.py │ │ ├── ball_query.py │ │ └── src │ │ │ ├── ball_query.cpp │ │ │ └── ball_query_cuda.cu │ ├── furthest_point_sample │ │ ├── __init__.py │ │ ├── furthest_point_sample.py │ │ ├── points_sampler.py │ │ ├── src │ │ │ ├── furthest_point_sample.cpp │ │ │ └── furthest_point_sample_cuda.cu │ │ └── utils.py │ ├── gather_points │ │ ├── __init__.py │ │ ├── gather_points.py │ │ └── src │ │ │ ├── gather_points.cpp │ │ │ └── gather_points_cuda.cu │ ├── group_points │ │ ├── __init__.py │ │ ├── group_points.py │ │ └── src │ │ │ ├── group_points.cpp │ │ │ └── group_points_cuda.cu │ ├── interpolate │ │ ├── __init__.py │ │ ├── src │ │ │ ├── interpolate.cpp │ │ │ ├── three_interpolate_cuda.cu │ │ │ └── three_nn_cuda.cu │ │ ├── three_interpolate.py │ │ └── three_nn.py │ ├── iou3d │ │ ├── __init__.py │ │ ├── iou3d_utils.py │ │ └── src │ │ │ ├── iou3d.cpp │ │ │ └── iou3d_kernel.cu │ ├── knn │ │ ├── __init__.py │ │ ├── knn.py │ │ └── src │ │ │ ├── knn.cpp │ │ │ └── knn_cuda.cu │ ├── norm.py │ ├── paconv │ │ ├── __init__.py │ │ ├── assign_score.py │ │ ├── paconv.py │ │ ├── src │ │ │ ├── assign_score_withk.cpp │ │ │ └── assign_score_withk_cuda.cu │ │ └── utils.py │ ├── pointnet_modules │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── paconv_sa_module.py │ │ ├── point_fp_module.py │ │ └── point_sa_module.py │ ├── roiaware_pool3d │ │ ├── __init__.py │ │ ├── points_in_boxes.py │ │ ├── roiaware_pool3d.py │ │ └── src │ │ │ ├── points_in_boxes_cpu.cpp │ │ │ ├── points_in_boxes_cuda.cu │ │ │ ├── roiaware_pool3d.cpp │ │ │ └── roiaware_pool3d_kernel.cu │ ├── sparse_block.py │ ├── spconv │ │ ├── __init__.py │ │ ├── conv.py │ │ ├── functional.py │ │ ├── include │ │ │ ├── paramsgrid.h │ │ │ ├── prettyprint.h │ │ │ ├── pybind11_utils.h │ │ │ ├── spconv │ │ │ │ ├── fused_spconv_ops.h │ │ │ │ ├── geometry.h │ │ │ │ ├── indice.cu.h │ │ │ │ ├── indice.h │ │ │ │ ├── maxpool.h │ │ │ │ ├── mp_helper.h │ │ │ │ ├── point2voxel.h │ │ │ │ ├── pool_ops.h │ │ │ │ ├── reordering.cu.h │ │ │ │ ├── reordering.h │ │ │ │ └── spconv_ops.h │ │ │ ├── tensorview │ │ │ │ ├── helper_kernel.cu.h │ │ │ │ ├── helper_launch.h │ │ │ │ └── tensorview.h │ │ │ ├── torch_utils.h │ │ │ └── utility │ │ │ │ └── timer.h │ │ ├── modules.py │ │ ├── ops.py │ │ ├── overwrite_spconv │ │ │ └── write_spconv2.py │ │ ├── pool.py │ │ ├── src │ │ │ ├── all.cc │ │ │ ├── indice.cc │ │ │ ├── indice_cuda.cu │ │ │ ├── maxpool.cc │ │ │ ├── maxpool_cuda.cu │ │ │ ├── reordering.cc │ │ │ └── reordering_cuda.cu │ │ ├── structure.py │ │ └── test_utils.py │ └── voxel │ │ ├── __init__.py │ │ ├── scatter_points.py │ │ ├── src │ │ ├── scatter_points_cpu.cpp │ │ ├── scatter_points_cuda.cu │ │ ├── voxelization.cpp │ │ ├── voxelization.h │ │ ├── voxelization_cpu.cpp │ │ └── voxelization_cuda.cu │ │ └── voxelize.py ├── runner │ ├── __init__.py │ └── epoch_based_runner.py ├── utils │ ├── __init__.py │ ├── collect_env.py │ ├── logger.py │ └── syncbn.py └── version.py ├── setup.cfg ├── setup.py └── tools ├── __init__.py ├── analysis_tools ├── analyze_logs.py └── benchmark.py ├── analyze.py ├── analyze.sh ├── create_data.py ├── create_data.sh ├── create_submission ├── data_converter ├── __init__.py ├── create_gt_database.py ├── indoor_converter.py ├── kitti_converter.py ├── kitti_data_utils.py ├── lyft_converter.py ├── lyft_data_fixer.py ├── nuimage_converter.py ├── nuscenes_converter.py ├── s3dis_data_utils.py ├── scannet_data_utils.py ├── sunrgbd_data_utils.py └── waymo_converter.py ├── dist_test.sh ├── dist_train.sh ├── get_watmo.sh ├── misc ├── browse_dataset.py ├── fuse_conv_bn.py ├── print_config.py └── visualize_results.py ├── model_converters ├── convert_votenet_checkpoints.py ├── publish_model.py └── regnet2mmdet.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/waymo-3d-3class-2sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/_base_/datasets/waymo-3d-3class-2sweep.py -------------------------------------------------------------------------------- /configs/_base_/datasets/waymo-3d-3class-3sweep-submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/_base_/datasets/waymo-3d-3class-3sweep-submission.py -------------------------------------------------------------------------------- /configs/_base_/datasets/waymo-3d-3class-3sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/_base_/datasets/waymo-3d-3class-3sweep.py -------------------------------------------------------------------------------- /configs/_base_/datasets/waymo-3d-3class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/_base_/datasets/waymo-3d-3class.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/schedules/cosine_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/_base_/schedules/cosine_2x.py -------------------------------------------------------------------------------- /configs/flatformer/flatformer_waymo_D1_2x_3class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/flatformer/flatformer_waymo_D1_2x_3class.py -------------------------------------------------------------------------------- /configs/flatformer/flatformer_waymo_D1_2x_3class_2f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/flatformer/flatformer_waymo_D1_2x_3class_2f.py -------------------------------------------------------------------------------- /configs/flatformer/flatformer_waymo_D1_2x_3class_3f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/flatformer/flatformer_waymo_D1_2x_3class_3f.py -------------------------------------------------------------------------------- /configs/flatformer/flatformer_waymo_D5_1x_3class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/flatformer/flatformer_waymo_D5_1x_3class.py -------------------------------------------------------------------------------- /configs/flatformer/flatformer_waymo_D5_1x_3class_3f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/configs/flatformer/flatformer_waymo_D5_1x_3class_3f.py -------------------------------------------------------------------------------- /mmdet3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/__init__.py -------------------------------------------------------------------------------- /mmdet3d/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/apis/__init__.py -------------------------------------------------------------------------------- /mmdet3d/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/apis/test.py -------------------------------------------------------------------------------- /mmdet3d/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/apis/train.py -------------------------------------------------------------------------------- /mmdet3d/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/anchor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/anchor/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/anchor/anchor_3d_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/anchor/anchor_3d_generator.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/assigners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/assigners/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/box_np_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/box_np_ops.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/coders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/coders/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/coders/anchor_free_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/coders/anchor_free_bbox_coder.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/coders/centerpoint_bbox_coders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/coders/centerpoint_bbox_coders.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/coders/delta_xyzwhlr_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/coders/delta_xyzwhlr_bbox_coder.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/coders/groupfree3d_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/coders/groupfree3d_bbox_coder.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/coders/partial_bin_based_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/coders/partial_bin_based_bbox_coder.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/iou_calculators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/iou_calculators/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/iou_calculators/iou3d_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/iou_calculators/iou3d_calculator.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/samplers/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/samplers/iou_neg_piecewise_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/samplers/iou_neg_piecewise_sampler.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/base_box3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/base_box3d.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/box_3d_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/box_3d_mode.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/cam_box3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/cam_box3d.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/coord_3d_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/coord_3d_mode.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/depth_box3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/depth_box3d.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/lidar_box3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/lidar_box3d.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/structures/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/structures/utils.py -------------------------------------------------------------------------------- /mmdet3d/core/bbox/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/bbox/transforms.py -------------------------------------------------------------------------------- /mmdet3d/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/evaluation/indoor_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/evaluation/indoor_eval.py -------------------------------------------------------------------------------- /mmdet3d/core/evaluation/kitti_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/evaluation/kitti_utils/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/evaluation/kitti_utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/evaluation/kitti_utils/eval.py -------------------------------------------------------------------------------- /mmdet3d/core/evaluation/kitti_utils/rotate_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/evaluation/kitti_utils/rotate_iou.py -------------------------------------------------------------------------------- /mmdet3d/core/evaluation/seg_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/evaluation/seg_eval.py -------------------------------------------------------------------------------- /mmdet3d/core/evaluation/waymo_utils/prediction_kitti_to_waymo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/evaluation/waymo_utils/prediction_kitti_to_waymo.py -------------------------------------------------------------------------------- /mmdet3d/core/points/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/points/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/points/base_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/points/base_points.py -------------------------------------------------------------------------------- /mmdet3d/core/points/cam_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/points/cam_points.py -------------------------------------------------------------------------------- /mmdet3d/core/points/depth_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/points/depth_points.py -------------------------------------------------------------------------------- /mmdet3d/core/points/lidar_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/points/lidar_points.py -------------------------------------------------------------------------------- /mmdet3d/core/post_processing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/post_processing/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/post_processing/box3d_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/post_processing/box3d_nms.py -------------------------------------------------------------------------------- /mmdet3d/core/post_processing/merge_augs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/post_processing/merge_augs.py -------------------------------------------------------------------------------- /mmdet3d/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/utils/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/utils/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/utils/gaussian.py -------------------------------------------------------------------------------- /mmdet3d/core/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/utils/visualize.py -------------------------------------------------------------------------------- /mmdet3d/core/visualizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/visualizer/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/visualizer/image_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/visualizer/image_vis.py -------------------------------------------------------------------------------- /mmdet3d/core/visualizer/open3d_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/visualizer/open3d_vis.py -------------------------------------------------------------------------------- /mmdet3d/core/visualizer/show_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/visualizer/show_result.py -------------------------------------------------------------------------------- /mmdet3d/core/voxel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/voxel/__init__.py -------------------------------------------------------------------------------- /mmdet3d/core/voxel/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/voxel/builder.py -------------------------------------------------------------------------------- /mmdet3d/core/voxel/voxel_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/core/voxel/voxel_generator.py -------------------------------------------------------------------------------- /mmdet3d/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/__init__.py -------------------------------------------------------------------------------- /mmdet3d/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/builder.py -------------------------------------------------------------------------------- /mmdet3d/datasets/custom_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/custom_3d.py -------------------------------------------------------------------------------- /mmdet3d/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /mmdet3d/datasets/kitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/kitti_dataset.py -------------------------------------------------------------------------------- /mmdet3d/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmdet3d/datasets/pipelines/data_augment_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/pipelines/data_augment_utils.py -------------------------------------------------------------------------------- /mmdet3d/datasets/pipelines/dbsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/pipelines/dbsampler.py -------------------------------------------------------------------------------- /mmdet3d/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /mmdet3d/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /mmdet3d/datasets/pipelines/test_time_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/pipelines/test_time_aug.py -------------------------------------------------------------------------------- /mmdet3d/datasets/pipelines/transforms_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/pipelines/transforms_3d.py -------------------------------------------------------------------------------- /mmdet3d/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/utils.py -------------------------------------------------------------------------------- /mmdet3d/datasets/waymo_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/datasets/waymo_dataset.py -------------------------------------------------------------------------------- /mmdet3d/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/backbones/second.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/backbones/second.py -------------------------------------------------------------------------------- /mmdet3d/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/builder.py -------------------------------------------------------------------------------- /mmdet3d/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/dense_heads/centerpoint_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/dense_heads/centerpoint_head.py -------------------------------------------------------------------------------- /mmdet3d/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/detectors/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/detectors/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/detectors/base.py -------------------------------------------------------------------------------- /mmdet3d/models/detectors/dynamic_voxelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/detectors/dynamic_voxelnet.py -------------------------------------------------------------------------------- /mmdet3d/models/detectors/single_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/detectors/single_stage.py -------------------------------------------------------------------------------- /mmdet3d/models/detectors/voxelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/detectors/voxelnet.py -------------------------------------------------------------------------------- /mmdet3d/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/losses/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/middle_encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/middle_encoders/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/middle_encoders/flatformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/middle_encoders/flatformer.py -------------------------------------------------------------------------------- /mmdet3d/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/necks/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/necks/second_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/necks/second_fpn.py -------------------------------------------------------------------------------- /mmdet3d/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/utils/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/utils/clip_sigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/utils/clip_sigmoid.py -------------------------------------------------------------------------------- /mmdet3d/models/utils/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/utils/mlp.py -------------------------------------------------------------------------------- /mmdet3d/models/voxel_encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/voxel_encoders/__init__.py -------------------------------------------------------------------------------- /mmdet3d/models/voxel_encoders/pillar_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/voxel_encoders/pillar_encoder.py -------------------------------------------------------------------------------- /mmdet3d/models/voxel_encoders/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/voxel_encoders/utils.py -------------------------------------------------------------------------------- /mmdet3d/models/voxel_encoders/voxel_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/models/voxel_encoders/voxel_encoder.py -------------------------------------------------------------------------------- /mmdet3d/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/ball_query/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/ball_query/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/ball_query/ball_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/ball_query/ball_query.py -------------------------------------------------------------------------------- /mmdet3d/ops/ball_query/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/ball_query/src/ball_query.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/ball_query/src/ball_query_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/ball_query/src/ball_query_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/furthest_point_sample/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/furthest_point_sample/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/furthest_point_sample/furthest_point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/furthest_point_sample/furthest_point_sample.py -------------------------------------------------------------------------------- /mmdet3d/ops/furthest_point_sample/points_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/furthest_point_sample/points_sampler.py -------------------------------------------------------------------------------- /mmdet3d/ops/furthest_point_sample/src/furthest_point_sample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/furthest_point_sample/src/furthest_point_sample.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/furthest_point_sample/src/furthest_point_sample_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/furthest_point_sample/src/furthest_point_sample_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/furthest_point_sample/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/furthest_point_sample/utils.py -------------------------------------------------------------------------------- /mmdet3d/ops/gather_points/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/gather_points/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/gather_points/gather_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/gather_points/gather_points.py -------------------------------------------------------------------------------- /mmdet3d/ops/gather_points/src/gather_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/gather_points/src/gather_points.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/gather_points/src/gather_points_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/gather_points/src/gather_points_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/group_points/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/group_points/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/group_points/group_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/group_points/group_points.py -------------------------------------------------------------------------------- /mmdet3d/ops/group_points/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/group_points/src/group_points.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/group_points/src/group_points_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/group_points/src/group_points_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/interpolate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/interpolate/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/interpolate/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/interpolate/src/interpolate.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/interpolate/src/three_interpolate_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/interpolate/src/three_interpolate_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/interpolate/src/three_nn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/interpolate/src/three_nn_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/interpolate/three_interpolate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/interpolate/three_interpolate.py -------------------------------------------------------------------------------- /mmdet3d/ops/interpolate/three_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/interpolate/three_nn.py -------------------------------------------------------------------------------- /mmdet3d/ops/iou3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/iou3d/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/iou3d/iou3d_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/iou3d/iou3d_utils.py -------------------------------------------------------------------------------- /mmdet3d/ops/iou3d/src/iou3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/iou3d/src/iou3d.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/iou3d/src/iou3d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/iou3d/src/iou3d_kernel.cu -------------------------------------------------------------------------------- /mmdet3d/ops/knn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/knn/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/knn/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/knn/knn.py -------------------------------------------------------------------------------- /mmdet3d/ops/knn/src/knn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/knn/src/knn.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/knn/src/knn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/knn/src/knn_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/norm.py -------------------------------------------------------------------------------- /mmdet3d/ops/paconv/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/paconv/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/paconv/assign_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/paconv/assign_score.py -------------------------------------------------------------------------------- /mmdet3d/ops/paconv/paconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/paconv/paconv.py -------------------------------------------------------------------------------- /mmdet3d/ops/paconv/src/assign_score_withk.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/paconv/src/assign_score_withk.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/paconv/src/assign_score_withk_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/paconv/src/assign_score_withk_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/paconv/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/paconv/utils.py -------------------------------------------------------------------------------- /mmdet3d/ops/pointnet_modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/pointnet_modules/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/pointnet_modules/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/pointnet_modules/builder.py -------------------------------------------------------------------------------- /mmdet3d/ops/pointnet_modules/paconv_sa_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/pointnet_modules/paconv_sa_module.py -------------------------------------------------------------------------------- /mmdet3d/ops/pointnet_modules/point_fp_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/pointnet_modules/point_fp_module.py -------------------------------------------------------------------------------- /mmdet3d/ops/pointnet_modules/point_sa_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/pointnet_modules/point_sa_module.py -------------------------------------------------------------------------------- /mmdet3d/ops/roiaware_pool3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/roiaware_pool3d/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/roiaware_pool3d/points_in_boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/roiaware_pool3d/points_in_boxes.py -------------------------------------------------------------------------------- /mmdet3d/ops/roiaware_pool3d/roiaware_pool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/roiaware_pool3d/roiaware_pool3d.py -------------------------------------------------------------------------------- /mmdet3d/ops/roiaware_pool3d/src/points_in_boxes_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/roiaware_pool3d/src/points_in_boxes_cpu.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/roiaware_pool3d/src/points_in_boxes_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/roiaware_pool3d/src/points_in_boxes_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/roiaware_pool3d/src/roiaware_pool3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/roiaware_pool3d/src/roiaware_pool3d.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/roiaware_pool3d/src/roiaware_pool3d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/roiaware_pool3d/src/roiaware_pool3d_kernel.cu -------------------------------------------------------------------------------- /mmdet3d/ops/sparse_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/sparse_block.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/conv.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/functional.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/paramsgrid.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/paramsgrid.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/prettyprint.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/prettyprint.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/pybind11_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/pybind11_utils.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/fused_spconv_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/fused_spconv_ops.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/geometry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/geometry.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/indice.cu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/indice.cu.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/indice.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/indice.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/maxpool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/maxpool.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/mp_helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/mp_helper.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/point2voxel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/point2voxel.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/pool_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/pool_ops.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/reordering.cu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/reordering.cu.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/reordering.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/reordering.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/spconv/spconv_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/spconv/spconv_ops.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/tensorview/helper_kernel.cu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/tensorview/helper_kernel.cu.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/tensorview/helper_launch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/tensorview/helper_launch.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/tensorview/tensorview.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/tensorview/tensorview.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/torch_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/torch_utils.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/include/utility/timer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/include/utility/timer.h -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/modules.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/ops.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/overwrite_spconv/write_spconv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/overwrite_spconv/write_spconv2.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/pool.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/src/all.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/src/all.cc -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/src/indice.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/src/indice.cc -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/src/indice_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/src/indice_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/src/maxpool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/src/maxpool.cc -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/src/maxpool_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/src/maxpool_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/src/reordering.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/src/reordering.cc -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/src/reordering_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/src/reordering_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/structure.py -------------------------------------------------------------------------------- /mmdet3d/ops/spconv/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/spconv/test_utils.py -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/__init__.py -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/scatter_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/scatter_points.py -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/src/scatter_points_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/src/scatter_points_cpu.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/src/scatter_points_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/src/scatter_points_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/src/voxelization.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/src/voxelization.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/src/voxelization.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/src/voxelization.h -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/src/voxelization_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/src/voxelization_cpu.cpp -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/src/voxelization_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/src/voxelization_cuda.cu -------------------------------------------------------------------------------- /mmdet3d/ops/voxel/voxelize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/ops/voxel/voxelize.py -------------------------------------------------------------------------------- /mmdet3d/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/runner/__init__.py -------------------------------------------------------------------------------- /mmdet3d/runner/epoch_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/runner/epoch_based_runner.py -------------------------------------------------------------------------------- /mmdet3d/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/utils/__init__.py -------------------------------------------------------------------------------- /mmdet3d/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/utils/collect_env.py -------------------------------------------------------------------------------- /mmdet3d/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/utils/logger.py -------------------------------------------------------------------------------- /mmdet3d/utils/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/utils/syncbn.py -------------------------------------------------------------------------------- /mmdet3d/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/mmdet3d/version.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/setup.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | from .data_converter import * 2 | -------------------------------------------------------------------------------- /tools/analysis_tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/analysis_tools/analyze_logs.py -------------------------------------------------------------------------------- /tools/analysis_tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/analysis_tools/benchmark.py -------------------------------------------------------------------------------- /tools/analyze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/analyze.py -------------------------------------------------------------------------------- /tools/analyze.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/analyze.sh -------------------------------------------------------------------------------- /tools/create_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/create_data.py -------------------------------------------------------------------------------- /tools/create_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/create_data.sh -------------------------------------------------------------------------------- /tools/create_submission: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/create_submission -------------------------------------------------------------------------------- /tools/data_converter/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/data_converter/create_gt_database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/create_gt_database.py -------------------------------------------------------------------------------- /tools/data_converter/indoor_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/indoor_converter.py -------------------------------------------------------------------------------- /tools/data_converter/kitti_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/kitti_converter.py -------------------------------------------------------------------------------- /tools/data_converter/kitti_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/kitti_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/lyft_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/lyft_converter.py -------------------------------------------------------------------------------- /tools/data_converter/lyft_data_fixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/lyft_data_fixer.py -------------------------------------------------------------------------------- /tools/data_converter/nuimage_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/nuimage_converter.py -------------------------------------------------------------------------------- /tools/data_converter/nuscenes_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/nuscenes_converter.py -------------------------------------------------------------------------------- /tools/data_converter/s3dis_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/s3dis_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/scannet_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/scannet_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/sunrgbd_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/sunrgbd_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/waymo_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/data_converter/waymo_converter.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/get_watmo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/get_watmo.sh -------------------------------------------------------------------------------- /tools/misc/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/misc/browse_dataset.py -------------------------------------------------------------------------------- /tools/misc/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/misc/fuse_conv_bn.py -------------------------------------------------------------------------------- /tools/misc/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/misc/print_config.py -------------------------------------------------------------------------------- /tools/misc/visualize_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/misc/visualize_results.py -------------------------------------------------------------------------------- /tools/model_converters/convert_votenet_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/model_converters/convert_votenet_checkpoints.py -------------------------------------------------------------------------------- /tools/model_converters/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/model_converters/publish_model.py -------------------------------------------------------------------------------- /tools/model_converters/regnet2mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/model_converters/regnet2mmdet.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/flatformer/HEAD/tools/train.py --------------------------------------------------------------------------------