├── .gitignore ├── README.md ├── config ├── CMDFusion-nuscenese-g-L.yaml ├── CMDFusion-semantickitti-g-all-ft.yaml ├── CMDFusion-semantickitti-g.yaml └── label_mapping │ ├── nuscenes-RCF.yaml │ ├── nuscenes.yaml │ ├── semantic-kitti-RCF.yaml │ ├── semantic-kitti-all.yaml │ ├── semantic-kitti-small.yaml │ ├── semantic-kitti-train.yaml │ └── semantic-kitti.yaml ├── dataloader ├── dataset.py └── pc_dataset.py ├── main.py ├── network ├── arch_2dpass.py ├── arch_cmd_fusion.py ├── base_model.py ├── base_model_distil.py ├── basic_block.py ├── spvcnn.py ├── spvcnn_g.py └── voxel_fea_generator.py ├── overview.png ├── requirements.txt └── utils ├── bar.py ├── bar_u.py ├── cp_weak_label_generate.py ├── instance_aug.py ├── instance_aug_bank.py ├── lovasz_loss.py ├── metric_util.py ├── read_weak_label_generate.py ├── schedulers.py ├── tsne.py ├── tsne_bi.py ├── turbo_cmap.py ├── validate_nus.py ├── vis_utils.py └── weak_label_generate.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/README.md -------------------------------------------------------------------------------- /config/CMDFusion-nuscenese-g-L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/CMDFusion-nuscenese-g-L.yaml -------------------------------------------------------------------------------- /config/CMDFusion-semantickitti-g-all-ft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/CMDFusion-semantickitti-g-all-ft.yaml -------------------------------------------------------------------------------- /config/CMDFusion-semantickitti-g.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/CMDFusion-semantickitti-g.yaml -------------------------------------------------------------------------------- /config/label_mapping/nuscenes-RCF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/label_mapping/nuscenes-RCF.yaml -------------------------------------------------------------------------------- /config/label_mapping/nuscenes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/label_mapping/nuscenes.yaml -------------------------------------------------------------------------------- /config/label_mapping/semantic-kitti-RCF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/label_mapping/semantic-kitti-RCF.yaml -------------------------------------------------------------------------------- /config/label_mapping/semantic-kitti-all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/label_mapping/semantic-kitti-all.yaml -------------------------------------------------------------------------------- /config/label_mapping/semantic-kitti-small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/label_mapping/semantic-kitti-small.yaml -------------------------------------------------------------------------------- /config/label_mapping/semantic-kitti-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/label_mapping/semantic-kitti-train.yaml -------------------------------------------------------------------------------- /config/label_mapping/semantic-kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/config/label_mapping/semantic-kitti.yaml -------------------------------------------------------------------------------- /dataloader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/dataloader/dataset.py -------------------------------------------------------------------------------- /dataloader/pc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/dataloader/pc_dataset.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/main.py -------------------------------------------------------------------------------- /network/arch_2dpass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/arch_2dpass.py -------------------------------------------------------------------------------- /network/arch_cmd_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/arch_cmd_fusion.py -------------------------------------------------------------------------------- /network/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/base_model.py -------------------------------------------------------------------------------- /network/base_model_distil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/base_model_distil.py -------------------------------------------------------------------------------- /network/basic_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/basic_block.py -------------------------------------------------------------------------------- /network/spvcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/spvcnn.py -------------------------------------------------------------------------------- /network/spvcnn_g.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/spvcnn_g.py -------------------------------------------------------------------------------- /network/voxel_fea_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/network/voxel_fea_generator.py -------------------------------------------------------------------------------- /overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/overview.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pytorch_lightning 2 | nuScenes-devkit 3 | spconv-cu113 4 | easydict 5 | thop -------------------------------------------------------------------------------- /utils/bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/bar.py -------------------------------------------------------------------------------- /utils/bar_u.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/bar_u.py -------------------------------------------------------------------------------- /utils/cp_weak_label_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/cp_weak_label_generate.py -------------------------------------------------------------------------------- /utils/instance_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/instance_aug.py -------------------------------------------------------------------------------- /utils/instance_aug_bank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/instance_aug_bank.py -------------------------------------------------------------------------------- /utils/lovasz_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/lovasz_loss.py -------------------------------------------------------------------------------- /utils/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/metric_util.py -------------------------------------------------------------------------------- /utils/read_weak_label_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/read_weak_label_generate.py -------------------------------------------------------------------------------- /utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/schedulers.py -------------------------------------------------------------------------------- /utils/tsne.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/tsne.py -------------------------------------------------------------------------------- /utils/tsne_bi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/tsne_bi.py -------------------------------------------------------------------------------- /utils/turbo_cmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/turbo_cmap.py -------------------------------------------------------------------------------- /utils/validate_nus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/validate_nus.py -------------------------------------------------------------------------------- /utils/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/vis_utils.py -------------------------------------------------------------------------------- /utils/weak_label_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jun-CEN/CMDFusion/HEAD/utils/weak_label_generate.py --------------------------------------------------------------------------------