STEM images colorization using Deep Convolutional Neural Networks.
A. Install and update using pip:
pip3 install color-spray
Or
git checkout https://github.com/fengwang/color_spray.git
cd color_spray
python3 -m pip install -e .
B. Dowload pretrained model files from https://drive.google.com/drive/folders/1dl-iNgROmSv71EpzNalh97zksKIBc90u?usp=sharing
, place them in your home folder, under path .color_spray/model
.
Command line:
color_spray INPUT_GRAY_IMAGE_PATH OUTPUT_RGB_IMAGE_PATH
Using Python API:
# uncomment the follow three lines if you have a Nvidia GPU but you do not want to enable it.
#import os
#os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
#os.environ["CUDA_VISIBLE_DEVICES"]=''
from color_spray import color_spray
rgb_image = color_spray( './a_gray_image.png', './an_rgb_image.png' )
- The training images are downloaded from PEXEL.
- BSD