TwinLiteNet: An Efficient and Lightweight Model for Driveable Area and Lane Segmentation in Self-Driving Cars

Requirement

See requirements.txt for additional dependencies and version requirements.

pip install -r requirements.txt

Data Preparation

• Download the images from images.

• Download the annotations of drivable area segmentation from segments.

• Download the annotations of lane line segmentation from lane.

/data
    bdd100k
        images
            train/
            val/
            test/
        segments
            train/
            val/
        lane
            train/
            val/

Pipeline

Train

python3 main.py

Test

python3 val.py

Inference

Images

python3 test_image.py

Visualize

Drive-able segmentation

Lane Detection

Acknowledgement

Our source code is inspired by:

• ESPNet
• YOLOP

Citation

If you find our paper and code useful for your research, please consider giving a star :star: and citation :pencil: :

@INPROCEEDINGS{10288646,
  author={Che, Quang-Huy and Nguyen, Dinh-Phuc and Pham, Minh-Quan and Lam, Duc-Khai},
  booktitle={2023 International Conference on Multimedia Analysis and Pattern Recognition (MAPR)}, 
  title={TwinLiteNet: An Efficient and Lightweight Model for Driveable Area and Lane Segmentation in Self-Driving Cars}, 
  year={2023},
  volume={},
  number={},
  pages={1-6},
  doi={10.1109/MAPR59823.2023.10288646}}

TwinLiteNetV2: A small stone can kill a giant

🚀 Coming soon!