🚀 SPIDER-Skin模型
SPIDER-skin模型是一個用於皮膚貼片級病理分類的深度學習模型。它隸屬於SPIDER數據集計劃,該計劃提供了一個大型、高質量、多器官的病理數據集,並帶有專家標註的標籤。若你想支持、贊助或獲取SPIDER數據和模型的商業許可,請聯繫models@hist.ai。
🚀 快速開始
模型適用於 1120×1120 的貼片。使用以下代碼片段加載並使用模型:
from transformers import AutoModel, AutoProcessor
from PIL import Image
model = AutoModel.from_pretrained("histai/SPIDER-skin-model", trust_remote_code=True)
processor = AutoProcessor.from_pretrained("histai/SPIDER-skin-model", trust_remote_code=True)
image = Image.open("path_to_image.png")
inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)
print(outputs.predicted_class_names)
✨ 主要特性
- 專為皮膚貼片級病理分類而訓練。
- 基於SPIDER數據集,數據質量高且有專家標註。
📦 安裝指南
文檔未提及具體安裝步驟,可參考 transformers
庫的安裝方式。
💻 使用示例
基礎用法
from transformers import AutoModel, AutoProcessor
from PIL import Image
model = AutoModel.from_pretrained("histai/SPIDER-skin-model", trust_remote_code=True)
processor = AutoProcessor.from_pretrained("histai/SPIDER-skin-model", trust_remote_code=True)
image = Image.open("path_to_image.png")
inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)
print(outputs.predicted_class_names)
📚 詳細文檔
模型描述
SPIDER-skin模型是一個用於皮膚貼片級病理分類的深度學習模型。它是SPIDER數據集計劃的一部分,該計劃提供了一個大型、高質量、多器官的病理數據集,並帶有專家標註的標籤。
若你想支持、贊助或獲取SPIDER數據和模型的商業許可,請聯繫models@hist.ai。
模型來源
訓練數據
該模型在 SPIDER-skin 數據集上進行訓練,這是SPIDER數據集的一個子集。該數據集包括:
類別 |
中心貼片數量 |
光化性角化病 |
4936 |
頂泌汗腺 |
6739 |
基底細胞癌 |
6446 |
原位癌 |
5478 |
膠原蛋白 |
6262 |
表皮 |
7449 |
脂肪 |
6525 |
毛囊 |
8343 |
炎症 |
5856 |
浸潤性黑色素瘤 |
9101 |
卡波西肉瘤 |
4778 |
角蛋白 |
6418 |
原位黑色素瘤 |
4545 |
默克爾細胞癌 |
5968 |
肌肉 |
6051 |
壞死 |
6842 |
神經 |
4735 |
痣 |
8937 |
皮脂腺 |
6639 |
脂溢性角化病 |
10311 |
日光性彈性組織變性 |
7613 |
鱗狀細胞癌 |
6051 |
血管 |
7673 |
疣 |
6158 |
總數統計:
- 159,854 箇中心貼片
- 2,696,987 個總貼片(包括上下文貼片)
- 3,784 個用於標註的總切片
結果
器官 |
準確率 |
精確率 |
F1分數 |
皮膚 |
0.940 |
0.936 |
0.938 |
📄 許可證
該模型遵循 CC BY - NC 4.0 許可協議,僅用於 研究用途。
🔗 引用
如果你使用了該模型,請引用以下內容:
@misc{nechaev2025spidercomprehensivemultiorgansupervised,
title={SPIDER: A Comprehensive Multi-Organ Supervised Pathology Dataset and Baseline Models},
author={Dmitry Nechaev and Alexey Pchelnikov and Ekaterina Ivanova},
year={2025},
eprint={2503.02876},
archivePrefix={arXiv},
primaryClass={eess.IV},
url={https://arxiv.org/abs/2503.02876},
}
🔍 更多信息
若要探索其他模型和SPIDER數據集,可訪問項目的 Hugging Face HistAI頁面 或 GitHub。
📞 聯繫方式
- 作者:Dmitry Nechaev, Alexey Pchelnikov, Ekaterina Ivanova
- 郵箱:dmitry@hist.ai, alex@hist.ai, kate@hist.ai