Image Colorization Tutorial - Using U-Net and GANs

I’ve done an in depth Tutorial on Image Colorization task using U-Net and Conditional GAN with PyTorch.
I’ve written a blog post about it on TowardsDataScience: Link
Also, all the project as a notebook along with the blog post explanations are available on my GitHub repo: Link

You can open the whole project directly on Google Colab and using the pretrianed weights, start colorizing your black and white images and learn a lot about the task and how it get solved using deep learning.
Open In Colab

Please star the GitHub repo if you liked the tutorial and the project. Thanks!

3 Likes

Cool, thanks for sharing it here! :slight_smile:

1 Like