├── .gitignore ├── Dockerfile ├── LICENSE ├── NUSCENES-GUIDE.md ├── README.md ├── config ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ └── config.cpython-37.pyc ├── config.py ├── label_mapping │ ├── nuscenes.yaml │ ├── semantic-kitti-multiscan.yaml │ └── semantic-kitti.yaml ├── nuScenes.yaml └── semantickitti.yaml ├── dataloader ├── __init__.py ├── dataset_nuscenes.py ├── dataset_semantickitti.py └── pc_dataset.py ├── demo_folder.py ├── img ├── leaderboard.png ├── leaderboard2.png └── pipeline.png ├── network ├── __init__.py ├── cylinder_fea_generator.py ├── cylinder_spconv_3d.py └── segmentator_3d_asymm_spconv.py ├── train.sh ├── train_cylinder_asym.py ├── train_cylinder_asym_nuscenes.py ├── train_nusc.sh └── utils ├── __init__.py ├── load_save_util.py ├── log_util.py ├── lovasz_losses.py └── metric_util.py /.gitignore: -------------------------------------------------------------------------------- 1 | # Created by .ignore support plugin (hsz.mobi) 2 | .idea 3 | .DS_Store -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/LICENSE -------------------------------------------------------------------------------- /NUSCENES-GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/NUSCENES-GUIDE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /config/__pycache__/config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/__pycache__/config.cpython-37.pyc -------------------------------------------------------------------------------- /config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/config.py -------------------------------------------------------------------------------- /config/label_mapping/nuscenes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/label_mapping/nuscenes.yaml -------------------------------------------------------------------------------- /config/label_mapping/semantic-kitti-multiscan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/label_mapping/semantic-kitti-multiscan.yaml -------------------------------------------------------------------------------- /config/label_mapping/semantic-kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/label_mapping/semantic-kitti.yaml -------------------------------------------------------------------------------- /config/nuScenes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/nuScenes.yaml -------------------------------------------------------------------------------- /config/semantickitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/config/semantickitti.yaml -------------------------------------------------------------------------------- /dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/dataloader/__init__.py -------------------------------------------------------------------------------- /dataloader/dataset_nuscenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/dataloader/dataset_nuscenes.py -------------------------------------------------------------------------------- /dataloader/dataset_semantickitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/dataloader/dataset_semantickitti.py -------------------------------------------------------------------------------- /dataloader/pc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/dataloader/pc_dataset.py -------------------------------------------------------------------------------- /demo_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/demo_folder.py -------------------------------------------------------------------------------- /img/leaderboard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/img/leaderboard.png -------------------------------------------------------------------------------- /img/leaderboard2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/img/leaderboard2.png -------------------------------------------------------------------------------- /img/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/img/pipeline.png -------------------------------------------------------------------------------- /network/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/network/__init__.py -------------------------------------------------------------------------------- /network/cylinder_fea_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/network/cylinder_fea_generator.py -------------------------------------------------------------------------------- /network/cylinder_spconv_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/network/cylinder_spconv_3d.py -------------------------------------------------------------------------------- /network/segmentator_3d_asymm_spconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/network/segmentator_3d_asymm_spconv.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/train.sh -------------------------------------------------------------------------------- /train_cylinder_asym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/train_cylinder_asym.py -------------------------------------------------------------------------------- /train_cylinder_asym_nuscenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/train_cylinder_asym_nuscenes.py -------------------------------------------------------------------------------- /train_nusc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/train_nusc.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/load_save_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/utils/load_save_util.py -------------------------------------------------------------------------------- /utils/log_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/utils/log_util.py -------------------------------------------------------------------------------- /utils/lovasz_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/utils/lovasz_losses.py -------------------------------------------------------------------------------- /utils/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinge008/Cylinder3D/HEAD/utils/metric_util.py --------------------------------------------------------------------------------