Image Colorization using Autoencoders [PyTorch]
Heroku - https://neural-image-colorizer.herokuapp.com/
Medium - https://medium.com/@pranav.kushare2001/colorize-your-black-and-white-photos-using-ai-4652a34e967
Input B/w image vs Output Colored Image
1] used Resnet18 for Encoder part
2] In decoder part latent representation (extracted features) are being upsampled
1] Their are n number of ways to imporve this model
2] we can use different encoders like resnet50, inceptionnet, Densenet ,Vggnet
3] I have just trained the base model because of time constraint
4] Results will improve in greater extent if model is trained for suitable number of epochs and dataset