Anime Character Classifier
Anime character classifier 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