Image Classifier to identify marvel character

🕸️ A machine learning web application that recognizes the user input image as one of the 6 actors in Marvel’s “The Avengers” series

A Convolutional Neural Network model in PyTorch that can classify 20 different marvel hero characters

In this repository we'll classify Marvel characters as heroes or villains based on their appearance!

This project focuses on image classification using the "Marvel Avengers" dataset. Two models were utilized: a classical model, Support Vector Machine Classifier (SVC), and a deep learning model, DenseNet201. The goal was to classify images into one of eight Marvel character labels and evaluate the models' performance.

The Final Project created by my team and I during the KCVanguard 2025 workshop organized by the Intelligent Computing and Vision Laboratory (KCV). Our group developed a model to classify images into 26 target superhero classes. The dataset we used was scraped from Bing image search and consists of characters from the Marvel and DC Universes.