
SA-UNet: 开源的SA-UNet代码
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:SA-UNet是一款开源的深度学习模型代码,基于U型网络架构并引入了自适应模块,专为医学图像分割任务优化设计。
该代码适用于论文《用于视网膜血管分割的空间注意U-Net》。我们在DRIVE和CHASE DB1数据集上报告了最新的性能结果。此代码由布达佩斯技术经济大学(CME)的郭长禄编写。
我们的训练与评估在Ubuntu 16.04系统下进行,但同样适用于Windows和其他操作系统环境。对于数据扩充部分包括:(1) 随机旋转; (2) 增加高斯噪声; (3) 色彩抖动;以及(4)水平、垂直和对角线翻转。
如果您不需要上述的数据增强操作,可以直接获取代码进行使用。训练时可以运行相关脚本,测试或评估则可以通过相应的命令执行。
环境需求:Keras 2.3.1 和 Tensorflow = 1.14.0
关于Keras:这是一个简洁且模块化程度高的神经网络库,用Python编写,并可在TensorFlow或者Theano上运行。它的设计重点在于快速实验与实现想法的便捷性。
全部评论 (0)
还没有任何评论哟~


