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

本项目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

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

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

文章来源于腾讯云开发者社区,点击查看原文