├── .github └── FUNDING.yml ├── .gitignore ├── LICENSE ├── README.md ├── config ├── __init__.py ├── genlanenet_openlane.py ├── once_eval_config.json ├── persformer_apollo.py ├── persformer_once.py └── persformer_openlane.py ├── data ├── Load_Data.py └── __init__.py ├── docs ├── INSTALL.md └── TRAIN_VAL.md ├── experiments ├── __init__.py ├── ddp.py └── runner.py ├── imgs ├── apollo_vis.png ├── openlane_vis.png ├── pipeline.png └── pipeline.svg ├── main_genlanenet.py ├── main_persformer.py ├── models ├── PersFormer.py ├── __init__.py ├── networks │ ├── Deform_ATTN.py │ ├── GeoNet3D_ext.py │ ├── Lane2D.py │ ├── Lane3D.py │ ├── Layers.py │ ├── Loss_crit.py │ ├── PE.py │ ├── Unet_parts.py │ ├── __init__.py │ ├── erfnet.py │ ├── feature_extractor.py │ ├── fpn.py │ └── libs │ │ ├── __init__.py │ │ ├── focal_loss.py │ │ ├── lane.py │ │ ├── layers.py │ │ └── matching.py ├── nms │ ├── setup.py │ └── src │ │ ├── nms.cpp │ │ ├── nms │ │ └── __init__.py │ │ └── nms_kernel.cu └── ops │ ├── functions │ ├── __init__.py │ └── ms_deform_attn_func.py │ ├── make.sh │ ├── modules │ ├── __init__.py │ └── ms_deform_attn.py │ ├── setup.py │ ├── src │ ├── cpu │ │ ├── ms_deform_attn_cpu.cpp │ │ └── ms_deform_attn_cpu.h │ ├── cuda │ │ ├── ms_deform_attn_cuda.cu │ │ ├── ms_deform_attn_cuda.h │ │ └── ms_deform_im2col_cuda.cuh │ ├── ms_deform_attn.h │ └── vision.cpp │ └── test.py ├── requirements.txt └── utils ├── MinCostFlow.py ├── __init__.py ├── eval_3D_lane.py ├── eval_3D_once.py └── utils.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/genlanenet_openlane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/config/genlanenet_openlane.py -------------------------------------------------------------------------------- /config/once_eval_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/config/once_eval_config.json -------------------------------------------------------------------------------- /config/persformer_apollo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/config/persformer_apollo.py -------------------------------------------------------------------------------- /config/persformer_once.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/config/persformer_once.py -------------------------------------------------------------------------------- /config/persformer_openlane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/config/persformer_openlane.py -------------------------------------------------------------------------------- /data/Load_Data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/data/Load_Data.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /docs/TRAIN_VAL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/docs/TRAIN_VAL.md -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/experiments/ddp.py -------------------------------------------------------------------------------- /experiments/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/experiments/runner.py -------------------------------------------------------------------------------- /imgs/apollo_vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/imgs/apollo_vis.png -------------------------------------------------------------------------------- /imgs/openlane_vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/imgs/openlane_vis.png -------------------------------------------------------------------------------- /imgs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/imgs/pipeline.png -------------------------------------------------------------------------------- /imgs/pipeline.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/imgs/pipeline.svg -------------------------------------------------------------------------------- /main_genlanenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/main_genlanenet.py -------------------------------------------------------------------------------- /main_persformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/main_persformer.py -------------------------------------------------------------------------------- /models/PersFormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/PersFormer.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/networks/Deform_ATTN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/Deform_ATTN.py -------------------------------------------------------------------------------- /models/networks/GeoNet3D_ext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/GeoNet3D_ext.py -------------------------------------------------------------------------------- /models/networks/Lane2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/Lane2D.py -------------------------------------------------------------------------------- /models/networks/Lane3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/Lane3D.py -------------------------------------------------------------------------------- /models/networks/Layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/Layers.py -------------------------------------------------------------------------------- /models/networks/Loss_crit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/Loss_crit.py -------------------------------------------------------------------------------- /models/networks/PE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/PE.py -------------------------------------------------------------------------------- /models/networks/Unet_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/Unet_parts.py -------------------------------------------------------------------------------- /models/networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/networks/erfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/erfnet.py -------------------------------------------------------------------------------- /models/networks/feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/feature_extractor.py -------------------------------------------------------------------------------- /models/networks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/fpn.py -------------------------------------------------------------------------------- /models/networks/libs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/networks/libs/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/libs/focal_loss.py -------------------------------------------------------------------------------- /models/networks/libs/lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/libs/lane.py -------------------------------------------------------------------------------- /models/networks/libs/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/libs/layers.py -------------------------------------------------------------------------------- /models/networks/libs/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/networks/libs/matching.py -------------------------------------------------------------------------------- /models/nms/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/nms/setup.py -------------------------------------------------------------------------------- /models/nms/src/nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/nms/src/nms.cpp -------------------------------------------------------------------------------- /models/nms/src/nms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/nms/src/nms/__init__.py -------------------------------------------------------------------------------- /models/nms/src/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/nms/src/nms_kernel.cu -------------------------------------------------------------------------------- /models/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/functions/__init__.py -------------------------------------------------------------------------------- /models/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /models/ops/make.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | python setup.py build install 3 | -------------------------------------------------------------------------------- /models/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/modules/__init__.py -------------------------------------------------------------------------------- /models/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /models/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/setup.py -------------------------------------------------------------------------------- /models/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /models/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /models/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /models/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/src/vision.cpp -------------------------------------------------------------------------------- /models/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/models/ops/test.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/MinCostFlow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/utils/MinCostFlow.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/eval_3D_lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/utils/eval_3D_lane.py -------------------------------------------------------------------------------- /utils/eval_3D_once.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/utils/eval_3D_once.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/PersFormer_3DLane/HEAD/utils/utils.py --------------------------------------------------------------------------------