GitHub - hbenbel/Colorization: Grayscale image colorization using a conditional DCGAN in Python and Pytorch

hbenbel / Colorization Public

Notifications You must be signed in to change notification settings
Fork 3
Star 0

Grayscale image colorization using a conditional DCGAN in Python and Pytorch

hbenbel.github.io/blog/colorization/

MIT license

0 stars 3 forks Branches Tags Activity

Star

Notifications

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 33 Commits
config		config
src		src
LICENSE		LICENSE
README		README
colorize.py		colorize.py
requirements.txt		requirements.txt
train.py		train.py

Repository files navigation

 ██████╗ ██████╗ ██╗      ██████╗ ██████╗ ██╗███████╗███████╗
██╔════╝██╔═══██╗██║     ██╔═══██╗██╔══██╗██║╚══███╔╝██╔════╝
██║     ██║   ██║██║     ██║   ██║██████╔╝██║  ███╔╝ █████╗  
██║     ██║   ██║██║     ██║   ██║██╔══██╗██║ ███╔╝  ██╔══╝  
╚██████╗╚██████╔╝███████╗╚██████╔╝██║  ██║██║███████╗███████╗
 ╚═════╝ ╚═════╝ ╚══════╝ ╚═════╝ ╚═╝  ╚═╝╚═╝╚══════╝╚══════╝
                                                             

DESCRIPTION
    Grayscale image colorization using a conditional DCGAN

USAGE
    First make sure that you have every packages in the requirements.txt file.

    The input image resolution should be either 32x32 or 256x256.
    To train using DCGAN change the config/train.json file to put the correct
    parameters and launch the training using:
        python train.py --config config/train.json

    To test the trained model, change the config/colorize.json file to put the
    correct parameters and launch the inference using:
        python colorize.py --config config/colorize.json

REFERENCES
    Colorful Image Colorization (https://arxiv.org/abs/1603.08511)
    Image Colorization with Generative Adversarial Networks (https://arxiv.org/abs/1803.05400)

CONTRIBUTORS
    Hussem Ben Belgacem