PyTorch医学图像分割开源库

发布时间:2021-10-28  栏目:医学, 图像处理, 机器视觉, 深度学习  评论:0 Comments

PyTorch医学图像分割开源库

该库特点:

  1. 支持2D和3D医学图像分割,可以修改hparam.py文件来确定是2D分割还是3D分割以及是否可以进行多分类。
  2. 支持绝大数主流分割模型,几乎提供了所有的2D和3D分割的算法。
  3. 兼容几乎所有的医学数据格式(例如 nii.gz, nii, mhd, nrrd, …),修改hparam.py的fold\_arch即可。

作者提供了训练和测试推断的代码,简单配置后训练和推断都仅需要一行命令。

已包含的分割模型:

  • 2D
  •  unet
  •  unet++
  •  miniseg
  •  segnet
  •  pspnet
  •  highresnet
  •  deeplab
  •  fcn
  • 3D
  •  unet3d
  •  densevoxelnet3d
  •  fcn3d
  •  vnet3d
  •  highresnert
  •  densenet3d

留下评论

You must be logged in to post a comment.

相册集

pix pix pix pix pix pix

关于自己

杨文龙,微软Principal Engineering Manager, 曾在各家公司担任影像技术资深总监、数据科学团队资深经理、ADAS算法总监、资深深度学习工程师等职位,热爱创新发明,专注于人工智能、深度学习、图像处理、机器学习、算法、自然语言处理及软件等领域,目前发明有国际专利19篇,中国专利28篇。

联系我

个人技术笔记

welonshen@gmail.com

2015 in Shanghai