This is a python script that fill colors in Black & White images. It uses Caffe (A Deep Learning framework) model and OpenCV.
To prevent the dependencies of this project from messing with your core python installation, you can create a virtualenv.
Now, use pip to install the dependencies using:
pip install -r requirements.txt
Now, we need to download the models, which can be done by the following:
for linux:
chmod +x get_models.sh
./get_models.shfor windows: run get_models.bat file
python fill_color.py --input /path/to/input/image output will be saved in same folder
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.