1.一种视杯视盘分割模型训练方法,其特征在于,包括:
获取样本图像和所述样本图像对应的图像标签,以根据所述样本图像和所述样本图像对应的图像标签构建样本数据;
将所述样本数据输入预设的神经网络,以得到预测的视杯视盘分割图像;
对所述图像标签和预测的视杯视盘分割图像分别进行投影,以得到所述图像标签对应的标签投影值和所述预测的视杯视盘分割图像的图像投影值;其中,所述投影为沿投影方向进行正投影;
基于所述预测的视杯视盘分割图像和对应的图像标签,利用分割损失函数公式计算分割损失函数的数值,以及基于所述标签投影值和所述图像投影值,利用投影损失函数公式计算投影损失函数的数值,以得到网络损失函数的数值,其中,所述分割损失函数用于计算所述预测的视杯视盘分割图像和对应的图像标签之间的损失,所述投影损失函数用于计算所述标签投影值和所述图像投影值之间的损失;
根据所述网络损失函数的数值对所述预设的神经网络进行训练,以得到视杯视盘分割模型。
2.根据权利要求1所述的视杯视盘分割模型训练方法,其特征在于,所述分割损失函数公式为:Lseg=‑[ytrue log ypred+(1‑ytrue)log(1‑ypred)]其中,Lseg表示分割损失函数的数值,ypred表示预设的神经网络根据样本图像预测出的视杯视盘分割图像,ytrue表示所述样本图像对应的图像标签。
3.根据权利要求1所述的视杯视盘分割模型训练方法,其特征在于,所述投影损失函数公式为:Lproj=||ptrue‑ppred||2
其中,Lproj表示投影损失函数的数值,ptrue表示标签投影值,ppred表示图像投影值。
4.根据权利要求1所述的视杯视盘分割模型训练方法,其特征在于,所述预设的神经网络包括特征提取层、卷积层和解码层;所述将所述样本数据输入预设的神经网络,以得到预测的视杯视盘分割图像,包括:通过所述特征提取层对所述样本图像进行特征提取,以得到所述样本图像对应的第一特征图像;
通过所述卷积层对所述第一特征图像进行卷积操作,以得到代表视杯视盘分割图像的第二特征图像;
将所述第二特征图像输入所述解码层,以得到预测的视杯视盘分割图像。
5.根据权利要求1所述的视杯视盘分割模型训练方法,其特征在于,在所述将所述样本数据输入预设的神经网络之前,所述方法包括:对所述样本图像进行预处理,所述预处理包括伸缩处理。
6.一种基于神经网络的杯盘比确定方法,其特征在于,包括:
获取眼底图像,并对所述眼底图像进行视盘区域检测,以得到视盘区域;
将所述视盘区域输入预先训练的视杯视盘分割模型,得到视杯视盘分割图像,所述视杯视盘分割模型为采用权利要求1至5中任一项所述的视杯视盘分割模型训练方法训练得到的模型;
基于所述视杯视盘分割图像确定杯盘比。
7.一种视杯视盘分割模型训练装置,其特征在于,包括:
样本构建模块,用于获取样本图像和所述样本图像对应的图像标签,以根据所述样本图像和所述样本图像对应的图像标签构建样本数据;
图像预测模块,用于将所述样本数据输入预设的神经网络,以得到预测的视杯视盘分割图像;
图像投影模块,用于对所述图像标签和预测的视杯视盘分割图像分别进行投影,以得到所述图像标签对应的标签投影值和所述预测的视杯视盘分割图像的图像投影值;其中,所述投影为沿投影方向进行正投影;
损失计算模块,用于基于所述预测的视杯视盘分割图像和对应的图像标签,利用分割损失函数公式计算分割损失函数的数值,以及基于所述标签投影值和所述图像投影值,利用投影损失函数公式计算投影损失函数的数值,以得到网络损失函数的数值,其中,所述分割损失函数用于计算所述预测的视杯视盘分割图像和对应的图像标签之间的损失,所述投影损失函数用于计算所述标签投影值和所述图像投影值之间的损失;
模型训练模块,用于根据所述网络损失函数的数值对所述预设的神经网络进行训练,以得到视杯视盘分割模型。
8.一种基于神经网络的杯盘比确定装置,其特征在于,包括:
图像检测模块,用于获取眼底图像,并对所述眼底图像进行视盘区域检测,以得到视盘区域;
网络预测模块,用于将所述视盘区域输入预先训练的视杯视盘分割模型,得到视杯视盘分割图像,所述视杯视盘分割模型为采用权利要求1至5中任一项所述的视杯视盘分割模型训练方法训练得到的模型;
杯盘比确定模块,用于基于所述视杯视盘分割图像确定杯盘比。
9.一种计算机设备,其特征在于,所述计算机设备包括存储器和处理器;
所述存储器用于存储计算机程序;
所述处理器,用于执行所述计算机程序并在执行所述计算机程序时实现如权利要求1至5中任一项所述的视杯视盘分割模型训练方法,或者实现如权利要求6所述的基于神经网络的杯盘比确定方法。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时使所述处理器实现如权利要求1至5中任一项所述的视杯视盘分割模型训练方法,或者实现如权利要求6所述的基于神经网络的杯盘比确定方法。