Skip to content
Put color back into monochromatic photos of nature.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
convnet
.gitignore
README.md
atomicfile.py
caffee_test.py
constants.py
flickr_scrape.py
image_segmentation.py
nps_photo_retrieve.py
requirements.txt
resize_crop.py
segment_images.py
test_svr.py
train_svr.py
util.py

README.md

Recolorizer

Imgur

Research Paper

Setup

First, install dependencies via pip:

pip install -r requirements.txt

We provide a script to automatically scrape images from Flickr. Run the script to acquire a dataset:

python flickr_scrape.py

Generate the SVR model by running

python train_svr.py

Finally, test the SVR model on a single photo by running

python test_svr.py <file name>
You can’t perform that action at this time.