| 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](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration: | |
| - Library: https://github.com/SkyTNT/anime-segmentation | |
| ## 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" | |
| ``` | |
| ```python | |
| from train import AnimeSegmentation | |
| model = AnimeSegmentation.from_pretrained("skytnt/anime-seg") | |
| ``` | |