Skip to content

karanchahal/driving-dirty

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

475 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

driving-dirty

DL Project 2020

How to run

First, install dependencies

# clone project   
git clone https://github.com/annikabrundyn/driving-dirty

# install project   
cd driving-dirty

# MAKE SURE TO INSTALL THE APPROPRIATE PYTORCH AND TORCHVISION

pip install -e .   
pip install -r requirements.txt --ignore-installed

Predicting test images

To run evaluations for class:

cd src/utils
python run_test.py --rm_ckpt_path '../../checkpoints/rm.ckpt'

Training the Autoencoder

python src/autoencoder/autoencoder.py --link '/scratch/ab8690/DLSP20Dataset/data' --gpus 1 --max_epochs 5 --batch_size 32

Some optional arguments to provide:

--link          link to where data is stored
--gpus          how many gpus available
--max_epochs    max number of epochs to train for

About

DL Project 2020

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 100.0%