A TensorFlow implementation of this Nvidia paper with some changes.
Download the dataset and extract into the repository folder
Use python train.py
to train the model
Use python run.py
to run the model on a live webcam feed
Use python run_dataset.py
to run the model on the dataset
To visualize training using Tensorboard use tensorboard --logdir=./logs
, then open http://0.0.0.0:6006/ into your web browser.