metadata
library_name: anime_segmentation
tags:
- image-segmentation
- model_hub_mixin
- pytorch_model_hub_mixin
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")