KITTI Road Segmentation

KITTI (Karlsruhe Institute of Technology and Toyota Technological Institute at Chicago) vision benchmark suite provides data for several tasks relevant to autonomous driving. One of the tasks is to detect the road/lane in images.

This code uses a custom U-Net architecture for road segmentation in the image, i.e, estimating pixels that correspond to the road/lane.

Prediction

U-Net's prediction on the test data are shown below. The predicted pixels for road/lane are shown in red.

Predictions for unmarked lane

Predictions for marked lane

Predictions for multiple lanes