Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/cfzd/Ultra-Fast-Lane-Detection
Ultra Fast Structure-aware Deep Lane Detection (ECCV 2020)
https://github.com/cfzd/Ultra-Fast-Lane-Detection
autonomous-driving autonomous-vehicles cnn lane-detection lane-detector lane-finding pytorch road-detection
Last synced: about 1 month ago
JSON representation
Ultra Fast Structure-aware Deep Lane Detection (ECCV 2020)
- Host: GitHub
- URL: https://github.com/cfzd/Ultra-Fast-Lane-Detection
- Owner: cfzd
- License: mit
- Created: 2020-05-23T11:11:34.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2022-12-14T06:50:09.000Z (over 1 year ago)
- Last Synced: 2024-07-21T21:42:14.747Z (about 2 months ago)
- Topics: autonomous-driving, autonomous-vehicles, cnn, lane-detection, lane-detector, lane-finding, pytorch, road-detection
- Language: Python
- Homepage:
- Size: 160 KB
- Stars: 1,771
- Watchers: 27
- Forks: 491
- Open Issues: 94
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- StarryDivineSky - cfzd/Ultra-Fast-Lane-Detection
README
# Ultra-Fast-Lane-Detection
PyTorch implementation of the paper "[Ultra Fast Structure-aware Deep Lane Detection](https://arxiv.org/abs/2004.11757)".
**\[July 18, 2022\] Updates: The new version of our method has been accepted by TPAMI 2022. Code is available [here](https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2)**.
\[June 28, 2021\] Updates: we will release an extended version, which improves **6.3** points of F1 on CULane with the ResNet-18 backbone compared with the ECCV version.
Updates: Our paper has been accepted by ECCV2020.
The evaluation code is modified from [SCNN](https://github.com/XingangPan/SCNN) and [Tusimple Benchmark](https://github.com/TuSimple/tusimple-benchmark).
Caffe model and prototxt can be found [here](https://github.com/Jade999/caffe_lane_detection).
# Demo
# Install
Please see [INSTALL.md](./INSTALL.md)
# Get started
First of all, please modify `data_root` and `log_path` in your `configs/culane.py` or `configs/tusimple.py` config according to your environment.
- `data_root` is the path of your CULane dataset or Tusimple dataset.
- `log_path` is where tensorboard logs, trained models and code backup are stored. ***It should be placed outside of this project.***
***
For single gpu training, run
```Shell
python train.py configs/path_to_your_config
```
For multi-gpu training, run
```Shell
sh launch_training.sh
```
or
```Shell
python -m torch.distributed.launch --nproc_per_node=$NGPUS train.py configs/path_to_your_config
```
If there is no pretrained torchvision model, multi-gpu training may result in multiple downloading. You can first download the corresponding models manually, and then restart the multi-gpu training.
Since our code has auto backup function which will copy all codes to the `log_path` according to the gitignore, additional temp file might also be copied if it is not filtered by gitignore, which may block the execution if the temp files are large. So you should keep the working directory clean.
***
Besides config style settings, we also support command line style one. You can override a setting like
```Shell
python train.py configs/path_to_your_config --batch_size 8
```
The ```batch_size``` will be set to 8 during training.
***
To visualize the log with tensorboard, run
```Shell
tensorboard --logdir log_path --bind_all
```
# Trained models
We provide two trained Res-18 models on CULane and Tusimple.
| Dataset | Metric paper | Metric This repo | Avg FPS on GTX 1080Ti | Model |
|:--------:|:------------:|:----------------:|:-------------------:|:-----------:|
| Tusimple | 95.87 | 95.82 | 306 | [GoogleDrive](https://drive.google.com/file/d/1WCYyur5ZaWczH15ecmeDowrW30xcLrCn/view?usp=sharing)/[BaiduDrive(code:bghd)](https://pan.baidu.com/s/1Fjm5yVq1JDpGjh4bdgdDLA) |
| CULane | 68.4 | 69.7 | 324 | [GoogleDrive](https://drive.google.com/file/d/1zXBRTw50WOzvUp6XKsi8Zrk3MUC3uFuq/view?usp=sharing)/[BaiduDrive(code:w9tw)](https://pan.baidu.com/s/19Ig0TrV8MfmFTyCvbSa4ag) |
For evaluation, run
```Shell
mkdir tmp
# This a bad example, you should put the temp files outside the project.
python test.py configs/culane.py --test_model path_to_culane_18.pth --test_work_dir ./tmp
python test.py configs/tusimple.py --test_model path_to_tusimple_18.pth --test_work_dir ./tmp
```
Same as training, multi-gpu evaluation is also supported.
# Visualization
We provide a script to visualize the detection results. Run the following commands to visualize on the testing set of CULane and Tusimple.
```Shell
python demo.py configs/culane.py --test_model path_to_culane_18.pth
# or
python demo.py configs/tusimple.py --test_model path_to_tusimple_18.pth
```
Since the testing set of Tusimple is not ordered, the visualized video might look bad and we **do not recommend** doing this.
# Speed
To test the runtime, please run
```Shell
python speed_simple.py
# this will test the speed with a simple protocol and requires no additional dependencies
python speed_real.py
# this will test the speed with real video or camera input
```
It will loop 100 times and calculate the average runtime and fps in your environment.
# Citation
```BibTeX
@InProceedings{qin2020ultra,
author = {Qin, Zequn and Wang, Huanyu and Li, Xi},
title = {Ultra Fast Structure-aware Deep Lane Detection},
booktitle = {The European Conference on Computer Vision (ECCV)},
year = {2020}
}
@ARTICLE{qin2022ultrav2,
author={Qin, Zequn and Zhang, Pengyi and Li, Xi},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
title={Ultra Fast Deep Lane Detection With Hybrid Anchor Driven Ordinal Classification},
year={2022},
volume={},
number={},
pages={1-14},
doi={10.1109/TPAMI.2022.3182097}
}
```
# Thanks
Thanks zchrissirhcz for the contribution to the compile tool of CULane, KopiSoftware for contributing to the speed test, and ustclbh for testing on the Windows platform.