Generative AI
An image colorization application that converts black and white images to colored ones using pix2pix model. Also includes interactive digit recognition webapp.
Check out the repository here
While working on this project, I have also built digit recognition models. You can test the performance of the models here.
Digit Recognition
Select a model from the list of models. Draw a digit in the canvas, and the predicted value is displayed below. Note. The model may take above 30 seconds to load as the backend Heroku app needs to boot up.
I could not upload more models due to space constraints on Heroku.
Image Colorization
I’m skipping rest of the models due to space constraints. Here is the main model that I have worked on. Upload a B/W image and my model will add color to it!
Note. Upload images of size 128*128 for the best results. If you upload larger images, it may just show a black image.
I was lazy and therefore didn’t bother to verify the file uploaded. Make sure you upload an image, else the app can crash!