【技术实现步骤摘要】
一种眼底图像质量增强模型的训练方法及装置
[0001]本申请属于图像处理
,尤其涉及一种眼底图像质量增强模型的训练方法及装置。
技术介绍
[0002]眼底图像对于眼底疾病的诊断有着重要意义,凭借着其成本低廉、易推广的优势,被广泛运用于各层级医疗机构。然而,由于眼底相机的传感器限制以及临床拍摄中光照等因素,眼底图像质量不均,而低质量眼底图像将会严重影响自动诊断算法(如:神经网络)的性能以及医生的判断。对于深度学习模型来说,其诊断及预测性能极度依赖训练数据的样本量及质量,低质量眼底图像无疑会严重干扰深度学习模型的准确性。而另一方面,眼底图像的获取成本非常高,在临床上很难对同一个病人进行多次拍摄以追求高质量眼底图像。
[0003]目前,针对眼底图像质量增强的算法主要有两种思路,第一种是依据眼底图像成像原理,使用传统计算机视觉中的滤波和卷积操作,结合先验知识,实现对某种特定低质量特征的增强;第二种方法是基于数据驱动,通过训练深度学习模型,学习低质量眼底图像到高质量眼底图像的映射。
[0004]但是第一种方法的技 ...
【技术保护点】
【技术特征摘要】
1.一种眼底图像质量增强模型的训练方法,其特征在于,包括:获取用于模型训练的第一眼底图像和第二眼底图像,所述第一眼底图像的图像质量优于所述第二眼底图像的图像质量;对所述第一眼底图像进行质量退化处理,获得与所述第一眼底图像对应的第三眼底图像;采用所述第一眼底图像、所述第二眼底图像和所述第三眼底图像,对预设的眼底图像质量增强网络进行半监督训练;采用预设的对抗网络对半监督训练后的所述眼底图像质量增强网络进行对抗训练;基于预设的损失函数,计算所述眼底图像质量增强网络的损失值;当所述损失值满足预设条件时,停止对所述眼底图像质量增强网络的训练,得到最终的眼底图像质量增强模型。2.如权利要求1所述的方法,其特征在于,所述采用所述第一眼底图像、所述第二眼底图像和所述第三眼底图像,对预设的眼底图像质量增强网络进行半监督学习,包括:采用所述第一眼底图像和所述第三眼底图像,对所述眼底图像质量增强网络进行监督训练;采用所述第一眼底图像和所述第二眼底图像,对所述眼底图像质量增强网络进行无监督训练。3.如权利要求2所述的方法,其特征在于,所述采用所述第一眼底图像和所述第三眼底图像,对所述眼底图像质量增强网络进行监督训练,包括:将所述第三眼底图像输入到所述眼底图像质量增强网络中,得到所述第三眼底图像的第三增强图像;采用预设的重要性评估算法计算所述第三眼底图像中的每个像素点的重要性;根据所述第三增强图像、所述第一眼底图像和所述第三眼底图像中每个像素点的重要性,对所述眼底图像质量增强网络的参数进行调整。4.如权利要求3所述的方法,其特征在于,所述采用所述第一眼底图像和所述第二眼底图像,对所述眼底图像质量增强网络进行无监督训练,包括:将所述第二眼底图像输入到所述眼底图像质量增强网络中,得到所述第二眼底图像的第二增强图像;采用预设的重要性评估算法计算所述第二眼底图像中的每个像素点的重要性;根据所述第二增强图像和所述第二眼底图像中每个像素点的重要性,对所述眼底图像质量增强网络的参数进行调整。5.如权利要求4所述的方法,其特征在于,所述对抗网络包括判别器,所述采用预设的对抗网络对半监督训练后的所述质量增强网络进行对抗训练,包括:以所述第一眼底图像作为基准,采用所述判别器判断所述第二增强图像的图像质量;根据所述第二增强图像的图像质量,对所述眼底图像质量增强网络的参数进行调整。6.如权利要求3或5任一项所述的方法,其特征在于,所述损失...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。