DEV Community

Francesco Tisiot
Francesco Tisiot

Posted on

Machine Learning challenge: Chihuahua vs Muffin with PostgreSQL and pgvector

Is it a Muffin or a Chihuahua?

I tried to solve the famous meme using #PostgreSQL pgvector extension. The results are really impressive!

Do you want to try? I created a Python notebook that can give you a jump start, check it out at https://go.aiven.io/muffin-vs-chihuahua

You can find the training dataset at https://www.kaggle.com/datasets/samuelcortinhas/muffin-vs-chihuahua-image-classification

You can find a managed PostgreSQL for free at https://go.aiven.io/francesco-signup

Top comments (0)