This program can colour B&W movies using deep learning.
Final SSIM score = 0.873
Some example images painted by this program:
-
Type this in command line to install all dependencies except PyTorch:
pip install -r requirements.txt
Dataset I used for training: COCO 2017 Unlabeled images (After filtering out grayscale images)
Model trained on this dataset for 256x256 images- Download from G-Drive
Note (For Non-Technical Users) - Users who have no knowledge of deep learning or pytorch are advised to only edit values in 'settings.py'. Most of the training and application usage can be handled using this file.
app.py - Run it to colour your images. This is basically a GUI for 'color_images.py'. A trained model should be on location MODEL_PATH mentioned in 'settings.py'.
colour_images.py - Run this file to colour your grayscale images using a trained model in python console. A trained model should be on location MODEL_PATH mentioned in 'settings.py'.
train_model.py - You can train your own image colouring model by running this script. All training settings can be adjusted from 'settings.py'.
colour_video.py - Run this file to colour your grayscale videos using a trained model. A trained model should be on location MODEL_PATH mentioned in 'settings.py'.
Heavily inspired from Colorizing black & white images with U-Net and conditional GAN