├── .gitignore ├── README.md ├── config ├── __init__.py ├── config_cpgnet_adam_bili_sample_ohem_fp16.py └── config_cpgnet_sgd_bili_sample_ohem_fp16.py ├── data └── nuscenes ├── datasets ├── __init__.py ├── data.py ├── nuscenes.yaml └── utils.py ├── evaluate.py ├── models ├── __init__.py └── cpgnet.py ├── networks ├── __init__.py ├── backbone.py ├── bird_view.py └── range_view.py ├── pipeline.sh ├── pytorch_lib ├── __init__.py └── src │ ├── atomics.cuh │ ├── point_deep.cpp │ ├── point_deep_cuda.cpp │ └── point_deep_cuda_kernel.cu ├── setup.py ├── train.py └── utils ├── __init__.py ├── builder.py ├── criterion.py ├── logger.py ├── lovasz_losses.py └── metric.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/config_cpgnet_adam_bili_sample_ohem_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/config/config_cpgnet_adam_bili_sample_ohem_fp16.py -------------------------------------------------------------------------------- /config/config_cpgnet_sgd_bili_sample_ohem_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/config/config_cpgnet_sgd_bili_sample_ohem_fp16.py -------------------------------------------------------------------------------- /data/nuscenes: -------------------------------------------------------------------------------- 1 | /mnt/nuscenes/ 2 | -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | from . import data -------------------------------------------------------------------------------- /datasets/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/datasets/data.py -------------------------------------------------------------------------------- /datasets/nuscenes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/datasets/nuscenes.yaml -------------------------------------------------------------------------------- /datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/datasets/utils.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/evaluate.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from . import cpgnet 2 | -------------------------------------------------------------------------------- /models/cpgnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/models/cpgnet.py -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/networks/__init__.py -------------------------------------------------------------------------------- /networks/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/networks/backbone.py -------------------------------------------------------------------------------- /networks/bird_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/networks/bird_view.py -------------------------------------------------------------------------------- /networks/range_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/networks/range_view.py -------------------------------------------------------------------------------- /pipeline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/pipeline.sh -------------------------------------------------------------------------------- /pytorch_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/pytorch_lib/__init__.py -------------------------------------------------------------------------------- /pytorch_lib/src/atomics.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/pytorch_lib/src/atomics.cuh -------------------------------------------------------------------------------- /pytorch_lib/src/point_deep.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/pytorch_lib/src/point_deep.cpp -------------------------------------------------------------------------------- /pytorch_lib/src/point_deep_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/pytorch_lib/src/point_deep_cuda.cpp -------------------------------------------------------------------------------- /pytorch_lib/src/point_deep_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/pytorch_lib/src/point_deep_cuda_kernel.cu -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/setup.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/utils/builder.py -------------------------------------------------------------------------------- /utils/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/utils/criterion.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/lovasz_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/utils/lovasz_losses.py -------------------------------------------------------------------------------- /utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GangZhang842/CPGNet/HEAD/utils/metric.py --------------------------------------------------------------------------------