├── .github ├── arch.png └── latency_f1score.png ├── .gitignore ├── LICENSE ├── README.md ├── clrnet ├── __init__.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── culane.py │ ├── llamas.py │ ├── process │ │ ├── __init__.py │ │ ├── generate_lane_line.py │ │ ├── process.py │ │ └── transforms.py │ ├── registry.py │ └── tusimple.py ├── engine │ ├── __init__.py │ ├── optimizer.py │ ├── registry.py │ ├── runner.py │ └── scheduler.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── dla34.py │ │ └── resnet.py │ ├── heads │ │ ├── __init__.py │ │ └── clr_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── focal_loss.py │ │ └── lineiou_loss.py │ ├── necks │ │ ├── __init__.py │ │ ├── fpn.py │ │ └── pafpn.py │ ├── nets │ │ ├── __init__.py │ │ └── detector.py │ ├── registry.py │ └── utils │ │ ├── __init__.py │ │ ├── dynamic_assign.py │ │ ├── roi_gather.py │ │ └── seg_decoder.py ├── ops │ ├── __init__.py │ ├── csrc │ │ ├── nms.cpp │ │ └── nms_kernel.cu │ └── nms.py └── utils │ ├── __init__.py │ ├── config.py │ ├── culane_metric.py │ ├── lane.py │ ├── llamas_metric.py │ ├── llamas_utils.py │ ├── logger.py │ ├── net_utils.py │ ├── recorder.py │ ├── registry.py │ ├── tusimple_metric.py │ └── visualization.py ├── configs └── clrnet │ ├── clr_dla34_culane.py │ ├── clr_dla34_llamas.py │ ├── clr_resnet101_culane.py │ ├── clr_resnet101_tusimple.py │ ├── clr_resnet18_culane.py │ ├── clr_resnet18_llamas.py │ ├── clr_resnet18_tusimple.py │ ├── clr_resnet34_culane.py │ └── clr_resnet34_tusimple.py ├── main.py ├── requirements.txt ├── setup.py └── tools └── generate_seg_tusimple.py /.github/arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/.github/arch.png -------------------------------------------------------------------------------- /.github/latency_f1score.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/.github/latency_f1score.png -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/README.md -------------------------------------------------------------------------------- /clrnet/__init__.py: -------------------------------------------------------------------------------- 1 | from .ops import * 2 | -------------------------------------------------------------------------------- /clrnet/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/__init__.py -------------------------------------------------------------------------------- /clrnet/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/base_dataset.py -------------------------------------------------------------------------------- /clrnet/datasets/culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/culane.py -------------------------------------------------------------------------------- /clrnet/datasets/llamas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/llamas.py -------------------------------------------------------------------------------- /clrnet/datasets/process/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/process/__init__.py -------------------------------------------------------------------------------- /clrnet/datasets/process/generate_lane_line.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/process/generate_lane_line.py -------------------------------------------------------------------------------- /clrnet/datasets/process/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/process/process.py -------------------------------------------------------------------------------- /clrnet/datasets/process/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/process/transforms.py -------------------------------------------------------------------------------- /clrnet/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/registry.py -------------------------------------------------------------------------------- /clrnet/datasets/tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/datasets/tusimple.py -------------------------------------------------------------------------------- /clrnet/engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /clrnet/engine/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/engine/optimizer.py -------------------------------------------------------------------------------- /clrnet/engine/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/engine/registry.py -------------------------------------------------------------------------------- /clrnet/engine/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/engine/runner.py -------------------------------------------------------------------------------- /clrnet/engine/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/engine/scheduler.py -------------------------------------------------------------------------------- /clrnet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/__init__.py -------------------------------------------------------------------------------- /clrnet/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/backbones/__init__.py -------------------------------------------------------------------------------- /clrnet/models/backbones/dla34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/backbones/dla34.py -------------------------------------------------------------------------------- /clrnet/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/backbones/resnet.py -------------------------------------------------------------------------------- /clrnet/models/heads/__init__.py: -------------------------------------------------------------------------------- 1 | from .clr_head import CLRHead 2 | -------------------------------------------------------------------------------- /clrnet/models/heads/clr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/heads/clr_head.py -------------------------------------------------------------------------------- /clrnet/models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /clrnet/models/losses/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/losses/accuracy.py -------------------------------------------------------------------------------- /clrnet/models/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/losses/focal_loss.py -------------------------------------------------------------------------------- /clrnet/models/losses/lineiou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/losses/lineiou_loss.py -------------------------------------------------------------------------------- /clrnet/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/necks/__init__.py -------------------------------------------------------------------------------- /clrnet/models/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/necks/fpn.py -------------------------------------------------------------------------------- /clrnet/models/necks/pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/necks/pafpn.py -------------------------------------------------------------------------------- /clrnet/models/nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/nets/__init__.py -------------------------------------------------------------------------------- /clrnet/models/nets/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/nets/detector.py -------------------------------------------------------------------------------- /clrnet/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/registry.py -------------------------------------------------------------------------------- /clrnet/models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /clrnet/models/utils/dynamic_assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/utils/dynamic_assign.py -------------------------------------------------------------------------------- /clrnet/models/utils/roi_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/utils/roi_gather.py -------------------------------------------------------------------------------- /clrnet/models/utils/seg_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/models/utils/seg_decoder.py -------------------------------------------------------------------------------- /clrnet/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/ops/__init__.py -------------------------------------------------------------------------------- /clrnet/ops/csrc/nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/ops/csrc/nms.cpp -------------------------------------------------------------------------------- /clrnet/ops/csrc/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/ops/csrc/nms_kernel.cu -------------------------------------------------------------------------------- /clrnet/ops/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/ops/nms.py -------------------------------------------------------------------------------- /clrnet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/__init__.py -------------------------------------------------------------------------------- /clrnet/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/config.py -------------------------------------------------------------------------------- /clrnet/utils/culane_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/culane_metric.py -------------------------------------------------------------------------------- /clrnet/utils/lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/lane.py -------------------------------------------------------------------------------- /clrnet/utils/llamas_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/llamas_metric.py -------------------------------------------------------------------------------- /clrnet/utils/llamas_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/llamas_utils.py -------------------------------------------------------------------------------- /clrnet/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/logger.py -------------------------------------------------------------------------------- /clrnet/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/net_utils.py -------------------------------------------------------------------------------- /clrnet/utils/recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/recorder.py -------------------------------------------------------------------------------- /clrnet/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/registry.py -------------------------------------------------------------------------------- /clrnet/utils/tusimple_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/tusimple_metric.py -------------------------------------------------------------------------------- /clrnet/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/clrnet/utils/visualization.py -------------------------------------------------------------------------------- /configs/clrnet/clr_dla34_culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_dla34_culane.py -------------------------------------------------------------------------------- /configs/clrnet/clr_dla34_llamas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_dla34_llamas.py -------------------------------------------------------------------------------- /configs/clrnet/clr_resnet101_culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_resnet101_culane.py -------------------------------------------------------------------------------- /configs/clrnet/clr_resnet101_tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_resnet101_tusimple.py -------------------------------------------------------------------------------- /configs/clrnet/clr_resnet18_culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_resnet18_culane.py -------------------------------------------------------------------------------- /configs/clrnet/clr_resnet18_llamas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_resnet18_llamas.py -------------------------------------------------------------------------------- /configs/clrnet/clr_resnet18_tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_resnet18_tusimple.py -------------------------------------------------------------------------------- /configs/clrnet/clr_resnet34_culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_resnet34_culane.py -------------------------------------------------------------------------------- /configs/clrnet/clr_resnet34_tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/configs/clrnet/clr_resnet34_tusimple.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/setup.py -------------------------------------------------------------------------------- /tools/generate_seg_tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Turoad/CLRNet/HEAD/tools/generate_seg_tusimple.py --------------------------------------------------------------------------------