torchxrayvision

开箱即用的胸部 X 光 AI 模型库,快速训练和部署诊断模型

TorchXRayVision 是一个面向胸部 X 光影像的开源工具库,集成了多个公开数据集和预训练模型,覆盖分类、分割和自编码器三大任务。它解决了医学影像研究中数据获取难、模型复现成本高的问题,提供统一的数据加载接口和标准化的模型权重,支持肺炎、肺结节、心脏病等多种胸部疾病的自动诊断。核心能力包括:一键加载 CheXpert、NIH ChestX-ray14、RSNA 等主流数据集,直接使用在大型数据集上预训练好的模型进行迁移学习,以及提供可解释性工具辅助医生决策。该库基于 PyTorch 构建,适合研究人员和开发者快速搭建胸部 X 光 AI 应用。

开源 free 开源模型
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 1197
维护状态 维护中
是否开源 是
定价模式 free

项目数据

分类开源模型
开发团队mlmed
所属国家
定价模式free
价格说明开源库,本地部署,完全免费
访问状态
是否开源是
开源协议NOASSERTION
主要语言Jupyter Notebook
技术栈/模型chest-radiographs,chest-xray,chest-xray-images,cxr,cxr-images,dataset,deep-learning,foundation-models,image-classification,machine-learning,medical,medical-ai,medical-application,medical-image-analysis,medical-image-processing,medical-imaging,pytorch,torchxrayvision,transfer-learning
GitHub 星标★ 1197
30天Star增速
HF 下载量
上线时间2020-03-04 00:00:00
最近更新2026-09-22 00:00:00
维护状态维护中
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数4

使用教程

难度:入门 约 10 分钟 部署方式:库/依赖 3 步

环境要求

  • 可用的 Python 环境(含 pip,用于执行 pip install)
  • 已安装 PyTorch(README 示例中导入了 torch、torchvision)
  • 已安装 scikit-image(README 示例中导入了 skimage)

安装与启动步骤

  1. 1安装库

    README 只给出这一条安装命令,通过 pip 从 PyPI 安装 torchxrayvision 包。建议在虚拟环境中执行,避免污染全局依赖。

    pip install torchxrayvision
  2. 2验证导入

    安装完成后导入该库,确认包已正确安装且无缺失依赖报错。README 示例中还会同时导入 skimage、torch、torchvision。

    python -c "import torchxrayvision as xrv; print('torchxrayvision import ok')"
  3. 3查阅官方文档

    README 指向在线文档,数据集加载、预训练模型等进阶用法需以官方文档为准,README 中未给出更多命令。

如何确认成功

执行 python -c "import torchxrayvision as xrv" 无报错,说明库安装成功。

常见问题

Q:安装命令是什么?

A:README 提供的安装方式为 pip install torchxrayvision,除此之外未列出其他安装步骤。

Q:有官方文档吗?

A:有。README 中给出文档地址 https://mlmed.org/torchxrayvision/ ,并提示 Documentation now online。

Q:有对应论文吗?

A:有。README 标注论文已上线,地址为 https://arxiv.org/abs/2111.00595 。

Q:怎么用预训练模型和数据集?

A:README 仅说明库提供公开胸部 X 光数据集统一接口和预训练模型,具体调用方式需查看官方文档。

注意事项

  • pip 包名 torchxrayvision 与 GitHub 仓库名一致,可直接安装。
  • README 节选中未给出预训练模型权重名称、数据集路径、端口等参数,不要臆造。
  • 官方文档:https://mlmed.org/torchxrayvision/ ,论文:https://arxiv.org/abs/2111.00595 。

核心亮点

  • 集成多个主流胸部 X 光数据集,统一接口,省去数据预处理时间
  • 提供预训练分类、分割和自编码器模型,可直接迁移学习
  • 支持模型可解释性分析,帮助理解诊断依据

不足之处

  • 主要针对胸部 X 光,领域较窄,不适用于其他医学影像
  • 依赖 PyTorch 生态,对 TensorFlow 用户不友好
  • 文档/社区待观察

适用场景

  • 科研人员快速复现胸部 X 光论文实验
  • 开发肺炎、肺结节等疾病的自动筛查系统
  • 医学影像教学和模型可解释性研究

替代项目

MONAI、MedicalZoo、XrayGLM

项目介绍

torchxrayvision 是计算机视觉领域的开源项目,由 mlmed 开发,2020 年首次发布。

在全站 13,014 个收录项目中,它的 GitHub 星标数(1,197)位列前 30%,在计算机视觉分类中处于中上游。

项目仍在小幅维护中,最近一次代码更新于 2026-09-22。开源库,本地部署,完全免费。

它主要面向的使用场景是:科研人员快速复现胸部X光论文实验。同类可对比的替代方案包括 MONAI、MedicalZoo、XrayGLM。

上一篇:MambaVision

下一篇:groundingLMM

同类项目推荐

modlens 开源

给纯文本编码代理装上眼睛,粘贴图片即刻获得结构化视觉证据。

The first vision plugin for DeepSeek Harness, and the vision bridge for every text-o···

★ 4019 2026-08-19
opencv 开源

开箱即用的视觉算法库,搞定图像视频处理与识别

Open Source Computer Vision Library

★ 90930 2026-08-10
OpenCVTutorials 开源

中文 OpenCV 教程,从入门到实战,边看边跑代码

OpenCV-Python4.1 中文文档

★ 1430 2026-08-20
DEIMv2 开源

用DINOv3做实时检测,又快又准,直接落地。

[DEIMv2] Real Time Object Detection Meets DINOv3

★ 2046 2026-08-14