color-spray 0.1.0

Library that colorizes gray STEM imagess using deep convolutional neural networks.

Color Spray

STEM images colorization using Deep Convolutional Neural Networks.

Installing

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.

Usage

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' )

Details

• The training images are downloaded from PEXEL.

License

• BSD