tum-phoenix / drive_ml

This is the repository for our Machine Learning applications
13 stars 4 forks source link

LaneDetection: spatialCNN (SCNN) - Implement/Test lane detection (open source dataset) #30

Open Christiaan1990 opened 4 years ago

Christiaan1990 commented 4 years ago

Goal:

Implement, train and test a spatialCNN algorithm and test it with the open source datasets CULane and TuSimple

Assigments:

Notes:

MaslinuPoimal commented 4 years ago

Note: this might be a helpful link.

In particular, we can try out the top paper which seems to be speed-oriented as well.

Hoff97 commented 4 years ago

A good working implementation of SCNN in pytorch can be found here: https://github.com/harryhan618/SCNN_Pytorch

Hoff97 commented 4 years ago

I tested a model trained on TuSimple on some simple images from the car. It doesnt look like the network can detect any lanes, probably because the training pictures look too different.