Anime Character Classifier

Anime character classifier where all the data was collected, trained on ResNet50. Trained and deployed using PyTorch.

The anime include: Attack On Titan, Dragon Ball Z/Super, Naruto, One Piece.

List Of Characters

[
    'Armin Arlert',
    'Tony Tony Chopper',
    'Eren Yeager',
    'Frieza',
    'Itachi Uchiha',
    'Kakashi Hatake',
    'Levi Ackerman',
    'Monkey D. Luffy',
    'Mikasa Ackerman',
    'Muten Roshi',
    'Naruto Uzumaki',
    'Reiner Braun',
    'Sakura Haruno',
    'Vinsmoke Sanji',
    'Sasuke Uchiha',
    'Son Gohan',
    'Son Goku',
    'Usopp',
    'Vegeta',
    'Roronoa Zoro'
]

Setup

Clone the repository:

git clone https://github.com/RudCodera8/AnimeClassifier

Navigate into the folder:

cd AnimeClassifier

For running CLI app:

Install dependencies:

pip install torch torchvision pillow requests

Run the CLI app:

python cli.py

For running the web app:

Install dependencies:

pip install -r requirements.txt

Run the app:

python app.py

Usage

• For the web app: Select an image and click on submit button
• For the CLI app:

usage: cli.py [-h] [-p PATH | -u URL]
args
optional arguments:

-h, --help            show this help message and exit
-p PATH, --path PATH  Path to image from your PC
-u URL, --url URL     URL of the image

Examples:

Web App:

CLI App:

Contributions for data collection:

Susil Kessav - Attack On Titan Data

Vamshi Krishna - Dragon Ball Z/Super Data

Abhijith Menon - Naruto Data

Vysnav - One Piece Data