【文件属性】：
文件名称：SA-UNet:SA-UNet的开源代码
文件大小：11.38MB
文件格式：ZIP
更新时间：2021-05-04 18:47:42
Python
概述
该代码适用于以下论文：用于视网膜血管分割的空间注意U-Net。 我们报告DRIVE和CHASE DB1数据集的最新性能。
布达佩斯技术经济大学（CME）的郭长禄（Changlu Guo）编写的代码。
我们在Ubuntu 16.04上进行培训和评估，它也适用于Windows和OS。
数据集
数据扩充：
（1）随机轮换； （2）增加高斯噪声； （3）色彩抖动； 2. （4）水平，垂直和对角线翻转。
如果您不想进行上述扩充，只需从我的链接下载即可。
快速开始
训练
运行或
测验
运行或Eval_chase.py
环境环境
凯拉斯2.3.1 Tensorflow = = 1.14.0
关于凯拉斯
Keras是一个极简的，高度模块化的神经网络库，用Python编写，能够在TensorFlow或Theano上运行。 它的开发着重于实现快速实验。 能够以最小的延迟将想法付诸
【文件预览】：
SA-UNet-master
----SA-UNet.png(29KB)
----requirements.txt(401B)
----keras_dataAug.py(6KB)
----Spatial_Attention.py(1KB)
----Eval_drive.py(1B)
----Model()
--------CHASE()
--------DRIVE()
----SA_UNet.py(9KB)
----Train_chase.py(5KB)
----README.md(3KB)
----.idea()
--------remote-mappings.xml(263B)
--------misc.xml(379B)
--------vcs.xml(180B)
--------deployment.xml(491B)
--------SA-UNet.iml(398B)
--------modules.xml(266B)
----CHASE()
--------download(124B)
----util.py(1KB)
----Train_drive.py(4KB)
----flip.py(476B)
----Eval_chase.py(1B)
----Dropblock.py(8KB)
----DRIVE()
--------download(83B)