├── .gitignore ├── README.md ├── configs └── flamnet │ ├── flamnet_dla34_culane.py │ └── flamnet_resnet34_culane.py ├── flamnet ├── __init__.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── culane.py │ ├── curvelanes.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 │ │ └── topformer.py │ ├── heads │ │ ├── __init__.py │ │ └── flamnet_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── focal_loss.py │ │ └── lineiou_loss.py │ ├── necks │ │ ├── __init__.py │ │ ├── fpn.py │ │ ├── pafpn.py │ │ └── ppam_dsaformer.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 ├── main.py ├── requirements.txt ├── setup.py └── tools ├── detect.py └── generate_seg_tusimple.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/README.md -------------------------------------------------------------------------------- /configs/flamnet/flamnet_dla34_culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/configs/flamnet/flamnet_dla34_culane.py -------------------------------------------------------------------------------- /configs/flamnet/flamnet_resnet34_culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/configs/flamnet/flamnet_resnet34_culane.py -------------------------------------------------------------------------------- /flamnet/__init__.py: -------------------------------------------------------------------------------- 1 | from .ops import * 2 | -------------------------------------------------------------------------------- /flamnet/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/__init__.py -------------------------------------------------------------------------------- /flamnet/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/base_dataset.py -------------------------------------------------------------------------------- /flamnet/datasets/culane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/culane.py -------------------------------------------------------------------------------- /flamnet/datasets/curvelanes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/curvelanes.py -------------------------------------------------------------------------------- /flamnet/datasets/llamas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/llamas.py -------------------------------------------------------------------------------- /flamnet/datasets/process/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/process/__init__.py -------------------------------------------------------------------------------- /flamnet/datasets/process/generate_lane_line.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/process/generate_lane_line.py -------------------------------------------------------------------------------- /flamnet/datasets/process/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/process/process.py -------------------------------------------------------------------------------- /flamnet/datasets/process/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/process/transforms.py -------------------------------------------------------------------------------- /flamnet/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/registry.py -------------------------------------------------------------------------------- /flamnet/datasets/tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/datasets/tusimple.py -------------------------------------------------------------------------------- /flamnet/engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flamnet/engine/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/engine/optimizer.py -------------------------------------------------------------------------------- /flamnet/engine/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/engine/registry.py -------------------------------------------------------------------------------- /flamnet/engine/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/engine/runner.py -------------------------------------------------------------------------------- /flamnet/engine/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/engine/scheduler.py -------------------------------------------------------------------------------- /flamnet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/__init__.py -------------------------------------------------------------------------------- /flamnet/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/backbones/__init__.py -------------------------------------------------------------------------------- /flamnet/models/backbones/dla34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/backbones/dla34.py -------------------------------------------------------------------------------- /flamnet/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/backbones/resnet.py -------------------------------------------------------------------------------- /flamnet/models/backbones/topformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/backbones/topformer.py -------------------------------------------------------------------------------- /flamnet/models/heads/__init__.py: -------------------------------------------------------------------------------- 1 | from .flamnet_head import FLAMNetHead -------------------------------------------------------------------------------- /flamnet/models/heads/flamnet_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/heads/flamnet_head.py -------------------------------------------------------------------------------- /flamnet/models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flamnet/models/losses/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/losses/accuracy.py -------------------------------------------------------------------------------- /flamnet/models/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/losses/focal_loss.py -------------------------------------------------------------------------------- /flamnet/models/losses/lineiou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/losses/lineiou_loss.py -------------------------------------------------------------------------------- /flamnet/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/necks/__init__.py -------------------------------------------------------------------------------- /flamnet/models/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/necks/fpn.py -------------------------------------------------------------------------------- /flamnet/models/necks/pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/necks/pafpn.py -------------------------------------------------------------------------------- /flamnet/models/necks/ppam_dsaformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/necks/ppam_dsaformer.py -------------------------------------------------------------------------------- /flamnet/models/nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/nets/__init__.py -------------------------------------------------------------------------------- /flamnet/models/nets/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/nets/detector.py -------------------------------------------------------------------------------- /flamnet/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/registry.py -------------------------------------------------------------------------------- /flamnet/models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flamnet/models/utils/dynamic_assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/utils/dynamic_assign.py -------------------------------------------------------------------------------- /flamnet/models/utils/roi_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/utils/roi_gather.py -------------------------------------------------------------------------------- /flamnet/models/utils/seg_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/models/utils/seg_decoder.py -------------------------------------------------------------------------------- /flamnet/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/ops/__init__.py -------------------------------------------------------------------------------- /flamnet/ops/csrc/nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/ops/csrc/nms.cpp -------------------------------------------------------------------------------- /flamnet/ops/csrc/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/ops/csrc/nms_kernel.cu -------------------------------------------------------------------------------- /flamnet/ops/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/ops/nms.py -------------------------------------------------------------------------------- /flamnet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/__init__.py -------------------------------------------------------------------------------- /flamnet/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/config.py -------------------------------------------------------------------------------- /flamnet/utils/culane_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/culane_metric.py -------------------------------------------------------------------------------- /flamnet/utils/lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/lane.py -------------------------------------------------------------------------------- /flamnet/utils/llamas_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/llamas_metric.py -------------------------------------------------------------------------------- /flamnet/utils/llamas_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/llamas_utils.py -------------------------------------------------------------------------------- /flamnet/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/logger.py -------------------------------------------------------------------------------- /flamnet/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/net_utils.py -------------------------------------------------------------------------------- /flamnet/utils/recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/recorder.py -------------------------------------------------------------------------------- /flamnet/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/registry.py -------------------------------------------------------------------------------- /flamnet/utils/tusimple_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/tusimple_metric.py -------------------------------------------------------------------------------- /flamnet/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/flamnet/utils/visualization.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/setup.py -------------------------------------------------------------------------------- /tools/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/tools/detect.py -------------------------------------------------------------------------------- /tools/generate_seg_tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RanHao-cq/FLAMNet/HEAD/tools/generate_seg_tusimple.py --------------------------------------------------------------------------------