基于多任务学习的青光眼智能诊断

魏宏博; 武劲圆; 陈磊; 冯梓毅; 游国栋

doi:10.13364/j.issn.1672-6510.20230080

天津科技大学学报 ›› 2024, Vol. 39 ›› Issue (02) : 59 -64. DOI: 10.13364/j.issn.1672-6510.20230080

基于多任务学习的青光眼智能诊断

魏宏博, 武劲圆, 陈磊, 冯梓毅, 游国栋

作者信息 +

Author information +

文章历史 +

PDF (1749K)

摘要

为了提高青光眼检测的准确率，降低青光眼的危害，本文提出一种基于多任务学习的青光眼智能诊断方法，将U-Net网络和VGG16网络结合，U-Net网络和VGG16网络共用U-Net网络的编码器部分，通过U-Net网络得到杯盘比(cup-to-discratio,CDR)，并且将CDR作为眼底图像的特征之一输入VGG16网络，实现眼底图像的青光眼分类。实验使用REFUGE挑战数据集进行验证，网络模型在训练后得到的工作特性曲线下面积为0.978 8，且视盘和视杯的分割准确率分别达到0.874 5和0.962 4，对比其他使用相同数据集的方法，本方法具有更高的青光眼分类准确率。