learnopencv/Fine-Tuning-SegFormer-For-Lane-Detection

Fine-Tuning SegFormer for Improved Lane Detection in Autonomous Vehicles

This repository contains the Jupyter Notebook for fine-tuning and inference using the SegFormer-b2 model on the Berkeley Deep Drive (BDD100K) dataset. This is part of the LearnOpenCV blog post - SegFormer 🤗 : Fine-Tuning for Improved Lane Detection in Autonomous Vehicles.

The notebook is one-click runnable and the dataset will download automatically.

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.