├── .gitignore ├── LICENSE ├── README.md ├── assets └── main.png ├── config ├── _base_ │ ├── base_res101_bs16xep100.py │ ├── openlane_1000_base_res101_bs16xep100_lidar.py │ └── optimizer.py └── release_iclr │ └── dv3dlane_openlane1000_base.py ├── data ├── Load_Data.py ├── __init__.py ├── data_utils.py ├── lane_transform.py ├── lidar_utils.py ├── transform.py └── utils.py ├── docs ├── install.md ├── prepare_data.md └── train_eval.md ├── experiments ├── __init__.py ├── ddp.py ├── gpu_utils.py ├── manager.py ├── model_utils.py ├── runner.py └── utils.py ├── main.py ├── models ├── __init__.py ├── backbone │ ├── PillarResNet.py │ ├── __init__.py │ └── base.py ├── depthnet.py ├── dv3dlane.py ├── dv3dlane_head.py ├── fusion_layer.py ├── head │ ├── __init__.py │ ├── bev_head.py │ └── center_head_todel.py ├── lidar_module_utils.py ├── mmpt_encoder.py ├── model │ ├── __init__.py │ └── pt_branch.py ├── model_init.py ├── ms2one.py ├── neck │ ├── __init__.py │ ├── misc.py │ └── rpn.py ├── norm_utils.py ├── ops │ ├── __init__.py │ └── pillar_ops │ │ ├── group_utils.py │ │ ├── pillar_modules.py │ │ ├── pillar_utils.py │ │ ├── scatter_utils.py │ │ ├── setup.py │ │ └── src │ │ ├── atomics.cuh │ │ ├── cuda_utils.h │ │ ├── group_ops.cpp │ │ ├── group_ops_gpu.cu │ │ ├── group_ops_gpu.h │ │ ├── pillar_api.cpp │ │ ├── pillar_ops.cpp │ │ ├── pillar_ops_gpu.cu │ │ ├── pillar_ops_gpu.h │ │ ├── scatter_ops.cpp │ │ ├── scatter_ops_gpu.cu │ │ └── scatter_ops_gpu.h ├── p2pc.py ├── pos_utils.py ├── pred_utils.py ├── pt_utils.py ├── readers │ ├── __init__.py │ ├── builder.py │ └── dynamic_pillar_encoder.py ├── registry.py ├── scatter_utils.py ├── sparse_ins.py ├── sparse_inst_loss.py ├── transformer_bricks.py └── utils.py ├── process_data.py ├── requirements.txt └── utils ├── MinCostFlow.py ├── __init__.py ├── config.py ├── eval_3D_lane.py ├── misc.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/README.md -------------------------------------------------------------------------------- /assets/main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/assets/main.png -------------------------------------------------------------------------------- /config/_base_/base_res101_bs16xep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/config/_base_/base_res101_bs16xep100.py -------------------------------------------------------------------------------- /config/_base_/openlane_1000_base_res101_bs16xep100_lidar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/config/_base_/openlane_1000_base_res101_bs16xep100_lidar.py -------------------------------------------------------------------------------- /config/_base_/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/config/_base_/optimizer.py -------------------------------------------------------------------------------- /config/release_iclr/dv3dlane_openlane1000_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/config/release_iclr/dv3dlane_openlane1000_base.py -------------------------------------------------------------------------------- /data/Load_Data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/data/Load_Data.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/data/data_utils.py -------------------------------------------------------------------------------- /data/lane_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/data/lane_transform.py -------------------------------------------------------------------------------- /data/lidar_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/data/lidar_utils.py -------------------------------------------------------------------------------- /data/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/data/transform.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/data/utils.py -------------------------------------------------------------------------------- /docs/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/docs/install.md -------------------------------------------------------------------------------- /docs/prepare_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/docs/prepare_data.md -------------------------------------------------------------------------------- /docs/train_eval.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/docs/train_eval.md -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/experiments/ddp.py -------------------------------------------------------------------------------- /experiments/gpu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/experiments/gpu_utils.py -------------------------------------------------------------------------------- /experiments/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/experiments/manager.py -------------------------------------------------------------------------------- /experiments/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/experiments/model_utils.py -------------------------------------------------------------------------------- /experiments/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/experiments/runner.py -------------------------------------------------------------------------------- /experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/experiments/utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/PillarResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/backbone/PillarResNet.py -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/backbone/base.py -------------------------------------------------------------------------------- /models/depthnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/depthnet.py -------------------------------------------------------------------------------- /models/dv3dlane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/dv3dlane.py -------------------------------------------------------------------------------- /models/dv3dlane_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/dv3dlane_head.py -------------------------------------------------------------------------------- /models/fusion_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/fusion_layer.py -------------------------------------------------------------------------------- /models/head/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/head/__init__.py -------------------------------------------------------------------------------- /models/head/bev_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/head/bev_head.py -------------------------------------------------------------------------------- /models/head/center_head_todel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/head/center_head_todel.py -------------------------------------------------------------------------------- /models/lidar_module_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/lidar_module_utils.py -------------------------------------------------------------------------------- /models/mmpt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/mmpt_encoder.py -------------------------------------------------------------------------------- /models/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/model/__init__.py -------------------------------------------------------------------------------- /models/model/pt_branch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/model/pt_branch.py -------------------------------------------------------------------------------- /models/model_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/model_init.py -------------------------------------------------------------------------------- /models/ms2one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ms2one.py -------------------------------------------------------------------------------- /models/neck/__init__.py: -------------------------------------------------------------------------------- 1 | from .rpn import RPNG -------------------------------------------------------------------------------- /models/neck/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/neck/misc.py -------------------------------------------------------------------------------- /models/neck/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/neck/rpn.py -------------------------------------------------------------------------------- /models/norm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/norm_utils.py -------------------------------------------------------------------------------- /models/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/ops/pillar_ops/group_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/group_utils.py -------------------------------------------------------------------------------- /models/ops/pillar_ops/pillar_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/pillar_modules.py -------------------------------------------------------------------------------- /models/ops/pillar_ops/pillar_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/pillar_utils.py -------------------------------------------------------------------------------- /models/ops/pillar_ops/scatter_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/scatter_utils.py -------------------------------------------------------------------------------- /models/ops/pillar_ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/setup.py -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/atomics.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/atomics.cuh -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/cuda_utils.h -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/group_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/group_ops.cpp -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/group_ops_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/group_ops_gpu.cu -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/group_ops_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/group_ops_gpu.h -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/pillar_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/pillar_api.cpp -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/pillar_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/pillar_ops.cpp -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/pillar_ops_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/pillar_ops_gpu.cu -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/pillar_ops_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/pillar_ops_gpu.h -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/scatter_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/scatter_ops.cpp -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/scatter_ops_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/scatter_ops_gpu.cu -------------------------------------------------------------------------------- /models/ops/pillar_ops/src/scatter_ops_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/ops/pillar_ops/src/scatter_ops_gpu.h -------------------------------------------------------------------------------- /models/p2pc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/p2pc.py -------------------------------------------------------------------------------- /models/pos_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/pos_utils.py -------------------------------------------------------------------------------- /models/pred_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/pred_utils.py -------------------------------------------------------------------------------- /models/pt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/pt_utils.py -------------------------------------------------------------------------------- /models/readers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/readers/__init__.py -------------------------------------------------------------------------------- /models/readers/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/readers/builder.py -------------------------------------------------------------------------------- /models/readers/dynamic_pillar_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/readers/dynamic_pillar_encoder.py -------------------------------------------------------------------------------- /models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/registry.py -------------------------------------------------------------------------------- /models/scatter_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/scatter_utils.py -------------------------------------------------------------------------------- /models/sparse_ins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/sparse_ins.py -------------------------------------------------------------------------------- /models/sparse_inst_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/sparse_inst_loss.py -------------------------------------------------------------------------------- /models/transformer_bricks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/transformer_bricks.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/models/utils.py -------------------------------------------------------------------------------- /process_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/process_data.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/MinCostFlow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/utils/MinCostFlow.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/eval_3D_lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/utils/eval_3D_lane.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JMoonr/dv-3dlane/HEAD/utils/utils.py --------------------------------------------------------------------------------