├── .gitignore ├── README.md ├── config ├── __init__.py └── semantickitti │ └── config_mvfcev2ctx_adam_wce_lossv2_single.py ├── data ├── SemanticKITTI └── object_bank_semkitti ├── datasets ├── __init__.py ├── copy_paste.py ├── data.py ├── semantic-kitti.yaml └── utils.py ├── evaluate.py ├── imgs ├── acc_vs_speed.PNG └── framework.PNG ├── light_trainer ├── __init__.py ├── recorder │ ├── __init__.py │ ├── base_recorder.py │ └── txt_recorder.py └── trainer.py ├── models ├── __init__.py ├── cfnet.py ├── common_utils.py ├── loss │ ├── __init__.py │ ├── criterion.py │ └── lovasz_losses.py ├── model_runner.py ├── model_utils │ ├── __init__.py │ ├── builder.py │ └── metric.py └── networks │ ├── __init__.py │ ├── backbone.py │ ├── fcn_2d.py │ ├── fpn.py │ ├── fusion_module.py │ ├── point2voxel.py │ └── voxel2point.py ├── pytorch_lib ├── __init__.py ├── panoptic_lib.py ├── setup.py └── src │ ├── atomics.cuh │ ├── pt_lib_cuda.cpp │ └── pt_lib_cuda_kernel.cu ├── requirements.txt ├── train.py └── utils ├── __init__.py └── config_parser.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/semantickitti/config_mvfcev2ctx_adam_wce_lossv2_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/config/semantickitti/config_mvfcev2ctx_adam_wce_lossv2_single.py -------------------------------------------------------------------------------- /data/SemanticKITTI: -------------------------------------------------------------------------------- 1 | /perception-hl/james.zhang7/dataset/SemanticKITTI -------------------------------------------------------------------------------- /data/object_bank_semkitti: -------------------------------------------------------------------------------- 1 | /perception-hl/james.zhang7/dataset/SemanticKITTI/object_bank_semkitti -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | from . import data -------------------------------------------------------------------------------- /datasets/copy_paste.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/datasets/copy_paste.py -------------------------------------------------------------------------------- /datasets/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/datasets/data.py -------------------------------------------------------------------------------- /datasets/semantic-kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/datasets/semantic-kitti.yaml -------------------------------------------------------------------------------- /datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/datasets/utils.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/evaluate.py -------------------------------------------------------------------------------- /imgs/acc_vs_speed.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/imgs/acc_vs_speed.PNG -------------------------------------------------------------------------------- /imgs/framework.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/imgs/framework.PNG -------------------------------------------------------------------------------- /light_trainer/__init__.py: -------------------------------------------------------------------------------- 1 | from . import recorder, trainer -------------------------------------------------------------------------------- /light_trainer/recorder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/light_trainer/recorder/__init__.py -------------------------------------------------------------------------------- /light_trainer/recorder/base_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/light_trainer/recorder/base_recorder.py -------------------------------------------------------------------------------- /light_trainer/recorder/txt_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/light_trainer/recorder/txt_recorder.py -------------------------------------------------------------------------------- /light_trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/light_trainer/trainer.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/cfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/cfnet.py -------------------------------------------------------------------------------- /models/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/common_utils.py -------------------------------------------------------------------------------- /models/loss/__init__.py: -------------------------------------------------------------------------------- 1 | from . import criterion -------------------------------------------------------------------------------- /models/loss/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/loss/criterion.py -------------------------------------------------------------------------------- /models/loss/lovasz_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/loss/lovasz_losses.py -------------------------------------------------------------------------------- /models/model_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/model_runner.py -------------------------------------------------------------------------------- /models/model_utils/__init__.py: -------------------------------------------------------------------------------- 1 | from . import builder, metric -------------------------------------------------------------------------------- /models/model_utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/model_utils/builder.py -------------------------------------------------------------------------------- /models/model_utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/model_utils/metric.py -------------------------------------------------------------------------------- /models/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/networks/__init__.py -------------------------------------------------------------------------------- /models/networks/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/networks/backbone.py -------------------------------------------------------------------------------- /models/networks/fcn_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/networks/fcn_2d.py -------------------------------------------------------------------------------- /models/networks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/networks/fpn.py -------------------------------------------------------------------------------- /models/networks/fusion_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/networks/fusion_module.py -------------------------------------------------------------------------------- /models/networks/point2voxel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/networks/point2voxel.py -------------------------------------------------------------------------------- /models/networks/voxel2point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/models/networks/voxel2point.py -------------------------------------------------------------------------------- /pytorch_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/pytorch_lib/__init__.py -------------------------------------------------------------------------------- /pytorch_lib/panoptic_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/pytorch_lib/panoptic_lib.py -------------------------------------------------------------------------------- /pytorch_lib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/pytorch_lib/setup.py -------------------------------------------------------------------------------- /pytorch_lib/src/atomics.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/pytorch_lib/src/atomics.cuh -------------------------------------------------------------------------------- /pytorch_lib/src/pt_lib_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/pytorch_lib/src/pt_lib_cuda.cpp -------------------------------------------------------------------------------- /pytorch_lib/src/pt_lib_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/pytorch_lib/src/pt_lib_cuda_kernel.cu -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | from . import config_parser -------------------------------------------------------------------------------- /utils/config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CFNet/HEAD/utils/config_parser.py --------------------------------------------------------------------------------