【开源公告】3D医疗影像预训练模型MedicalNet开源

许多研究表明,深度学习的发展非常依赖数据量,在医疗图像领域,目前缺乏基于大数据基础的专用预训练模型。

本项目 MedicalNet 将多个 3D 医疗数据集集合成大数据集,基于此数据集提供了完整的 3D-ResNet 系列预训练模型与相应的迁移学习训练代码。

MedicalNet适用的场景

  • MedicalNet 提供的预训练网络可迁移到任何 3D 医疗影像的 AI 应用中,包括但不限于分割、检测、分类等任务。

  • 尤其适用小数据医疗影像 AI 场景,能加快网络收敛,提升网络性能。

MedicalNet 功能介绍

  • 支持单卡、多卡训练

  • 支持 Python3.7 PyTorch-0.4.1

  • 支持任何 3D 医疗影像的迁移学习任务

  • 支持 3D-ResNet 全系列模型的迁移学习任务

  • 支持多种感受野的迁移学习

MedicalNet性能展示

上图是在相同的迭代次数下,不同预训练方式的测试结果,相对于从头训练,MedicalNet能明显加快网络收敛速度,提升性能。 更多细节请参考文章[Med3D: Transfer Learning for 3D Medical Image Analysis](https://arxiv.org/abs/1904.00625)。

MedicalNet 未来规划

  • 收集更多数据提升预训练模型性能

  • 增加 3D 超声预训练数据

  • 完成 3D 轻量级( 3D-MobileNet 系列, 3D-ShuffleNet 系列等)预训练模型

  • 完成 2D 医学预训练模型

数据量是全球医疗影像AI落地的最大痛点之一,我们期望通过建立一个平台为千万的医疗影像AI提供大数据基础模型支持。这是一个良好的开端,如果您在使用过程中发现Bug或者有新特性建议,欢迎fork并提交Merge Request来解决问题,我们期待并感谢您的贡献。

MedicalNet 正式开源

Github 开源地址:

https://github.com/Tencent/MedicalNet

(点击文末阅读原文直接访问)

请给 MedicalNet 一个 Star !

欢迎提出你的 issue 和 PR!

NeuralClassifier 国内镜像地址:

https://git.code.tencent.com/Tencent_Open_Source/MedicalNet

腾讯工蜂源码系统为开源开发者提供完整、最新的腾讯开源项目国内镜像

我来评几句
登录后评论

已发表评论数()

相关站点

+订阅
热门文章