Draw like Bob Ross using the power of Neural Networks (With PyTorch)!
Switch branches/tags
Nothing to show
Clone or download
Permalink
Failed to load latest commit information.
aae updated model Apr 5, 2017
pretrained_models updated model Apr 5, 2017
static/js added color wheel Apr 3, 2017
templates added color wheel Apr 3, 2017
visuals prebundles Apr 5, 2017
.gitignore prebundles Apr 5, 2017
LICENSE Initial commit Apr 2, 2017
README.md Update README.md Apr 5, 2017
app.py prebundles Apr 5, 2017
preprocess.py added preprocessing Apr 3, 2017
requirements.txt updated requirements Apr 3, 2017
scrapper.sh basic gan Apr 2, 2017

README.md

Draw like Bob Ross using the power of Neural Networks! (+ Pytorch)

Sample

Learning Process Visualization

webm

Getting started

Install dependecies

Requires python3.x

pip install -r requirements.txt
pip install http://download.pytorch.org/whl/cu75/torch-0.1.11.post5-cp36-cp36m-linux_x86_64.whl 
pip install torchvision

Run server (using pretrained model)

python app.py --resume pretrained_models/450epoch_aae.tar

Navigate to 127.0.0.1:5000

Run server (from scatch)

1. Scrapping data

./scrapper.sh

2. Preprocess data (should take around 5-10 mins)

python preprocess.py

3. Train data

cd aae
python train.py

4. Run server

cd PROJECT_ROOT
python app.py --resume TRAINEDMODEL.path.tar

Navigate to 127.0.0.1:5000