arkel23 / animesion

For holding anime-related object classification and detection models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Animesion

An end-to-end framework for anime character recognition and tagging.

Try it on Google Colab!

Install requirements by first creating a conda environment and installing with conda then pip packages for packages that cant be found in conda

conda create --name animesion --file requirements_conda.txt
conda activate animesion
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
pip install -r requirements.txt
cd classification_tagging/models & pip install -e . & python download_convert_models.py & cd ../../
cd utilities/loss-landscapes/ & pip install -e . & cd ../../..

About

For holding anime-related object classification and detection models

License:MIT License


Languages

Language:Jupyter Notebook 81.1%Language:Python 18.9%