├── Dockerfile ├── README.md ├── configs ├── OccFusion.py ├── OccFusion_occ3d.py ├── OccFusion_openoccupancy.py ├── OccFusion_semanticKITTI.py └── _base_ │ └── default_runtime.py ├── docs ├── dataset.md ├── getting_started.md └── install.md ├── figs ├── nuscene_night.png ├── nuscene_rainy.png ├── nuscene_whole.png ├── overrall.jpg ├── range_vari.png ├── semanticKITTI.png └── vis.png ├── occfusion ├── __init__.py ├── bottleneckaspp.py ├── custom_pack.py ├── data_preprocessor.py ├── efficientvitblock.py ├── evaluate.py ├── fusion.py ├── loading.py ├── loss.py ├── main.py ├── multi_scale_inverse_matrixVT.py ├── nuscenes_dataset.py ├── occhead.py ├── semantickitti_dataset.py ├── svfe.py └── window_attention.py └── tools ├── analysis_tools ├── analyze_logs.py ├── benchmark.py └── get_flops.py ├── create_data.py ├── create_data.sh ├── dataset_converters ├── create_gt_database.py ├── indoor_converter.py ├── kitti_converter.py ├── kitti_data_utils.py ├── kitti_unzip.sh ├── lyft_converter.py ├── lyft_data_fixer.py ├── nuimage_converter.py ├── nuscenes_converter.py ├── nuscenes_unzip.sh ├── s3dis_data_utils.py ├── scannet_data_utils.py ├── semantickitti_converter.py ├── semantickitti_unzip.sh ├── sunrgbd_data_utils.py ├── update_infos_to_v2.py └── waymo_converter.py ├── deployment ├── mmdet3d2torchserve.py ├── mmdet3d_handler.py └── test_torchserver.py ├── dist_test.sh ├── dist_train.sh ├── get_params.py ├── misc ├── browse_dataset.py ├── fuse_conv_bn.py ├── print_config.py └── visualize_results.py ├── model_converters ├── convert_h3dnet_checkpoints.py ├── convert_votenet_checkpoints.py ├── publish_model.py └── regnet2mmdet.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py ├── train.py ├── update_data_coords.py └── update_data_coords.sh /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/Dockerfile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/README.md -------------------------------------------------------------------------------- /configs/OccFusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/configs/OccFusion.py -------------------------------------------------------------------------------- /configs/OccFusion_occ3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/configs/OccFusion_occ3d.py -------------------------------------------------------------------------------- /configs/OccFusion_openoccupancy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/configs/OccFusion_openoccupancy.py -------------------------------------------------------------------------------- /configs/OccFusion_semanticKITTI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/configs/OccFusion_semanticKITTI.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /docs/dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/docs/dataset.md -------------------------------------------------------------------------------- /docs/getting_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/docs/getting_started.md -------------------------------------------------------------------------------- /docs/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/docs/install.md -------------------------------------------------------------------------------- /figs/nuscene_night.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/figs/nuscene_night.png -------------------------------------------------------------------------------- /figs/nuscene_rainy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/figs/nuscene_rainy.png -------------------------------------------------------------------------------- /figs/nuscene_whole.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/figs/nuscene_whole.png -------------------------------------------------------------------------------- /figs/overrall.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/figs/overrall.jpg -------------------------------------------------------------------------------- /figs/range_vari.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/figs/range_vari.png -------------------------------------------------------------------------------- /figs/semanticKITTI.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/figs/semanticKITTI.png -------------------------------------------------------------------------------- /figs/vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/figs/vis.png -------------------------------------------------------------------------------- /occfusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/__init__.py -------------------------------------------------------------------------------- /occfusion/bottleneckaspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/bottleneckaspp.py -------------------------------------------------------------------------------- /occfusion/custom_pack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/custom_pack.py -------------------------------------------------------------------------------- /occfusion/data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/data_preprocessor.py -------------------------------------------------------------------------------- /occfusion/efficientvitblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/efficientvitblock.py -------------------------------------------------------------------------------- /occfusion/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/evaluate.py -------------------------------------------------------------------------------- /occfusion/fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/fusion.py -------------------------------------------------------------------------------- /occfusion/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/loading.py -------------------------------------------------------------------------------- /occfusion/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/loss.py -------------------------------------------------------------------------------- /occfusion/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/main.py -------------------------------------------------------------------------------- /occfusion/multi_scale_inverse_matrixVT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/multi_scale_inverse_matrixVT.py -------------------------------------------------------------------------------- /occfusion/nuscenes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/nuscenes_dataset.py -------------------------------------------------------------------------------- /occfusion/occhead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/occhead.py -------------------------------------------------------------------------------- /occfusion/semantickitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/semantickitti_dataset.py -------------------------------------------------------------------------------- /occfusion/svfe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/svfe.py -------------------------------------------------------------------------------- /occfusion/window_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/occfusion/window_attention.py -------------------------------------------------------------------------------- /tools/analysis_tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/analysis_tools/analyze_logs.py -------------------------------------------------------------------------------- /tools/analysis_tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/analysis_tools/benchmark.py -------------------------------------------------------------------------------- /tools/analysis_tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/analysis_tools/get_flops.py -------------------------------------------------------------------------------- /tools/create_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/create_data.py -------------------------------------------------------------------------------- /tools/create_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/create_data.sh -------------------------------------------------------------------------------- /tools/dataset_converters/create_gt_database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/create_gt_database.py -------------------------------------------------------------------------------- /tools/dataset_converters/indoor_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/indoor_converter.py -------------------------------------------------------------------------------- /tools/dataset_converters/kitti_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/kitti_converter.py -------------------------------------------------------------------------------- /tools/dataset_converters/kitti_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/kitti_data_utils.py -------------------------------------------------------------------------------- /tools/dataset_converters/kitti_unzip.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/kitti_unzip.sh -------------------------------------------------------------------------------- /tools/dataset_converters/lyft_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/lyft_converter.py -------------------------------------------------------------------------------- /tools/dataset_converters/lyft_data_fixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/lyft_data_fixer.py -------------------------------------------------------------------------------- /tools/dataset_converters/nuimage_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/nuimage_converter.py -------------------------------------------------------------------------------- /tools/dataset_converters/nuscenes_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/nuscenes_converter.py -------------------------------------------------------------------------------- /tools/dataset_converters/nuscenes_unzip.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/nuscenes_unzip.sh -------------------------------------------------------------------------------- /tools/dataset_converters/s3dis_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/s3dis_data_utils.py -------------------------------------------------------------------------------- /tools/dataset_converters/scannet_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/scannet_data_utils.py -------------------------------------------------------------------------------- /tools/dataset_converters/semantickitti_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/semantickitti_converter.py -------------------------------------------------------------------------------- /tools/dataset_converters/semantickitti_unzip.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/semantickitti_unzip.sh -------------------------------------------------------------------------------- /tools/dataset_converters/sunrgbd_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/sunrgbd_data_utils.py -------------------------------------------------------------------------------- /tools/dataset_converters/update_infos_to_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/update_infos_to_v2.py -------------------------------------------------------------------------------- /tools/dataset_converters/waymo_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dataset_converters/waymo_converter.py -------------------------------------------------------------------------------- /tools/deployment/mmdet3d2torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/deployment/mmdet3d2torchserve.py -------------------------------------------------------------------------------- /tools/deployment/mmdet3d_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/deployment/mmdet3d_handler.py -------------------------------------------------------------------------------- /tools/deployment/test_torchserver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/deployment/test_torchserver.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/get_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/get_params.py -------------------------------------------------------------------------------- /tools/misc/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/misc/browse_dataset.py -------------------------------------------------------------------------------- /tools/misc/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/misc/fuse_conv_bn.py -------------------------------------------------------------------------------- /tools/misc/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/misc/print_config.py -------------------------------------------------------------------------------- /tools/misc/visualize_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/misc/visualize_results.py -------------------------------------------------------------------------------- /tools/model_converters/convert_h3dnet_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/model_converters/convert_h3dnet_checkpoints.py -------------------------------------------------------------------------------- /tools/model_converters/convert_votenet_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/model_converters/convert_votenet_checkpoints.py -------------------------------------------------------------------------------- /tools/model_converters/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/model_converters/publish_model.py -------------------------------------------------------------------------------- /tools/model_converters/regnet2mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/model_converters/regnet2mmdet.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/update_data_coords.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/update_data_coords.py -------------------------------------------------------------------------------- /tools/update_data_coords.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DanielMing123/OCCFusion/HEAD/tools/update_data_coords.sh --------------------------------------------------------------------------------