├── .gitignore ├── LICENSE.md ├── Makefile ├── README.md ├── docker └── Dockerfile-mmlab-cu111 ├── docs └── INSTALL.md ├── media └── framework.png ├── projects ├── configs │ └── vedet_vovnet_p4_1600x640_2vview_2frame.py └── mmdet3d_plugin │ ├── __init__.py │ ├── core │ └── bbox │ │ ├── array_converter.py │ │ ├── assigners │ │ ├── __init__.py │ │ └── hungarian_assigner_3d.py │ │ ├── coders │ │ ├── __init__.py │ │ └── nms_free_coder.py │ │ ├── iou_calculators │ │ ├── __init__.py │ │ └── iou3d_calculator.py │ │ ├── match_costs │ │ ├── __init__.py │ │ └── match_cost.py │ │ └── util.py │ ├── datasets │ ├── __init__.py │ ├── nuscenes_dataset.py │ └── pipelines │ │ ├── __init__.py │ │ ├── loading.py │ │ └── transform_3d.py │ └── models │ ├── backbones │ ├── __init__.py │ ├── convnext.py │ ├── vovnet.py │ └── vovnetcp.py │ ├── dense_heads │ ├── __init__.py │ └── vedet_head.py │ ├── detectors │ ├── __init__.py │ └── vedet.py │ ├── losses │ ├── Sigmoid_ce_loss.py │ └── __init__.py │ ├── necks │ ├── __init__.py │ └── cp_fpn.py │ └── utils │ ├── __init__.py │ ├── grid_mask.py │ ├── petr_transformer.py │ ├── positional_encoding.py │ └── vedet_transformer.py └── tools ├── analysis_tools ├── analyze_logs.py ├── benchmark.py └── get_flops.py ├── build-dataset.py ├── create_data.py ├── create_data.sh ├── 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 ├── map_api.py ├── nuimage_converter.py ├── nuscenes_converter.py ├── nuscenes_converter_seg.py ├── s3dis_data_utils.py ├── scannet_data_utils.py ├── sunrgbd_data_utils.py └── waymo_converter.py ├── dist_test.sh ├── dist_train.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 ├── rename_depth_cache.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/LICENSE.md -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/README.md -------------------------------------------------------------------------------- /docker/Dockerfile-mmlab-cu111: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/docker/Dockerfile-mmlab-cu111 -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /media/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/media/framework.png -------------------------------------------------------------------------------- /projects/configs/vedet_vovnet_p4_1600x640_2vview_2frame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/configs/vedet_vovnet_p4_1600x640_2vview_2frame.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/array_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/array_converter.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/assigners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/assigners/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_3d.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/coders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/coders/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/coders/nms_free_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/coders/nms_free_coder.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/iou_calculators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/iou_calculators/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/iou_calculators/iou3d_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/iou_calculators/iou3d_calculator.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/match_costs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/match_costs/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/core/bbox/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/core/bbox/util.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/datasets/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/datasets/nuscenes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/datasets/nuscenes_dataset.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/backbones/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/backbones/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/backbones/convnext.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/backbones/vovnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/backbones/vovnet.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/backbones/vovnetcp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/backbones/vovnetcp.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/dense_heads/vedet_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/dense_heads/vedet_head.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/detectors/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/detectors/vedet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/detectors/vedet.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/losses/Sigmoid_ce_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/losses/Sigmoid_ce_loss.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/losses/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/necks/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/necks/cp_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/necks/cp_fpn.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/utils/__init__.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/utils/grid_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/utils/grid_mask.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/utils/petr_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/utils/petr_transformer.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/utils/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/utils/positional_encoding.py -------------------------------------------------------------------------------- /projects/mmdet3d_plugin/models/utils/vedet_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/projects/mmdet3d_plugin/models/utils/vedet_transformer.py -------------------------------------------------------------------------------- /tools/analysis_tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/analysis_tools/analyze_logs.py -------------------------------------------------------------------------------- /tools/analysis_tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/analysis_tools/benchmark.py -------------------------------------------------------------------------------- /tools/analysis_tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/analysis_tools/get_flops.py -------------------------------------------------------------------------------- /tools/build-dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/build-dataset.py -------------------------------------------------------------------------------- /tools/create_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/create_data.py -------------------------------------------------------------------------------- /tools/create_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/create_data.sh -------------------------------------------------------------------------------- /tools/data_converter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/__init__.py -------------------------------------------------------------------------------- /tools/data_converter/create_gt_database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/create_gt_database.py -------------------------------------------------------------------------------- /tools/data_converter/indoor_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/indoor_converter.py -------------------------------------------------------------------------------- /tools/data_converter/kitti_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/kitti_converter.py -------------------------------------------------------------------------------- /tools/data_converter/kitti_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/kitti_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/lyft_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/lyft_converter.py -------------------------------------------------------------------------------- /tools/data_converter/lyft_data_fixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/lyft_data_fixer.py -------------------------------------------------------------------------------- /tools/data_converter/map_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/map_api.py -------------------------------------------------------------------------------- /tools/data_converter/nuimage_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/nuimage_converter.py -------------------------------------------------------------------------------- /tools/data_converter/nuscenes_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/nuscenes_converter.py -------------------------------------------------------------------------------- /tools/data_converter/nuscenes_converter_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/nuscenes_converter_seg.py -------------------------------------------------------------------------------- /tools/data_converter/s3dis_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/s3dis_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/scannet_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/scannet_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/sunrgbd_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/sunrgbd_data_utils.py -------------------------------------------------------------------------------- /tools/data_converter/waymo_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/data_converter/waymo_converter.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/misc/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/misc/browse_dataset.py -------------------------------------------------------------------------------- /tools/misc/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/misc/fuse_conv_bn.py -------------------------------------------------------------------------------- /tools/misc/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/misc/print_config.py -------------------------------------------------------------------------------- /tools/misc/visualize_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/misc/visualize_results.py -------------------------------------------------------------------------------- /tools/model_converters/convert_votenet_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/model_converters/convert_votenet_checkpoints.py -------------------------------------------------------------------------------- /tools/model_converters/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/model_converters/publish_model.py -------------------------------------------------------------------------------- /tools/model_converters/regnet2mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/model_converters/regnet2mmdet.py -------------------------------------------------------------------------------- /tools/rename_depth_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/rename_depth_cache.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TRI-ML/VEDet/HEAD/tools/train.py --------------------------------------------------------------------------------