CE网
该手稿已被TMI接受。
请在运行main.py之前启动“ visdom”。 然后,运行main.py文件。
我们已上传DRIVE数据集以运行视网膜血管检测。 其他医学数据集将在下一次提交中上载。
提交的文件主要包括:
network / cenet.py中的体系结构(称为CE-Net)
loss.py中的多类骰子损失
data.py中的数据扩充
更新:我们修改了损失功能。 不会发生cuda错误(或警告)。
更新:测试代码已上传。 此外,我们发布了一个预训练模型,该模型在DRIVE数据集的AUC评分中达到0.9819。
1