微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

MedicalNet 3D 医疗影像预训练模型

程序名称:MedicalNet

授权协议: MIT

操作系统: 跨平台

开发语言: Python

MedicalNet 介绍

MedicalNet 是一个 3D 医疗影像预训练模型,其将多个 3D 医疗数据集集合成大数据集,基于此数据集提供了完整的 3D-resnet
系列预训练模型与相应的迁移学习训练代码

适用场景

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

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

特性

  • 支持单卡与多卡训练

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

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

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

环境依赖

  • Python 3.7.0
  • PyTorch-0.4.1
  • CUDA Version 9.0
  • CUDNN 7.0.5

性能

Network Pretrain LungSeg(Dice) NoduleCls(accuracy)
3D-resnet10 Train from scratch 71.30% 79.80%
MedicalNet 87.16% 86.87%
3D-resnet18 Train from scratch 75.22% 80.80%
MedicalNet 87.26% 88.89%
3D-resnet34 Train from scratch 76.82% 83.84%
MedicalNet 89.31% 89.90%
3D-resnet50 Train from scratch 71.75% 84.85%
MedicalNet 93.31% 89.90%
3D-resnet101 Train from scratch 72.10% 81.82%
MedicalNet 92.79% 90.91%
3D-resnet152 Train from scratch 73.29% 73.74%
MedicalNet 92.33% 90.91%
3D-resnet200 Train from scratch 71.29% 76.77%
MedicalNet 92.06% 90.91%

MedicalNet 官网

https://github.com/Tencent/MedicalNet

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐