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