Anime Seg

🚀 动漫分割模型

本项目提供了一个动漫图像分割模型，借助相关库和工具，能方便地加载和使用该模型进行动漫图像的分割处理，为动漫图像的处理带来便利。

🚀 快速开始

此模型已通过 PytorchModelHubMixin 集成推送到模型中心。

• 库链接：https://github.com/SkyTNT/anime-segmentation

📦 安装指南

下载训练脚本

wget https://raw.githubusercontent.com/SkyTNT/anime-segmentation/refs/heads/main/train.py

安装依赖库

pip install pytorch_lightning kornia timm "huggingface_hub>=0.22"

💻 使用示例

基础用法

# 从 train 模块导入 AnimeSegmentation 类
from train import AnimeSegmentation
# 从预训练模型加载 AnimeSegmentation 模型
model = AnimeSegmentation.from_pretrained("skytnt/anime-seg")