oskar-j / awesome-image-coloring

A curated list of awesome AI-powered image coloring frameworks, libraries and software.

Awesome Software for Image Coloring

A curated list of awesome AI-powered image coloring frameworks, libraries and software. Inspired by josephmisiti/awesome-machine-learning. It's a good idea to explore the GitHub topic as well - Topic "Image colorization". In comparison to the awesome list MarkMoHR/Awesome-Image-Colorization (which focuses on research papers), I focus on practical open-source software.

Considerations

Most of the software runs in Python, and requires some kind of an AI frameworks (e.g. Tensorflow) - which means you may need a GPU with a configured CUDA toolkit to run it in a reasonable time.

Frameworks and libraries

🐍 Python

Tensorflow

• Automatic Image Colorization - Automatic Image Colorization using TensorFlow based on Residual Encoder Network http://tinyclouds.org/colorize/
• Image Colorization using Convolutional Networks - Image colorization using CNNs in tensorflow.
• Image and video colorizer - Image and video colorizer is package for automatic image and video colorization. Models are already trained.
• PIC - Probabilistic Image Colorization - Probabilistic Image Colorization https://arxiv.org/abs/1705.04258
• Photo Coloring Using End2end CNN based Model! - A Deep Learning based coloring tool, which can color a black-white or gray picture.

Tensorflow with GANs

• Image Colorization with Generative Adversarial Networks - Grayscale Image Colorization with Generative Adversarial Networks. https://arxiv.org/abs/1803.05400
• Image-colorization-using-CycleGAN - Colorization of grayscale images using CycleGAN in TensorFlow.

Keras

• Coloring Black and White photos with Neural Networks - Coloring black and white images with deep learning.
• JadeBlue96 - Recolorizing grayscaled historical paintings and photos with Deep Learning using an Autoencoder CNN.
• Image-Coloring - Deep Neural Net for coloring grayscale images using local and global image features
• Image-Colorization - Automatic Image Colorization using a Convolutional Network (U-Net)

Fast.AI

• DeOldify - A Deep Learning based project for colorizing and restoring old images.

Caffee

• Colorful Image Colorization - Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016. http://richzhang.github.io/colorization/
• Interactive Deep Colorization - Deep learning software for colorizing black and white images with a few clicks. https://richzhang.github.io/ideepcolor/

PyTorch

• Interactive Deep Colorization in PyTorch - PyTorch reimplementation of Interactive Deep Colorization https://richzhang.github.io/ideepcolor/
• Automatic Image Colorization - Automatic Image Colorization with Simultaneous Classification – based on "Let there be Color!".
• Image colorization with GANs - Image colorization with generative adversarial networks on the CIFAR10 dataset.
• Colorful Image Colorization PyTorch - A from-scratch PyTorch implementation of "Colorful Image Colorization" by Zhang et al. created for the Deep Learning in Data Science course at KTH Stockholm.
• Colorful Image Colorization - Pytorch implementation of the paper Colorful Image Colorization https://arxiv.org/abs/1603.08511
• Square-Images-Colorization - Colorization algorithms for images depicting cities squares

C++

• Beyond Landscapes: An Exemplar-based Image colorization method - Exemplar-based Image Colorization method based on superpixel segmentation and classification.

C#

• StyleTransfer-Colorization-SuperResolution - Demonstration implementations of neural network image processing algorithms.

Language-based colorization

• SketchySceneColorization - Language-based Colorization of Scene Sketches. (SIGGRAPH Asia 2019) https://sketchyscene.github.io/SketchySceneColorization/

Implementations / apps

iOS

• Colorizer iOS - Transform grayscale photos to color photos in iOS

📚 Relevant knowledge, books and papers

• Awesome-Image-Colorization - A collection of Deep Learning based Image Colorization and Video Colorization papers.

• Build a Photo Restoration App with Python - YouTube tutorial from AssemblyAI on how to build a photo restoration app with Python and Flask.

🕶️ Related awesome lists

• Awesome Machine Learning - A curated list of awesome Machine Learning frameworks, libraries and software.
• Awesome Deep Learning - A curated list of awesome Deep Learning tutorials, projects and communities.
• Awesome Deep Vision - A curated list of deep learning resources for computer vision.