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!


Cool, thanks for sharing it here! :slight_smile:

1 Like