--- 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") ```