├── .github └── FUNDING.yml ├── .gitignore ├── LICENSE ├── README.md ├── figs ├── lane_segment.jpg └── pipeline.png ├── projects ├── __init__.py ├── bevformer │ ├── __init__.py │ └── modules │ │ ├── __init__.py │ │ ├── custom_base_transformer_layer.py │ │ ├── decoder.py │ │ ├── encoder.py │ │ ├── multi_scale_deformable_attn_function.py │ │ ├── spatial_cross_attention.py │ │ └── temporal_self_attention.py ├── configs │ ├── lanesegnet_r50_8x1_24e_olv2_subset_A.py │ └── lanesegnet_r50_8x1_24e_olv2_subset_A_mapele_bucket.py └── lanesegnet │ ├── __init__.py │ ├── core │ ├── __init__.py │ ├── lane │ │ ├── __init__.py │ │ ├── area_coder.py │ │ ├── lane_coder.py │ │ ├── lane_hungarian_assigner.py │ │ ├── match_cost.py │ │ └── util.py │ └── visualizer │ │ ├── __init__.py │ │ └── lane_segment.py │ ├── datasets │ ├── __init__.py │ ├── openlanev2_evaluate_custom.py │ ├── openlanev2_subset_A_lanesegnet_dataset.py │ ├── openlanev2_subset_A_mapele_bucket_dataset.py │ └── pipelines │ │ ├── __init__.py │ │ ├── formating.py │ │ ├── loading.py │ │ ├── transform_3d.py │ │ └── transform_3d_lane.py │ ├── models │ ├── __init__.py │ ├── dense_heads │ │ ├── __init__.py │ │ ├── deformable_detr_head.py │ │ ├── laneseg_head.py │ │ └── relationship_head.py │ ├── detectors │ │ ├── __init__.py │ │ ├── lanesegnet.py │ │ └── lanesegnet_mapele_bucket.py │ └── modules │ │ ├── __init__.py │ │ ├── bevformer_constructer.py │ │ ├── lane_attention.py │ │ ├── laneseg_decoder.py │ │ └── laneseg_transformer.py │ ├── thirdparty │ ├── __init__.py │ ├── map_loss.py │ ├── maptr_assigner.py │ ├── maptr_decoder.py │ ├── maptr_head.py │ └── maptr_transformer.py │ └── utils │ ├── __init__.py │ └── builder.py ├── requirements.txt └── tools ├── data_process.py ├── dist_test.sh ├── dist_train.sh ├── test.py └── train.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/README.md -------------------------------------------------------------------------------- /figs/lane_segment.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/figs/lane_segment.jpg -------------------------------------------------------------------------------- /figs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/figs/pipeline.png -------------------------------------------------------------------------------- /projects/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/bevformer/__init__.py: -------------------------------------------------------------------------------- 1 | from .modules import * 2 | -------------------------------------------------------------------------------- /projects/bevformer/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/bevformer/modules/__init__.py -------------------------------------------------------------------------------- /projects/bevformer/modules/custom_base_transformer_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/bevformer/modules/custom_base_transformer_layer.py -------------------------------------------------------------------------------- /projects/bevformer/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/bevformer/modules/decoder.py -------------------------------------------------------------------------------- /projects/bevformer/modules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/bevformer/modules/encoder.py -------------------------------------------------------------------------------- /projects/bevformer/modules/multi_scale_deformable_attn_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/bevformer/modules/multi_scale_deformable_attn_function.py -------------------------------------------------------------------------------- /projects/bevformer/modules/spatial_cross_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/bevformer/modules/spatial_cross_attention.py -------------------------------------------------------------------------------- /projects/bevformer/modules/temporal_self_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/bevformer/modules/temporal_self_attention.py -------------------------------------------------------------------------------- /projects/configs/lanesegnet_r50_8x1_24e_olv2_subset_A.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/configs/lanesegnet_r50_8x1_24e_olv2_subset_A.py -------------------------------------------------------------------------------- /projects/configs/lanesegnet_r50_8x1_24e_olv2_subset_A_mapele_bucket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/configs/lanesegnet_r50_8x1_24e_olv2_subset_A_mapele_bucket.py -------------------------------------------------------------------------------- /projects/lanesegnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/lane/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/lane/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/lane/area_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/lane/area_coder.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/lane/lane_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/lane/lane_coder.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/lane/lane_hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/lane/lane_hungarian_assigner.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/lane/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/lane/match_cost.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/lane/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/lane/util.py -------------------------------------------------------------------------------- /projects/lanesegnet/core/visualizer/__init__.py: -------------------------------------------------------------------------------- 1 | from .lane_segment import draw_annotation_bev 2 | -------------------------------------------------------------------------------- /projects/lanesegnet/core/visualizer/lane_segment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/core/visualizer/lane_segment.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/openlanev2_evaluate_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/openlanev2_evaluate_custom.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/openlanev2_subset_A_lanesegnet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/openlanev2_subset_A_lanesegnet_dataset.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/openlanev2_subset_A_mapele_bucket_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/openlanev2_subset_A_mapele_bucket_dataset.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/pipelines/transform_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/pipelines/transform_3d.py -------------------------------------------------------------------------------- /projects/lanesegnet/datasets/pipelines/transform_3d_lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/datasets/pipelines/transform_3d_lane.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/dense_heads/deformable_detr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/dense_heads/deformable_detr_head.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/dense_heads/laneseg_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/dense_heads/laneseg_head.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/dense_heads/relationship_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/dense_heads/relationship_head.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/detectors/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/detectors/lanesegnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/detectors/lanesegnet.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/detectors/lanesegnet_mapele_bucket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/detectors/lanesegnet_mapele_bucket.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/modules/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/modules/bevformer_constructer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/modules/bevformer_constructer.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/modules/lane_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/modules/lane_attention.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/modules/laneseg_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/modules/laneseg_decoder.py -------------------------------------------------------------------------------- /projects/lanesegnet/models/modules/laneseg_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/models/modules/laneseg_transformer.py -------------------------------------------------------------------------------- /projects/lanesegnet/thirdparty/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/thirdparty/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/thirdparty/map_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/thirdparty/map_loss.py -------------------------------------------------------------------------------- /projects/lanesegnet/thirdparty/maptr_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/thirdparty/maptr_assigner.py -------------------------------------------------------------------------------- /projects/lanesegnet/thirdparty/maptr_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/thirdparty/maptr_decoder.py -------------------------------------------------------------------------------- /projects/lanesegnet/thirdparty/maptr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/thirdparty/maptr_head.py -------------------------------------------------------------------------------- /projects/lanesegnet/thirdparty/maptr_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/thirdparty/maptr_transformer.py -------------------------------------------------------------------------------- /projects/lanesegnet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/utils/__init__.py -------------------------------------------------------------------------------- /projects/lanesegnet/utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/projects/lanesegnet/utils/builder.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/tools/data_process.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenDriveLab/LaneSegNet/HEAD/tools/train.py --------------------------------------------------------------------------------