Awesome
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
:snake: 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
:books: 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.
:dark_sunglasses: 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.