🚀 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