Muzalevsky / roadLaneSegmentation

Road lane segmentation based on Apolloscape
MIT License
2 stars 0 forks source link

Road Lane Segmentation

[![PythonSupported](https://img.shields.io/badge/python-3.9-brightgreen.svg)](https://python3statement.org/#sections50-why) [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) [![Pre-commit](https://img.shields.io/badge/pre--commit-enabled-brightgreen?logo=pre-commit&logoColor=white)](https://pre-commit.com/) [![poetry](https://img.shields.io/endpoint?url=https://python-poetry.org/badge/v0.json)](https://python-poetry.org/) ![pytorch](https://img.shields.io/badge/PyTorch-EE4C2C?style=for-the-badge&logo=pytorch&logoColor=white) ![ubuntu](https://img.shields.io/badge/Ubuntu-E95420?style=for-the-badge&logo=ubuntu&logoColor=white) This project focuses on the segmentation of city road lane markings using deep machine learning techniques. ![DemoGIF](demo/test_camera_6(2).gif) ## Trained Models

T204 trained on datasets with errors

## Metrics | Class label | Class name | IoU - The latest | IoU - T204 | The latest DICE | T204 DICE | | :---------: | :------------------------: | :--------------: | :--------: | :-------------: | :-------: | | background | background | 0.998 | 0.998 | 0.999 | 0.998 | | SYD | solid yellow dividing line | 0.986 | 0.823 | 0.987 | 0.899 | | BWG | broken white guiding line | 0.693 | 0.568 | 0.775 | 0.682 | | SWD | solid white dividing line | 0.794 | 0.617 | 0.881 | 0.727 | | SWS | solid white stop line | 0.114 | 0.432 | 0.162 | 0.54 | | CWYZ | crosswalk | 0.386 | 0.56 | 0.496 | 0.687 | | - | - | - | - | - | - | | micro | | 0.995 | 0.995 | 0.998 | 0.998 | | macro | | 0.662 | 0.667 | 0.717 | 0.756 | ## Articles

A. R. Muzalevskiy, E. V. Serykh, M. M. Kopichev, E. V. Druian and M. A. Chernyshev, "Lane Marking Semantic Segmentation Using Convolutional Neural Networks," 2023 XXVI International Conference on Soft Computing and Measurements (SCM), Saint Petersburg, Russian Federation, 2023, pp. 123-126, doi: 10.1109/SCM58628.2023.10159034.

## Repository contents

System Requirements