torchxrayvision

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

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 259 🐛 待处理 33 👀 关注 15
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 1217
维护状态 较活跃
是否开源 开源
定价模式 🏷️完全免费 开源库,本地部署,完全免费

项目预览

torchxrayvision 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队mlmed
定价模式🏷️完全免费
价格说明开源库,本地部署,完全免费
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议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 星标★ 1217
上线时间2020-03-04 00:00:00
代码更新2026-10-18
仓库体积57.1 MB
本站数据更新2026-11-11
维护状态较活跃
收录时间2026-08-09
浏览次数5

项目介绍

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

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 1217

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

难度:入门 约 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

上一篇:MambaVision

下一篇:groundingLMM

同类项目推荐

ITK 开源

医学图像分割配准,C++/Python 直接调用

Insight Toolkit (ITK) -- Official Repository. ITK builds on a proven, spatially-ori···

★ 1675 2026-10-15
SurgicalDINO 开源

用基础模型适配器,让内窥镜图像秒变深度图,助力手术导航。

[IPCAI 2024 (IJCARS special issue)] Surgical-DINO: Adapter Learning of Foundation Mo···

★ 70 2026-08-09
SARATR-X 开源

用基础模型让雷达图像识别更准更省标注

A Foundation Model for SAR Target Recognition

★ 279 2026-08-09
bootcamp 开源

一套代码搞定图像、音频、分子等非结构化搜索

Dealing with all unstructured data, such as reverse image search, audio search, mole···

★ 2451 2026-08-09
GC-VLN 开源

不训练模型,用图约束让机器人听懂指令找路

[CoRL 2025] GC-VLN: Instruction as Graph Constraints for Training-free Vision-and-La···

★ 88 2026-08-09
traffic-monitor 开源

摄像头加雷达,边缘端自己算车流和车速

Traffic Monitor built with edge ML for object detection and radar for speed monitori···

★ 135 2026-09-20
CShade 开源

给 ReShade 装上 AI 引擎,实时提升游戏和视频画质

Motion graphics shaders for ReShade.

★ 65 2026-08-09
GenHowTo 开源

从教学视频学习动作生成,让 AI 看懂操作步骤

Code for the paper "GenHowTo: Learning to Generate Actions and State Transformations···

★ 54 2026-08-09
zed-sdk 开源

接上 ZED 相机,几行代码让机器人看懂三维空间

⚡️The spatial perception framework for rapidly building smart robots and spaces

★ 1246 2026-10-17