Today, colorization is done by hand in Photoshop, a picture can take up to one month to colorize. It requires extensive research. A face alone needs up to 20 layers of pink, green and blue shades to get it just right. But something changed this year when Amir Avni used neural networks to troll the subreddit /r/Colorization - a community where people colorize historical black and white images manually using Photoshop. They were astonished with Amir’s deep learning bot - what could take up to a month of manual labour could now be done in just a few seconds.
Click this button to open a Workspace on FloydHub that will train this model.
Fascinated by Amir’s neural network, Emill reproduced it and documented the process in the famous blog post: Colorizing B&W Photos with Neural Networks. In this notebook we will reproduce Emil's work by using the Full Version of his experiments.
We will:
- Preprocess the image data for this CV task
- Build and train the
colornetmodel using Keras and Tensorflow - Evaluate our model on the test set
- Run the model on your own black&white and colored pictures!
You can easily spin up a serve job on FloydHub to demo your model through an interactive web site. Just run the following command from workspace terminal or your local machine:
floyd run --mode serveYou should be able to see the following page when visiting the FloydHub serve url:
