MedicalNet

小样本医学影像,用预训练3D-ResNet直接微调提精度

开源 free 行业应用
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 2272
维护状态 活跃
是否开源 是
定价模式 free

项目预览

项目数据

分类行业应用
开发团队Tencent
所属国家
官网地址
定价模式free
价格说明开源免费,提供3D-ResNet预训练模型及代码
访问状态
是否开源是
开源协议NOASSERTION
主要语言Python
技术栈/模型
GitHub 星标★ 2272
30天Star增速
HF 下载量
上线时间2019-07-17 00:00:00
代码更新—
仓库体积—
本站数据更新—
维护状态活跃
中文支持
访问方式
移动端支持
综合评分
收录时间2026-10-17
浏览次数0

项目介绍

MedicalNet 是一个面向医学影像的 3D 深度学习预训练模型项目,核心是提供一系列在医学 CT/MRI 数据上预训练的 3D-ResNet 权重及配套代码。医学影像标注成本极高、单中心数据量通常只有几十到几百例,直接训练 3D 网络极易过拟合。该项目通过大规模多病种医学数据预训练,让研究者在小样本任务上做迁移学习,从而提升分割、分类、检测等下游任务的精度。它解决了医学 3D 数据稀缺、从零训练困难的问题,提供不同深度(如 ResNet-10/18/34/50 等)的预训练模型,并给出加载权重、微调及推理的示例,是医学影像 AI 领域较早且被广泛引用的迁移学习基线。

Star 增长趋势

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

正在加载星标数据…

技术标签

使用教程

—

难度:进阶 约 15 分钟 部署方式:库/依赖 6 步

环境要求

  • Python 3.7.0(README 明确指定该版本)
  • pip(用于安装 requirements.txt 中的依赖)
  • PyTorch 环境(本项目为 Pytorch 实现,依赖由 requirements.txt 安装)
  • 可存放 3D 医学影像数据与预训练权重的磁盘空间

安装与启动步骤

  1. 1克隆代码仓库

    从 GitHub 拉取 MedicalNet 源码到本地任意目录,网络不通时可改用镜像或手动下载压缩包。

    git clone https://github.com/Tencent/MedicalNet.git
  2. 2进入项目目录

    切换到刚克隆下来的仓库根目录,后续安装与运行命令都在此目录执行。

    cd MedicalNet
  3. 3准备 Python 3.7.0

    按 README 要求安装 Python 3.7.0,建议用 conda 单独建虚拟环境,避免与其他项目的依赖冲突。

    conda create -n medicalnet python=3.7.0 -y
  4. 4安装项目依赖

    在激活的 Python 3.7 环境中执行依赖安装,requirements.txt 位于仓库根目录,安装过程较慢请耐心等待。

    pip install -r requirements.txt
  5. 5获取预训练权重

    README 节选中未给出权重下载链接,请到 GitHub 仓库页面获取 resnet_10/18/34/50_23dataset.pth,并按表格匹配参数字段。

  6. 6按参数加载权重

    以 23 数据集权重为例,模型名与参数一一对应:resnet_10_23dataset.pth 用 --model resnet --model_depth 10 --resnet_shortcut B。

关键配置

配置项必填说明示例
--model是指定骨干网络类型,预训练权重均为 resnetresnet
--model_depth是ResNet 深度,需与所用权重文件名一致10
--resnet_shortcut是残差捷径类型,深度 10/50 用 B,18/34 用 AB
resnet_10_23dataset.pth否基于 23 个医学数据集预训练的 ResNet-10 权重文件resnet_10_23dataset.pth
resnet_50_23dataset.pth否基于 23 个医学数据集预训练的 ResNet-50 权重文件resnet_50_23dataset.pth

如何确认成功

权重文件下载完成且与 --model / --model_depth / --resnet_shortcut 参数严格匹配,代码即可加载权重开展微调。

常见问题

Q:可以用 Python 3.8 及以上版本吗?

A:README 明确要求 Python 3.7.0,更高版本可能导致依赖不兼容,建议严格按该版本创建虚拟环境。

Q:预训练权重从哪里下载?

A:本次节选未包含下载地址,请到 MedicalNet 的 GitHub 仓库页面查找权重下载入口,不要使用来路不明的权重文件。

Q:不同深度模型该怎么选?

A:数据量越小越建议用浅层模型(如 depth 10)减少过拟合;注意 depth 10/50 用 shortcut B,18/34 用 shortcut A。

Q:能否用于自己的分割/分类任务?

A:可以,项目定位就是迁移学习基线,加载预训练权重后把最后的分类/分割头改成自己的类别数再微调即可。

注意事项

  • 项目依赖版本较旧(Python 3.7.0),建议使用独立 conda 虚拟环境安装,避免污染主环境。
  • 使用代码或预训练权重时需按 README 中的 BibTeX 引用 Med3D 论文。
  • 项目基于 MIT License 发布,商用前请阅读 LICENSE 文件确认条款。
  • README 节选未提供数据预处理与推理脚本的具体命令,实际训练命令请以仓库完整文档为准。

核心亮点

  • 提供多深度3D-ResNet预训练权重,可直接加载用于小样本医学任务迁移学习
  • 基于大规模多病种医学数据预训练,相比从零训练能显著缓解过拟合
  • 代码结构清晰,包含权重加载、微调与推理示例,便于快速复现和二次开发

不足之处

  • 预训练数据未完全开源,复现完整训练流程受限
  • 仅覆盖3D-ResNet系列,缺少Transformer等新架构的预训练模型

适用场景

  • 医院或高校用少量标注CT做肺结节/肝脏分割
  • 科研人员做医学影像分类的迁移学习基线对比
  • 企业快速搭建医学影像AI原型验证

替代项目

MONAI、nnU-Net

上一篇:worldmonitor

下一篇:finance-skills

同类项目推荐

xmgai-like 开源

换脸、艺术二维码随手玩,副业灵感一包带走。

本开源 AI 副业搞钱项目集成了强大功能,包括 AI 艺术二维码生成、AI 换脸、Delle3 绘画等···

★ 59 2026-08-09
spoken-to-signed-translation 开源

把口语文本一路转成手语视频的流水线

a text-to-gloss-to-pose-to-video pipeline for spoken to signed language translation

★ 103 2026-09-11
hub 开源

已退役的YOLO训练平台,迁移到Ultralytics Platform

Historical Ultralytics HUB repository — HUB shut down on July 31, 2026 and was repl···

★ 249 2026-08-12