This model has been pushed to the Hub using the PytorchModelHubMixin integration:

How to load the model

wget https://raw.githubusercontent.com/SkyTNT/anime-segmentation/refs/heads/main/train.py
pip install pytorch_lightning kornia timm "huggingface_hub>=0.22"
from train import AnimeSegmentation
model = AnimeSegmentation.from_pretrained("skytnt/anime-seg")
Downloads last month
61
Safetensors
Model size
51M params
Tensor type
F32
Β·
Inference Examples
Inference API (serverless) does not yet support anime_segmentation models for this pipeline type.

Spaces using skytnt/anime-seg 100