3 个月前

IterNet:利用血管网络中结构冗余进行视网膜图像分割

IterNet:利用血管网络中结构冗余进行视网膜图像分割

摘要

视网膜血管分割在视网膜血管疾病诊断中具有重要意义。为进一步提升血管分割性能,本文提出一种基于UNet架构的新模型——IterNet,该模型能够从已分割的血管图像中自动识别出被遮蔽的细微结构,而非依赖原始输入图像。IterNet由多个迭代的微型UNet组成,其深度可达普通UNet的4倍。此外,IterNet采用权值共享与跳跃连接机制,有效促进模型训练;因此,即使在如此庞大的网络结构下,模型仅需10至20张标注图像即可完成学习,无需预训练或任何先验知识。在三个主流数据集(DRIVE、CHASE-DB1和STARE)上,IterNet分别取得了0.9816、0.9851和0.9881的AUC值,目前在相关文献中表现最优。源代码已公开。

代码仓库

amri369/Pytorch-Iternet
pytorch
GitHub 中提及

基准测试

基准方法指标
retinal-vessel-segmentation-on-chase_db1IterNet
AUC: 0.9851
F1 score: 0.8073
retinal-vessel-segmentation-on-driveIterNet
AUC: 0.9816
F1 score: 0.8205

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
IterNet:利用血管网络中结构冗余进行视网膜图像分割 | 论文 | HyperAI超神经