CN113538470B - 一种基于神经网络的图像层间边界确定方法及装置 - Google Patents
一种基于神经网络的图像层间边界确定方法及装置 Download PDFInfo
- Publication number
- CN113538470B CN113538470B CN202110666798.3A CN202110666798A CN113538470B CN 113538470 B CN113538470 B CN 113538470B CN 202110666798 A CN202110666798 A CN 202110666798A CN 113538470 B CN113538470 B CN 113538470B
- Authority
- CN
- China
- Prior art keywords
- boundary
- image
- result
- boundary position
- neural network
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
- 238000013528 artificial neural network Methods 0.000 title claims abstract description 85
- 239000011229 interlayer Substances 0.000 title claims abstract description 77
- 238000000034 method Methods 0.000 title claims abstract description 70
- 238000013527 convolutional neural network Methods 0.000 claims abstract description 33
- 239000010410 layer Substances 0.000 claims description 104
- 238000012549 training Methods 0.000 claims description 40
- 238000012937 correction Methods 0.000 claims description 31
- 230000011218 segmentation Effects 0.000 claims description 30
- 238000012545 processing Methods 0.000 claims description 23
- 238000003062 neural network model Methods 0.000 claims description 18
- 238000009499 grossing Methods 0.000 claims description 11
- 238000010606 normalization Methods 0.000 claims description 8
- 238000004364 calculation method Methods 0.000 claims description 7
- 230000004913 activation Effects 0.000 claims description 5
- 230000009286 beneficial effect Effects 0.000 abstract description 10
- 238000012014 optical coherence tomography Methods 0.000 description 10
- 238000004590 computer program Methods 0.000 description 8
- 238000012216 screening Methods 0.000 description 7
- 230000008569 process Effects 0.000 description 6
- 238000004422 calculation algorithm Methods 0.000 description 5
- 238000010586 diagram Methods 0.000 description 5
- 230000002349 favourable effect Effects 0.000 description 4
- 238000004458 analytical method Methods 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- 230000006870 function Effects 0.000 description 3
- 238000002372 labelling Methods 0.000 description 3
- 238000011176 pooling Methods 0.000 description 3
- 238000007476 Maximum Likelihood Methods 0.000 description 2
- 230000008859 change Effects 0.000 description 2
- 238000006243 chemical reaction Methods 0.000 description 2
- 201000010099 disease Diseases 0.000 description 2
- 208000037265 diseases, disorders, signs and symptoms Diseases 0.000 description 2
- 238000003384 imaging method Methods 0.000 description 2
- 238000007477 logistic regression Methods 0.000 description 2
- 238000013139 quantization Methods 0.000 description 2
- 210000001525 retina Anatomy 0.000 description 2
- 230000004256 retinal image Effects 0.000 description 2
- 238000005070 sampling Methods 0.000 description 2
- 238000007619 statistical method Methods 0.000 description 2
- 241000579895 Chlorostilbon Species 0.000 description 1
- 238000001514 detection method Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 238000003745 diagnosis Methods 0.000 description 1
- 239000006185 dispersion Substances 0.000 description 1
- 230000000694 effects Effects 0.000 description 1
- 239000010976 emerald Substances 0.000 description 1
- 229910052876 emerald Inorganic materials 0.000 description 1
- 238000000605 extraction Methods 0.000 description 1
- 238000010191 image analysis Methods 0.000 description 1
- 238000003709 image segmentation Methods 0.000 description 1
- 230000001788 irregular Effects 0.000 description 1
- 239000010977 jade Substances 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 238000007781 pre-processing Methods 0.000 description 1
- ZLIBICFPKPWGIZ-UHFFFAOYSA-N pyrimethanil Chemical compound CC1=CC(C)=NC(NC=2C=CC=CC=2)=N1 ZLIBICFPKPWGIZ-UHFFFAOYSA-N 0.000 description 1
- 230000002207 retinal effect Effects 0.000 description 1
- 239000010979 ruby Substances 0.000 description 1
- 229910001750 ruby Inorganic materials 0.000 description 1
- 230000003068 static effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 238000012360 testing method Methods 0.000 description 1
- 239000013598 vector Substances 0.000 description 1
- 230000000007 visual effect Effects 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/12—Edge-based segmentation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- Biomedical Technology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种基于神经网络的图像层间边界确定方法及装置,该方法包括:将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,该预测边界结果包括N个分层边界以及与分层边界对应的N个边界通道;对于任一边界通道中的边界位置,确定与该边界位置相对应的目标像素点集;计算目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度;根据该边界位置对应的分布集中度,确定该边界位置的确定度信息。可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,有利于提高对图像分层边界识别的准确性。
Description
技术领域
本发明涉及图像分割技术领域,尤其涉及一种基于神经网络的图像层间边界确定方法及装置。
背景技术
随着计算机和医疗技术的快速发展,OCT(Optical Coherence Tomography,光学相干层析成像)技术已被广泛应用于眼底疾病的诊断设备中,对眼科疾病的检测和治疗有着重要意义。OCT属于一种高灵敏度、高分辨率、高速度、无入侵的断层扫描成像方式,利用光的相干性对眼底扫描成像,每次扫描被称为一个A-scan,相邻连续的多次扫描组合在一起称为一个B-scan,B-scan也即通常所看到的OCT截面图,是医学诊断中OCT最主要的成像方式。
在对OCT截面图(如OCT视网膜图像)进行分析处理的实际应用中,精确并自动分割OCT截面图中不同组织层并获取组织层的边界(也即分层边界)具有重大意义。现有技术中,对于分层边界的识别,主要是采用图像搜索或图像识别来实现的,通过搜索或识别图像的灰度、梯度、扫描轮廓等特征达到分层边界的识别目的。然而实践发现,这种方法无法确定识别出的分层边界的置信度,存在分层边界被误判的风险。
发明内容
本发明提供了一种基于神经网络的图像层间边界确定方法及装置,能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,提高对图像分层边界识别的准确性及效率。
为了解决上述技术问题,本发明第一方面公开了一种基于神经网络的图像层间边界确定方法,所述方法包括:
将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,所述预测边界结果包括所述待检测图像的N个分层边界、与N个所述分层边界对应的N个边界通道以及N个所述边界通道分别对应的概率值集合,其中,所述N大于等于1,每个所述边界通道对应的概率值集合包括所述待检测图像中所有像素点属于每个所述边界通道的概率值;
对于任一所述边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,根据所述预测边界结果,计算所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度;
对于任一所述边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,该边界位置的确定度信息用于确定该边界位置的准确度。
作为一种可选的实施方式,在本发明第一方面中,所述对于任一所述边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,包括:
对于任一所述边界通道中的任一边界位置,确定该边界通道对应的分层边界的分布方向,当确定出该边界通道对应的分层边界的分布方向是横向时,将该边界通道中该边界位置处的整列像素点确定为与该边界位置相对应的目标像素点集;当确定出该边界通道对应的分层边界的分布方向是纵向时,将该边界位置处的整行像素点确定为与该边界位置相对应的目标像素点集。
作为一种可选的实施方式,在本发明第一方面中,所述根据所述预测边界结果,计算所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度,包括:
根据所述预测边界结果,对所述目标像素点集中所有像素点属于该边界通道的概率值执行Softmax回归操作,得到所述目标像素点集中所有像素点属于该边界通道的回归概率值;
对所述目标像素点集中所有像素点属于该边界通道的回归概率值执行概率分布统计操作,得到所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,将所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度确定为该边界位置对应的分布集中度。
作为一种可选的实施方式,在本发明第一方面中,所述对于任一所述边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,包括:
对于任一所述边界通道中的任一边界位置,判断该边界位置的分布集中度是否大于预设阈值,当判断出该边界位置的分布集中度大于所述预设阈值时,将用于表示该边界位置为实际边界位置的标识信息确定为该边界位置的确定度信息;
当判断出该边界位置的分布集中度小于等于所述预设阈值时,将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息。
作为一种可选的实施方式,在本发明第一方面中,对于任一所述边界通道中的任一边界位置,所述将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息之后,所述方法还包括:
根据该边界位置,从所述待检测图像中确定出局部难例样本图片;
将所述局部难例样本图片输入至第二卷积神经网络模型,得到该边界位置对应的纠偏量;
根据该边界位置对应的纠偏量,计算该边界位置对应的纠偏边界位置,并将所述纠偏边界位置确定为实际边界位置。
作为一种可选的实施方式,在本发明第一方面中,所述第一卷积神经网络模型包括:
深度为M的编码层以及与M层所述编码层对应的M层解码层,所述M大于等于1;
其中,相邻两层所述编码层之间通过下采样单元连接,相邻两层所述解码层之间通过上采样单元连接;
其中,每个所述编码层以及每个所述解码层均包括若干卷积单元,每个所述卷积单元中依次包括卷积层、归一化层、激活层。
作为一种可选的实施方式,在本发明第一方面中,在所述将待检测图像输入至第一卷积神经网络模型之前,所述方法还包括:
对预训练数据集执行图像增强操作,得到目标训练数据集;
将所述目标训练数据集输入到预置神经网络中,得到预分割结果,所述预分割结果包括图像层分类结果以及图像层间边界结果;
计算所述预分割结果与预先确定出的标记结果之间的联合损失,所述预先确定出的标记结果包括预先标记分类结果以及预先标记层间边界结果;
将所述联合损失进行反向传播,通过预设周期长度的迭代训练得到第一卷积神经网络模型。
作为一种可选的实施方式,在本发明第一方面中,所述联合损失的计算公式为:
L=λlabel_diceLlabel_dice+λlabel_ceLlabel_ce+λbd_ceLbd_ce+λbd_l1Lbd_l1;
其中,L为所述联合损失,Llabel_dice为所述图像层分类结果与所述预先标记分类结果之间的Dice损失,λlabel_dice为所述Dice损失对应的权重系数,Llabel_ce为所述图像层分类结果与所述预先标记分类结果之间的分类交叉熵损失,λlabel_ce为所述分类交叉熵损失对应的权重系数,Lbd_ce为所述图像层间边界结果与所述预先标记层间边界结果之间的边界交叉熵损失,λbd_ce为所述边界交叉熵损失对应的权重系数,Lbd_l1为所述图像层间边界结果与所述预先标记层间边界结果之间的平滑损失,λbd_l1为所述平滑损失对应的权重系数。
本发明第二方面公开了一种基于神经网络的图像层间边界确定装置,所述装置包括:
第一处理模块,用于将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,所述预测边界结果包括所述待检测图像的N个分层边界、与N个所述分层边界对应的N个边界通道以及N个所述边界通道分别对应的概率值集合,其中,所述N大于等于1,每个所述边界通道对应的概率值集合包括所述待检测图像中所有像素点属于每个所述边界通道的概率值;
第一确定模块,用于对于任一所述边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,根据所述预测边界结果,计算所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度;
第二确定模块,用于对于任一所述边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,该边界位置的确定度信息用于确定该边界位置的准确度。
作为一种可选的实施方式,在本发明第二方面中,对于任一所述边界通道中的任一边界位置,所述第一确定模块确定与该边界位置相对应的目标像素点集的具体方式为:
对于任一所述边界通道中的任一边界位置,确定该边界通道对应的分层边界的分布方向,当确定出该边界通道对应的分层边界的分布方向是横向时,将该边界通道中该边界位置处的整列像素点确定为与该边界位置相对应的目标像素点集;当确定出该边界通道对应的分层边界的分布方向是纵向时,将该边界位置处的整行像素点确定为与该边界位置相对应的目标像素点集。
作为一种可选的实施方式,在本发明第二方面中,所述第一确定模块根据所述预测边界结果,计算所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度的具体方式为:
根据所述预测边界结果,对所述目标像素点集中所有像素点属于该边界通道的概率值执行Softmax回归操作,得到所述目标像素点集中所有像素点属于该边界通道的回归概率值;
对所述目标像素点集中所有像素点属于该边界通道的回归概率值执行概率分布统计操作,得到所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,将所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度确定为该边界位置对应的分布集中度。
作为一种可选的实施方式,在本发明第二方面中,对于任一所述边界通道中的任一边界位置,所述第二确定模块根据该边界位置对应的分布集中度,确定该边界位置的确定度信息的具体方式为:
对于任一所述边界通道中的任一边界位置,判断该边界位置的分布集中度是否大于预设阈值,当判断出该边界位置的分布集中度大于所述预设阈值时,将用于表示该边界位置为实际边界位置的标识信息确定为该边界位置的确定度信息;
当判断出该边界位置的分布集中度小于等于所述预设阈值时,将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息。
作为一种可选的实施方式,在本发明第二方面中,所述装置还包括:
第三确定模块,用于对于任一所述边界通道中的任一边界位置,当所述第二确定模块判断出该边界位置的分布集中度小于等于所述预设阈值时,根据该边界位置,从所述待检测图像中确定出局部难例样本图片;
第二处理模块,用于将所述局部难例样本图片输入至第二卷积神经网络模型,得到该边界位置对应的纠偏量;
第四确定模块,用于根据该边界位置对应的纠偏量,计算该边界位置对应的纠偏边界位置,并将所述纠偏边界位置确定为实际边界位置。
作为一种可选的实施方式,在本发明第二方面中,所述第一卷积神经网络模型包括:
深度为M的编码层以及与M层所述编码层对应的M层解码层,所述M大于等于1;
其中,相邻两层所述编码层之间通过下采样单元连接,相邻两层所述解码层之间通过上采样单元连接;
其中,每个所述编码层以及每个所述解码层均包括若干卷积单元,每个所述卷积单元中依次包括卷积层、归一化层、激活层。
作为一种可选的实施方式,在本发明第二方面中,所述装置还包括:
第三处理模块,用于对预训练数据集执行图像增强操作,得到目标训练数据集;
分割模块,用于将所述目标训练数据集输入到预置神经网络中,得到预分割结果,所述预分割结果包括图像层分类结果以及图像层间边界结果;
第五确定模块,用于计算所述预分割结果与预先确定出的标记结果之间的联合损失,所述预先确定出的标记结果包括预先标记分类结果以及预先标记层间边界结果;
训练模块,用于将所述联合损失进行反向传播,通过预设周期长度的迭代训练得到第一卷积神经网络模型,以触发所述第一确定模块执行所述将待检测图像输入至第一卷积神经网络模型,得到预测边界结果的操作。
作为一种可选的实施方式,在本发明第二方面中,所述联合损失的计算公式为:
L=λlabel_diceLlabel_dice+λlabel_ceLlabel_ce+λbd_ceLbd_ce+λbd_l1Lbd_l1;
其中,L为所述联合损失,Llabel_dice为所述图像层分类结果与所述预先标记分类结果之间的Dice损失,λlabel_dice为所述Dice损失对应的权重系数,Llabel_ce为所述图像层分类结果与所述预先标记分类结果之间的分类交叉熵损失,λlabel_ce为所述分类交叉熵损失对应的权重系数,Lbd_ce为所述图像层间边界结果与所述预先标记层间边界结果之间的边界交叉熵损失,λbd_ce为所述边界交叉熵损失对应的权重系数,Lbd_l1为所述图像层间边界结果与所述预先标记层间边界结果之间的平滑损失,λbd_l1为所述平滑损失对应的权重系数。
本发明第三方面公开了另一种基于神经网络的图像层间边界确定装置,所述装置包括:
存储有可执行程序代码的存储器;
与所述存储器耦合的处理器;
所述处理器调用所述存储器中存储的所述可执行程序代码,执行本发明第一方面公开的一种基于神经网络的图像层间边界确定方法中的部分或全部步骤。
本发明第四方面公开了一种计算机存储介质,所述计算机存储介质存储有计算机程序代码,所述计算机程序代码被调用时,用于执行本发明第一方面公开的一种基于神经网络的图像层间边界确定方法中的部分或全部步骤。
与现有技术相比,本发明具有以下有益效果:
本发明中,将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,该预测边界结果包括N个分层边界以及与分层边界对应的N个边界通道;对于任一边界通道中的边界位置,确定与该边界位置相对应的目标像素点集;计算目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度;根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,该边界位置的确定度信息用于确定该边界位置的准确度。可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,通过一种量化的方式来确定出预测边界结果的确定度,有利于提高对图像分层边界识别的准确性,提高识别效率和可靠性,便于灵活筛选不同确定度程度的边界预测水平,进而可以降低后续图像处理的复杂度和整体调试的难度,同时可以排除不确定度高的点,并有利于后续使用纠正方法纠正模型输出。
附图说明
为了更清楚地说明本发明实施例中的技术方案,下面将对实施例描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
图1是本发明实施例公开的一种基于神经网络的图像层间边界确定方法的流程示意图;
图2是本发明实施例公开的另一种基于神经网络的图像层间边界确定方法的流程示意图;
图3是本发明实施例公开的一种基于神经网络的图像层间边界确定装置的结构示意图;
图4是本发明实施例公开的另一种基于神经网络的图像层间边界确定装置的结构示意图。
图5是本发明实施例公开的又一种基于神经网络的图像层间边界确定装置的结构示意图。
具体实施方式
为了使本技术领域的人员更好地理解本发明方案,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
本发明的说明书和权利要求书及上述附图中的术语“第一”、“第二”等是用于区别不同对象,而不是用于描述特定顺序。此外,术语“包括”和“具有”以及它们任何变形,意图在于覆盖不排他的包含。例如包含了一系列步骤或单元的过程、方法、装置、产品或端没有限定于已列出的步骤或单元,而是可选地还包括没有列出的步骤或单元,或可选地还包括对于这些过程、方法、产品或端固有的其他步骤或单元。
在本文中提及“实施例”意味着,结合实施例描述的特定特征、结构或特性可以包含在本发明的至少一个实施例中。在说明书中的各个位置出现该短语并不一定均是指相同的实施例,也不是与其它实施例互斥的独立的或备选的实施例。本领域技术人员显式地和隐式地理解的是,本文所描述的实施例可以与其它实施例相结合。
本发明涉及一种基于神经网络的图像层间边界确定方法及装置,该基于神经网络的图像层间边界确定方法及装置能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息。本发明一个或多个实施例可以应用于任意需要进行分层边界识别的图像,包括可以直接输入到神经网络进行分析的截面图片(例如OCT图像),也包括经过预处理得到的特征提取图像(例如RGB图像经过卷积网络处理之后得到的单颜色通道的特征分割图),还可以包括结合图像的采样差值算法、图像超分辨率重建算法等对图像进行划分后的图像,本发明实施例不做限定。
实施例一
请参阅图1,图1是本发明实施例公开的一种基于神经网络的图像层间边界确定方法的流程示意图。其中,该方法可以应用于图像分层判定装置中,该图像分层判定装置可以是一个独立的装置,也可以集成在图片或视频处理设备中,本发明实施例不做限定。如图1所示,该基于神经网络的图像层间边界确定方法可以包括以下操作:
100、将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,该预测边界结果包括N个分层边界以及与分层边界对应的N个边界通道。
本发明实施例中,通过第一卷积神经网络模型,可以得到待检测图像的预测边界结果,该预测边界结果包括该待检测图像中所有像素属于某两层间边界的可能性。具体的,该预测边界结果包括该待检测图像的N(N大于等于1)个分层边界、与N个分层边界对应的N个边界通道以及N个边界通道分别对应的概率值集合,其中,每个边界通道对应的概率值集合包括该待检测图像中所有像素点属于每个边界通道的概率值;
101、对于任一边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集。
本发明实施例中,在得到预测边界结果之后,对于任一边界通道中的任一边界位置,首先确定出与该边界位置相对应的目标像素点集,该目标像素点集用于进行概率统计分析。目标像素点集并不限定具体的范围和数目,但选取的目标像素点集样本应当是与该边界位置相对应、相关联的,例如以该边界位置为中心的正方形、圆形区域或者其他不规则图形区域范围内的所有像素点的集合。
需要说明的是,对每一个边界位置进行目标像素点集选取时,各个边界位置对应的目标像素点集之间可以有相互重叠的像素点,也可以是完全相互独立的像素点集,本发明实施例不做限定。
102、根据预测边界结果,计算目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度。
本发明实施例中,在确定该边界位置对应的目标像素点集之后,计算该边界位置对应的目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,例如可以计算目标像素点集对应的所有概率值的峰度值K,其中K的计算公式如下:
其中,K为目标像素点集对应的所有概率值的峰度值,n为该目标像素点集中所有像素点的总数目,xi为目标像素点集中第i个像素点属于该边界通道的概率值,x为目标像素点集中所有像素点属于该边界通道的概率值的平均值。
需要说明的是,本发明实施例中,并不限定所使用的分布集中度的具体形式,除峰度值反映分布集中度外,还可以使用偏度、方差、标准差等其他评价概率分布集中度的参数,本发明实施例不作限定。
103、对于任一边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息。
本发明实施例中,通过计算分析每个通道的概率分布集中度情况,用来衡量该通道该位置处分层边界的确定度信息,该边界位置的确定度信息用于确定该边界位置的准确度。以峰度值K为例,可以通过峰度值K反映每个通道每列中概率值的分散情况,置信度分布越集中的列,K值越大,确定度越高,置信度分布越分散的列,K值越小,不确定度越高。
可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,通过一种量化的方式来确定出预测边界结果的确定度,有利于提高对图像分层边界识别的准确性,进而可以降低后续图像处理的复杂度和整体调试的难度,同时可以排除不确定度高的点,并有利于后续使用纠正方法纠正模型输出。
进一步可选的,步骤100中将待检测图像输入至第一卷积神经网络模型,得到的预测边界结果还可以包括该待检测图像中所有像素属于某个分层的可能性。具体的,该预测边界结果还可以包括该待检测图像的B(B大于等于2)个分层、B+1(通道数目等于分层数目加1层背景)个分层通道以及B+1个分层通道分别对应的概率值集合,其中每个分层通道对应的概率值集合包括该待检测图像中所有像素点属于每个分层通道的概率值。
可见,本发明不仅能够基于神经网络准确分割图像的分层并获取分层边界信息,还能够获取分层信息,进而确定识别出的分层信息的准确度信息,通过一种量化的方式来确定出分层结果的确定度,有利于提高对图像分层分割的准确性,进而可以降低后续图像处理的复杂度和整体调试的难度,同时可以排除不确定度高的点,并有利于后续使用纠正方法纠正模型输出。
在一个可选的实施例中,对于任一边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,可以选择一种更优的确定方法:确定该边界通道对应的分层边界的分布方向,当确定出该边界通道对应的分层边界的分布方向是横向时,将该边界通道中的该边界位置处的整列像素点确定为与该边界位置相对应的目标像素点集;当确定出该边界通道对应的分层边界的分布方向是纵向时,将该边界位置处的整行像素点确定为与该边界位置相对应的目标像素点集。例如,视网膜的分层基本上都是横向的层在竖直方向上的叠加,边界信息是个数为W(图像像素级宽度)的y坐标的高度值组成,所以可以将该边界位置处的一整列的所有像素点组成为目标像素点集,进而对该列像素点的分析结果作为该边界位置处的分析结果。
可见,本发明通过优化边界位置处的目标像素点集的取样方式,能够更准确的评价该边界位置处的确定度水平,进而有利于提高对图像分层边界准确度,提高识别效率和可靠性。
在一个可选的实施例中,根据预测边界结果,计算目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度,可以包括:
根据预测边界结果,对目标像素点集中所有像素点属于该边界通道的概率值执行Softmax回归操作,得到目标像素点集中所有像素点属于该边界通道的回归概率值;
对目标像素点集中所有像素点属于该边界通道的回归概率值执行概率分布统计操作,得到目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,将目标像素点集中所有像素点属于该边界通道的概率值的分布集中度确定为该边界位置对应的分布集中度。
本发明实施例中,通过Softmax算法,对目标像素点集中所有像素点属于该边界通道个概率值进行归一化的逻辑回归,将所有概率值转化为回归概率值,即将所有概率值映射到0-1区间的回归概率值,且所有回归概率值的和为1,进而对回归概率值进行概率统计分析。需要说明的是,本发明实施例中还可以使用其他归一化准则,例如logistic回归算法,本发明不做限定,但应当说明的是Softmax是本方案的优选方案。
可见,本发明通过归一化的转化方式,可以最小化预测边界与实际边界之间的损失,进行最大似然估计,提高概率分布统计样本的可靠性,可以更加准确地实现对边界位置的识别,提高识别和分割边界的可靠性。
在该可选的实施例中,进一步可选的,在得到每一通道每一分界位置的Softmax结果之后,可以将该Softmax结果与像素的对应位置编号进行点积运算,最终可以获得该分界在该位置出的点积结果,也即层间边界的分界信息。例如,以视网膜图片分析为例,对每一通道每一列求Softmax得到结果B1,再与该列每行编号点积运算后,获得该分界在该列的位置得到B2,B2即为视网膜组织层间分界信息。
在一个可选的实施例中,对于任一边界通道中的任一边界位置,可以通过预设阈值判断的方式,判断该边界位置的分布集中度是否大于预设阈值,当判断出该边界位置的分布集中度大于该预设阈值时,将用于表示该边界位置为实际位置的标识信息确定为该边界位置的确定度信息;当判断出该边界位置的分布集中度小于等于该预设阈值时,将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息。例如在视网膜OCT组织层识别过程中,以峰度值K作为衡量分布集中度的方式,当K值小于等于70的时,反映相应位置处的概率分布越分散,不确定度越高,也越容易出现预测错误的情况,当K值大于70时,说明相应位置处的概率分布比较集中,不确定度较低,本发明得到的预测边界位置的为实际边界位置的置信度较高。需要说明的是,本发明中提到的阈值70只是作为示例,实际方案中可以根据实际情况需要,动态调整对应的阈值范围和水平,以适应不同图像的分割和判断需要。同样,本发明并不限定每个通道使用同一个阈值水平,可以根据实际情况调整不同通道的预设阈值,比如对于其中个别不重要的分层边界不需要很高的确定度水平,可以适当降低对应分层边界对应的预设阈值。
可见,本发明通过预设阈值的方式,可以根据实际需要,动态调整控制图像识别过程中的确定度水平,便于灵活筛选不同确定度程度的边界预测水平,降低后续纠正模型的处理数据量,提高纠正模型的处理速率。
在一个可选的实施例中,本发明实施例中第一卷积神经网络模型可以包括:多个编码层和与编码成相对应的解码层,每个编码层或者解码层由若干个卷积单元组成,每个卷积单元中依次包括卷积层(卷积核大小可以是3x3、5x5、7x3,本发明实施例不做限定)、归一化层(批归一化层、实例归一化层或无,本发明实施例不作限定)、激活层(ReLU或其他变种,如PReLU、LeakyReLU,本发明实施例不做限定)。其中同级的编码层与解码层之间可以带有短连接,短连接有利于梯度传播,使得神经网络更容易训练。
需要说明的是,本发明实施例中,两个或多个卷积单元中也可以带有短连接。相邻两层编码层之间可以通过下采样单元连接,下采样单元可以为步幅为预设值的卷积或者最大池化层或平均池化层,相邻两层解码层之间可以通过上采样单元连接,上采样单元可以为步幅为预设值的反卷积或上采样层。
可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,通过一种量化的方式来确定出预测边界结果的确定度,有利于提高对图像分层边界识别的准确性,提高识别效率和可靠性,便于灵活筛选不同确定度程度的边界预测水平,进而可以降低后续图像处理的复杂度和整体调试的难度,同时可以排除不确定度高的点,并有利于后续使用纠正方法纠正模型输出。
实施例二
请参阅图2,图2是本发明实施例公开的另一种基于神经网络的图像层间边界确定方法的流程示意图。其中,该方法可以应用于图像分层判定装置中,该图像分层判定装置可以是一个独立的装置,也可以集成在图片或视频处理设备中,本发明实施例不做限定。如图2所示,该基于神经网络的图像层间边界确定方法可以包括以下操作:
201、将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,该预测边界结果包括N个分层边界以及与分层边界对应的N个边界通道。
202、对于任一边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集。
203、根据预测边界结果,计算目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度。
204、对于任一边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息。
本发明实施例中,针对步骤201-步骤204的其它描述,请分别对应参照实施例一中针对步骤100-步骤103的详细描述,本发明实施例不再赘述。
可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,通过一种量化的方式来确定出预测边界结果的确定度,有利于提高对图像分层边界识别的准确性,进而可以降低后续图像处理的复杂度和整体调试的难度,同时可以排除不确定度高的点,并有利于后续使用纠正方法纠正模型输出。
在一个可选的实施例中,在步骤203将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息之后,该方法还可以包括:
205、根据该边界位置,从待检测图像中确定出局部难例样本图片,对该局部难例样本图片进行纠正。
本发明实施例中,通过找出K值小于某固定值的点,可以找出第一神经网络模型预测不确定度高的点,可以通过插值纠正不确定的预测点。其中一些插值的方法如:a.通过相邻附近点信息插值纠正;b.同一分界前后帧插值纠正;c.通过对整个对齐后的3D(所有b-scan)图像边界进行预测,对不确定度高的点使用2维插值进行纠正。上述所指明的方法只是作为示例,实际可以根据实际情况进行选择,本发明实施例不做限定。
可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而在确定识别出的边界信息的准确度信息之后,对于不确定度高的点,通过纠正算法对不确定点进行纠正操作,进而可以使得分层结果更加准确。
在该可选的实施例中,进一步可选的,本发明实施例也可以通过筛选出不确定区域,专门训练第二神经网络模型,同时接受组织层边界输入和相应位置的峰度信息,预测边界的偏移量,通过比对标注信息得到损失从而训练得到纠正神经网络,也即第二神经网络。例如,详细的纠正过程可以如下:a、通过第一神经网络模型中训练得到的预测边界结果,结合Kurtosis或其它度量,筛选出不确定度高的区域,如H*32的大小(H为图片的高度,宽度可任意,一般应大于不确定度高的区域的宽度),作为新的训练样本。这些样本代表了对于第一神经网络来说的难以确定的难例样本;b、建立纠偏网络,也即第二神经网络,经过若干个卷积单元及池化层,最后通过若干全连接层,输出长度为k*W的纠偏向量(k条分层线,W为图像的宽度)。第二神经网络的输入为经过第一神经网络筛选的局部难例样本图片(置信度低的图片局部)、第一神经网络预测的分层位置及置信度信息,第二神经网络的输出为对应不同层边界的修正量,例如第一神经网络预测边界的位置为126,而真正的位置为132,则第二神经网络通过学习,预测出纠偏量为5.5(真实纠偏量为6),通过两个网络结合,可以使得分层结果更加准确。
可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而在确定识别出的边界信息的准确度信息之后,通过纠偏神经网络模型对第一神经网络模型的输出结果进行纠正,可以得到精确的纠偏量,进而通过两个神经网络模型的结合,使得分层结果更加准确。
在一个可选的实施例中,在将待检测图像输入至第一卷积神经网络模型之前,该方法还包括:
200、通过训练数据集对预设神经网络模型进行训练,得到第一卷积神经网络模型。
本发明实施例中,对预设神经网络模型进行训练的方法,可以包括如下操作:
对预训练数据集执行图像增强操作,得到目标训练数据集。
本发明实施例中,对预训练数据集执行图像增强操作,其中图像增强操作用于增加模型健壮性,图像增强操作包括上下移动、左右翻转、上下翻转、对比度变化、亮度变化等,本发明实施例不做限定,进而得到目标训练数据集。
将目标训练数据集输入到预置神经网络中,得到预分割结果,预分割结果包括图像层分类结果以及图像层间边界结果。
本发明实施例中,将目标训练数据集输入到阈值神经网络中,进行前向传播,以得到预分割结果。其中,预分割结果可以包括图像层分类结果以及图像层间边界结果。
计算预分割结果与预先确定出的标记结果之间的联合损失,预先确定出的标记结果包括预先标记分类结果以及预先标记层间边界结果。
本发明实施例中,在对预置神经网络进行训练的过程中,由于需要不断训练,因此本发明实施例采用了预分割结果与预先确定出的标记结果之间的损失函数配合作用来对预置神经网络进行不断训练。其中,预先确定出的标记结果包括预先标记分类结果以及预先标记层间边界结果。例如,专家标注的像素级图像或者其他自动判定方式确定出来的图像。
将联合损失进行反向传播,通过预设周期长度的迭代训练得到第一卷积神经网络模型。
本发明实施例中,将联合损失值进行反向传播,并通过预设周期长度的迭代训练得到第一卷积神经网络。其中,联合损失用来表示预分割预测结果与其相对应的经过one-hot编码后的专家像素级标记图像之间的误差。通过联合损失函数公式计算预分割预测图与经过one-hot编码后的专家像素级标记图像之间的损失值进行反向传播,以便于对权重参数进行更新,通过预设周期长度的迭代训练,以最小化损失函数值,从而得到训练好的第一卷积神经网络用于测试。
其中,联合损失的计算公式为:
L=λlabel_diceLlabel_dice+λlabel_ceLlabel_ce+λbd_ceLbd_ce+λbd_l1Lbd_l1;
其中,L为联合损失,Llabel_dice为图像层分类结果与预先标记分类结果之间的Dice损失,λlabel_dice为Dice损失对应的权重系数,Llabel_ce为图像层分类结果与预先标记分类结果之间的分类交叉熵损失,λlabel_ce为分类交叉熵损失对应的权重系数,Lbd_ce为图像层间边界结果与预先标记层间边界结果之间的边界交叉熵损失,λbd_ce为边界交叉熵损失对应的权重系数,Lbd_l1为图像层间边界结果与预先标记层间边界结果之间的平滑损失,λbd_l1为平滑损失对应的权重系数。
可见,本发明能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,可以提高训练的可靠性,得到最优的第一神经网络模型,进而提高神经网络模型应用的广泛性,满足实际应用需求。
实施例三
请参阅图3,图3是本发明实施例公开的一种基于神经网络的图像层间边界确定装置的结构示意图。需要说明的是,该动静判定装置参照的是实施例一和实施例二所描述的基于神经网络的图像层间边界确定方法中的步骤,详细的描述在本实施例中就不做赘述,如图3所示,该基于神经网络的图像层间边界确定装置可以包括:
第一处理模块301,用于将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,该预测边界结果包括该待检测图像的N个分层边界、与N个分层边界对应的N个边界通道以及N个边界通道分别对应的概率值集合,其中,N大于等于1,每个边界通道对应的概率值集合包括该待检测图像中所有像素点属于每个边界通道的概率值;
第一确定模块302,用于对于任一边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,根据预测边界结果,计算该目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度;
第二确定模块303,用于对于任一边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,该边界位置的确定度信息用于确定该边界位置的准确度。
可见,本发明所描述的基于神经网络的图像层间边界确定装置,能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,通过一种量化的方式来确定出预测边界结果的确定度,有利于提高对图像分层边界识别的准确性,进而可以降低后续图像处理的复杂度和整体调试的难度,同时可以排除不确定度高的点,并有利于后续使用纠正方法纠正模型输出。
在一个可选的实施例中,对于任一边界通道中的任一边界位置,第一确定模块302确定与该边界位置相对应的目标像素点集的具体方式为:
对于任一边界通道中的任一边界位置,确定该边界通道对应的分层边界的分布方向,当确定出该边界通道对应的分层边界的分布方向是横向时,将该边界通道中该边界位置处的整列像素点确定为与该边界位置相对应的目标像素点集;当确定出该边界通道对应的分层边界的分布方向是纵向时,将该边界位置处的整行像素点确定为与该边界位置相对应的目标像素点集。
可见,本发明所描述的基于神经网络的图像层间边界确定装置,能够更准确的评价该边界位置处的确定度水平,进而有利于提高对图像分层边界准确度,提高识别效率和可靠性。
在一个可选的实施例中,第一确定模块302根据预测边界结果,计算目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度的具体方式为:
根据预测边界结果,对目标像素点集中所有像素点属于该边界通道的概率值执行Softmax回归操作,得到目标像素点集中所有像素点属于该边界通道的回归概率值;
对目标像素点集中所有像素点属于该边界通道的回归概率值执行概率分布统计操作,得到目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,将目标像素点集中所有像素点属于该边界通道的概率值的分布集中度确定为该边界位置对应的分布集中度。
可见,本发明所描述的基于神经网络的图像层间边界确定装置,能够通过归一化的转化方式,可以最小化预测边界与实际边界之间的损失,进行最大似然估计,提高概率分布统计样本的可靠性,可以更加准确地实现对边界位置的识别,提高识别和分割边界的可靠性。
在一个可选的实施例中,对于任一边界通道中的任一边界位置,第二确定模块303根据该边界位置对应的分布集中度,确定该边界位置的确定度信息的具体方式为:
对于任一边界通道中的任一边界位置,判断该边界位置的分布集中度是否大于预设阈值,当判断出该边界位置的分布集中度大于预设阈值时,将用于表示该边界位置为实际边界位置的标识信息确定为该边界位置的确定度信息;
当判断出该边界位置的分布集中度小于等于预设阈值时,将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息。
可见,本发明所描述的基于神经网络的图像层间边界确定装置,能够通过预设阈值的方式,可以根据实际需要,动态调整控制图像识别过程中的确定度水平,便于灵活筛选不同确定度程度的边界预测水平,降低后续纠正模型的处理数据量,提高纠正模型的处理速率。
在一个可选的实施例中,如图4所示,该基于神经网络的图像层间边界确定装置还可以包括:
第三确定模块304,用于对于任一边界通道中的任一边界位置,当第二确定模块303判断出该边界位置的分布集中度小于等于预设阈值时,根据该边界位置,从待检测图像中确定出局部难例样本图片;
第二处理模块305,用于将局部难例样本图片输入至第二卷积神经网络模型,得到该边界位置对应的纠偏量;
第四确定模块306,用于根据该边界位置对应的纠偏量,计算该边界位置对应的纠偏边界位置,并将纠偏边界位置确定为实际边界位置。
可见,本发明所描述的基于神经网络的图像层间边界确定装置,能够基于神经网络准确分割图像的分层并获取分层边界信息,进而在确定识别出的边界信息的准确度信息之后,通过纠偏神经网络模型对第一神经网络模型的输出结果进行纠正,可以得到精确的纠偏量,进而通过两个神经网络模型的结合,使得分层结果更加准确。
在一个可选的实施例中,该基于神经网络的图像层间边界确定装置中第一卷积神经网络模型包括:
深度为M的编码层以及与M层编码层对应的M层解码层,M大于等于1;
其中,相邻两层编码层之间通过下采样单元连接,相邻两层解码层之间通过上采样单元连接;
其中,每个编码层以及每个解码层均包括若干卷积单元,每个卷积单元中依次包括卷积层、归一化层、激活层。
在一个可选的实施例中,该基于神经网络的图像层间边界确定装置还可以包括:
第三处理模块307,用于对预训练数据集执行图像增强操作,得到目标训练数据集;
分割模块308,用于将目标训练数据集输入到预置神经网络中,得到预分割结果,预分割结果包括图像层分类结果以及图像层间边界结果;
第五确定模块309,用于计算预分割结果与预先确定出的标记结果之间的联合损失,预先确定出的标记结果包括预先标记分类结果以及预先标记层间边界结果;
训练模块310,用于将联合损失进行反向传播,通过预设周期长度的迭代训练得到第一卷积神经网络模型,以触发第一确定模块执行将待检测图像输入至第一卷积神经网络模型,得到预测边界结果的操作,继而触发执行第一处理模块301执行将待检测图像输入至第一卷积神经网络模型,得到预测边界结果的操作。
可见,本发明所描述的基于神经网络的图像层间边界确定装置,能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,可以提高训练的可靠性,得到最优的第一神经网络模型,进而提高神经网络模型应用的广泛性,满足实际应用需求。
在一个可选的实施例中,该基于神经网络的图像层间边界确定装置中的联合损失的计算公式为:
L=λlabel_diceLlabel_dice+λlabel_ceLlabel_ce+λbd_ceLbd_ce+λbd_l1Lbd_l1;
其中,L为联合损失,Llabel_dice为图像层分类结果与预先标记分类结果之间的Dice损失,λlabel_dice为Dice损失对应的权重系数,Llabel_ce为图像层分类结果与预先标记分类结果之间的分类交叉熵损失,λlabel_ce为分类交叉熵损失对应的权重系数,Lbd_ce为图像层间边界结果与预先标记层间边界结果之间的边界交叉熵损失,λbd_ce为边界交叉熵损失对应的权重系数,Lbd_l1为图像层间边界结果与预先标记层间边界结果之间的平滑损失,λbd_l1为所述平滑损失对应的权重系数。
可见,本发明所描述的基于神经网络的图像层间边界确定装置,能够基于神经网络准确分割图像的分层并获取分层边界信息,进而确定识别出的边界信息的准确度信息,通过一种量化的方式来确定出预测边界结果的确定度,有利于提高对图像分层边界识别的准确性,提高识别效率和可靠性,便于灵活筛选不同确定度程度的边界预测水平,进而可以降低后续图像处理的复杂度和整体调试的难度,同时可以排除不确定度高的点,并有利于后续使用纠正方法纠正模型输出。
实施例四
请参阅图5,图5是本发明实施例公开的另一种基于神经网络的图像层间边界确定装置的结构示意图。如图5所示,该基于神经网络的图像层间边界确定装置可以包括:
存储有可执行程序代码的存储器401;
与存储器401耦合的处理器402;
处理器402调用存储器401中存储的可执行程序代码,用于执行实施例一或实施例二所描述的基于神经网络的图像层间边界确定方法中的步骤。
实施例五
本发明实施例公开了一种计算机读存储介质,其存储用于电子数据交换的计算机程序,其中,该计算机程序使得计算机执行实施例一或实施例二所描述的基于神经网络的图像层间边界确定方法中的步骤。
实施例六
本发明实施例公开了一种计算机程序产品,该计算机程序产品包括存储了计算机程序的非瞬时性计算机可读存储介质,且该计算机程序可操作来使计算机执行实施例一或实施例二所描述的基于神经网络的图像层间边界确定方法中的步骤。
以上所描述的装置实施例仅是示意性的,其中所述作为分离部件说明的模块可以是或者也可以不是物理上分开的,作为模块显示的部件可以是或者也可以不是物理模块,即可以位于一个地方,或者也可以分布到多个网络模块上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。本领域普通技术人员在不付出创造性的劳动的情况下,即可以理解并实施。
通过以上的实施例的具体描述,本领域的技术人员可以清楚地了解到各实施方式可借助软件加必需的通用硬件平台的方式来实现,当然也可以通过硬件。基于这样的理解,上述技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品可以存储在计算机可读存储介质中,存储介质包括只读存储器(Read-Only Memory,ROM)、随机存储器(Random Access Memory,RAM)、可编程只读存储器(Programmable Read-only Memory,PROM)、可擦除可编程只读存储器(ErasableProgrammable Read Only Memory,EPROM)、一次可编程只读存储器(One-timeProgrammable Read-Only Memory,OTPROM)、电子抹除式可复写只读存储器(Electrically-Erasable Programmable Read-Only Memory,EEPROM)、只读光盘(CompactDisc Read-Only Memory,CD-ROM)或其他光盘存储器、磁盘存储器、磁带存储器、或者能够用于携带或存储数据的计算机可读的任何其他介质。
需要说明的是本说明书各部分操作所需的计算机程序代码可以用任意一种或多种程序语言编写,包括面向对象编程语言如Java、Scala、Smalltalk、Eiffel、JADE、Emerald、C++、C#、VB.NET、Python等,常规程序化编程语言如C语言、Visual Basic、Fortran2003、Perl、COBOL 2002、PHP、ABAP,动态编程语言如Python、Ruby和Groovy,或其他编程语言等。该程序编码可以完全在计算机(PC、嵌入式智能设备等)上运行、或作为独立的软件包在用户计算机上运行、或部分在用户计算机上运行部分在远程计算机运行、或完全在远程计算机或服务器上运行。在后种情况下,远程计算机可以通过任何网络形式与用户计算机连接,比如局域网(LAN)或广域网(WAN),或连接至外部计算机(例如通过因特网),或在云计算环境中,或作为服务使用如软件即服务(SaaS)。
最后应说明的是:本发明实施例公开的一种基于神经网络的图像层间边界确定方法及装置所揭露的仅为本发明较佳实施例而已,仅用于说明本发明的技术方案,而非对其限制;尽管参照前述实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解;其依然可以对前述各项实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或替换,并不使相应的技术方案的本质脱离本发明各项实施例技术方案的精神和范围。
Claims (8)
1.一种基于神经网络的图像层间边界确定方法,其特征在于,所述方法包括:
将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,所述预测边界结果包括所述待检测图像的N个分层边界、与N个所述分层边界对应的N个边界通道以及N个所述边界通道分别对应的概率值集合,其中,所述N大于等于1,每个所述边界通道对应的概率值集合包括所述待检测图像中所有像素点属于每个所述边界通道的概率值;
对于任一所述边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,根据所述预测边界结果,计算所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度;
对于任一所述边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,该边界位置的确定度信息用于确定该边界位置的准确度;
以及,在所述将待检测图像输入至第一卷积神经网络模型之前,所述方法还包括:
对预训练数据集执行图像增强操作,得到目标训练数据集;
将所述目标训练数据集输入到预置神经网络中,得到预分割结果,所述预分割结果包括图像层分类结果以及图像层间边界结果;
计算所述预分割结果与预先确定出的标记结果之间的联合损失,所述预先确定出的标记结果包括预先标记分类结果以及预先标记层间边界结果;
将所述联合损失进行反向传播,通过预设周期长度的迭代训练得到第一卷积神经网络模型;
其中,所述联合损失的计算公式为:
L=λlabel_diceLlabel_dice+λlabel_ceLlabel_ce+λbd_ceLbd_ce+λbd_l1Lbd_l1;
其中,L为所述联合损失,Llabel_dice为所述图像层分类结果与所述预先标记分类结果之间的Dice损失,λlabel_dice为所述Dice损失对应的权重系数,Llabel_ce为所述图像层分类结果与所述预先标记分类结果之间的分类交叉熵损失,λlabel_ce为所述分类交叉熵损失对应的权重系数,Lbd_ce为所述图像层间边界结果与所述预先标记层间边界结果之间的边界交叉熵损失,λbd_ce为所述边界交叉熵损失对应的权重系数,Lbd_l1为所述图像层间边界结果与所述预先标记层间边界结果之间的平滑损失,λbd_l1为所述平滑损失对应的权重系数。
2.根据权利要求1所述的基于神经网络的图像层间边界确定方法,其特征在于,所述对于任一所述边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,包括:
对于任一所述边界通道中的任一边界位置,确定该边界通道对应的分层边界的分布方向,当确定出该边界通道对应的分层边界的分布方向是横向时,将该边界通道中该边界位置处的整列像素点确定为与该边界位置相对应的目标像素点集;当确定出该边界通道对应的分层边界的分布方向是纵向时,将该边界位置处的整行像素点确定为与该边界位置相对应的目标像素点集。
3.根据权利要求2所述的基于神经网络的图像层间边界确定方法,其特征在于,所述根据所述预测边界结果,计算所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度,包括:
根据所述预测边界结果,对所述目标像素点集中所有像素点属于该边界通道的概率值执行Softmax回归操作,得到所述目标像素点集中所有像素点属于该边界通道的回归概率值;
对所述目标像素点集中所有像素点属于该边界通道的回归概率值执行概率分布统计操作,得到所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,将所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度确定为该边界位置对应的分布集中度。
4.根据权利要求3所述的基于神经网络的图像层间边界确定方法,其特征在于,所述对于任一所述边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,包括:
对于任一所述边界通道中的任一边界位置,判断该边界位置的分布集中度是否大于预设阈值,当判断出该边界位置的分布集中度大于所述预设阈值时,将用于表示该边界位置为实际边界位置的标识信息确定为该边界位置的确定度信息;
当判断出该边界位置的分布集中度小于等于所述预设阈值时,将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息。
5.根据权利要求4所述的基于神经网络的图像层间边界确定方法,其特征在于,对于任一所述边界通道中的任一边界位置,所述将用于表示该边界位置为非实际边界位置的标识信息确定为该边界位置的确定度信息之后,所述方法还包括:
根据该边界位置,从所述待检测图像中确定出局部难例样本图片;
将所述局部难例样本图片输入至第二卷积神经网络模型,得到该边界位置对应的纠偏量;
根据该边界位置对应的纠偏量,计算该边界位置对应的纠偏边界位置,并将所述纠偏边界位置确定为实际边界位置。
6.根据权利要求1所述的基于神经网络的图像层间边界确定方法,其特征在于,所述第一卷积神经网络模型包括:
深度为M的编码层以及与M层所述编码层对应的M层解码层,所述M大于等于1;
其中,相邻两层所述编码层之间通过下采样单元连接,相邻两层所述解码层之间通过上采样单元连接;
其中,每个所述编码层以及每个所述解码层均包括若干卷积单元,每个所述卷积单元中依次包括卷积层、归一化层、激活层。
7.一种基于神经网络的图像层间边界确定装置,其特征在于,所述装置包括:
第一处理模块,用于将待检测图像输入至第一卷积神经网络模型,得到预测边界结果,所述预测边界结果包括所述待检测图像的N个分层边界、与N个所述分层边界对应的N个边界通道以及N个所述边界通道分别对应的概率值集合,其中,所述N大于等于1,每个所述边界通道对应的概率值集合包括所述待检测图像中所有像素点属于每个所述边界通道的概率值;
第一确定模块,用于对于任一所述边界通道中的任一边界位置,确定与该边界位置相对应的目标像素点集,根据所述预测边界结果,计算所述目标像素点集中所有像素点属于该边界通道的概率值的分布集中度,得到该边界位置对应的分布集中度;
第二确定模块,用于对于任一所述边界通道中的任一边界位置,根据该边界位置对应的分布集中度,确定该边界位置的确定度信息,该边界位置的确定度信息用于确定该边界位置的准确度;
以及,所述装置还包括:
第三处理模块,用于对预训练数据集执行图像增强操作,得到目标训练数据集;
分割模块,用于将所述目标训练数据集输入到预置神经网络中,得到预分割结果,所述预分割结果包括图像层分类结果以及图像层间边界结果;
第五确定模块,用于计算所述预分割结果与预先确定出的标记结果之间的联合损失,所述预先确定出的标记结果包括预先标记分类结果以及预先标记层间边界结果;
训练模块,用于将所述联合损失进行反向传播,通过预设周期长度的迭代训练得到第一卷积神经网络模型,以触发所述第一确定模块执行所述将待检测图像输入至第一卷积神经网络模型,得到预测边界结果的操作;
其中,所述联合损失的计算公式为:
L=λlabel_diceLlabel_dice+λlabel_ceLlabel_ce+λbd_ceLbd_ce+λbd_l1Lbd_l1;
其中,L为所述联合损失,Llabel_dice为所述图像层分类结果与所述预先标记分类结果之间的Dice损失,λlabel_dice为所述Dice损失对应的权重系数,Llabel_ce为所述图像层分类结果与所述预先标记分类结果之间的分类交叉熵损失,λlabel_ce为所述分类交叉熵损失对应的权重系数,Lbd_ce为所述图像层间边界结果与所述预先标记层间边界结果之间的边界交叉熵损失,λbd_ce为所述边界交叉熵损失对应的权重系数,Lbd_l1为所述图像层间边界结果与所述预先标记层间边界结果之间的平滑损失,λbd_l1为所述平滑损失对应的权重系数。
8.一种基于神经网络的图像层间边界确定装置,其特征在于,所述装置包括:
存储有可执行程序代码的存储器;
与所述存储器耦合的处理器;
所述处理器调用所述存储器中存储的所述可执行程序代码,执行如权利要求1-6任一项所述的基于神经网络的图像层间边界确定方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110666798.3A CN113538470B (zh) | 2021-06-16 | 2021-06-16 | 一种基于神经网络的图像层间边界确定方法及装置 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110666798.3A CN113538470B (zh) | 2021-06-16 | 2021-06-16 | 一种基于神经网络的图像层间边界确定方法及装置 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN113538470A CN113538470A (zh) | 2021-10-22 |
CN113538470B true CN113538470B (zh) | 2024-02-23 |
Family
ID=78125011
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110666798.3A Active CN113538470B (zh) | 2021-06-16 | 2021-06-16 | 一种基于神经网络的图像层间边界确定方法及装置 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113538470B (zh) |
Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109003267A (zh) * | 2017-08-09 | 2018-12-14 | 深圳科亚医疗科技有限公司 | 从3d图像自动检测目标对象的计算机实现方法和系统 |
CN111583291A (zh) * | 2020-04-20 | 2020-08-25 | 中山大学 | 基于深度学习的视网膜层和积液区域的层分割方法及系统 |
CN111784721A (zh) * | 2020-07-01 | 2020-10-16 | 华南师范大学 | 基于深度学习的超声内窥图像智能分割与量化方法及系统 |
CN112465847A (zh) * | 2020-11-27 | 2021-03-09 | 深圳点猫科技有限公司 | 一种基于预测清晰边界的边缘检测方法、装置及设备 |
Family Cites Families (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200327674A1 (en) * | 2019-04-10 | 2020-10-15 | Nvidia Corporation | Identifying Image Segmentation Quality Using Neural Networks |
-
2021
- 2021-06-16 CN CN202110666798.3A patent/CN113538470B/zh active Active
Patent Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109003267A (zh) * | 2017-08-09 | 2018-12-14 | 深圳科亚医疗科技有限公司 | 从3d图像自动检测目标对象的计算机实现方法和系统 |
CN111583291A (zh) * | 2020-04-20 | 2020-08-25 | 中山大学 | 基于深度学习的视网膜层和积液区域的层分割方法及系统 |
CN111784721A (zh) * | 2020-07-01 | 2020-10-16 | 华南师范大学 | 基于深度学习的超声内窥图像智能分割与量化方法及系统 |
CN112465847A (zh) * | 2020-11-27 | 2021-03-09 | 深圳点猫科技有限公司 | 一种基于预测清晰边界的边缘检测方法、装置及设备 |
Non-Patent Citations (1)
Title |
---|
基于双通道卷积神经网络的视频目标移除取证算法;白珊山等;《信号处理》;第36卷(第9期);第1415-1421页 * |
Also Published As
Publication number | Publication date |
---|---|
CN113538470A (zh) | 2021-10-22 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
US11776119B2 (en) | Confidence-based method and system for analyzing images of a retina | |
US11227683B2 (en) | Methods and systems for characterizing anatomical features in medical images | |
US20210142477A1 (en) | Bone Age Assessment And Height Prediction Model, System Thereof And Prediction Method Thereof | |
CN111862044B (zh) | 超声图像处理方法、装置、计算机设备和存储介质 | |
CN109949280B (zh) | 图像处理方法、装置、设备存储介质及生长发育评估系统 | |
KR102250954B1 (ko) | 뇌 mri 영상의 뇌 영역별 분할을 통한 치매 예측 장치 및 방법 | |
CN111179372A (zh) | 图像衰减校正方法、装置、计算机设备和存储介质 | |
CN113689342A (zh) | 一种图像质量优化的方法及系统 | |
CN110858399B (zh) | 用于提供虚拟断层扫描中风后续检查图像的方法和装置 | |
CN111612756B (zh) | 冠状动脉特异性钙化检测方法及装置 | |
CN110717894B (zh) | 癌症靶向治疗疗效评估方法及装置、设备、存储介质 | |
CN110956628B (zh) | 图片等级分类方法、装置、计算机设备和存储介质 | |
Sambyal et al. | Towards reducing aleatoric uncertainty for medical imaging tasks | |
CN113724203B (zh) | 应用于oct图像中目标特征分割的模型训练方法及装置 | |
CN115187566A (zh) | 基于mra影像的颅内动脉瘤检测方法及装置 | |
CN113298800B (zh) | 一种ct血管造影cta源图像的处理方法、装置及设备 | |
CN113538470B (zh) | 一种基于神经网络的图像层间边界确定方法及装置 | |
CN112652031A (zh) | 伪影校正分析方法、电子设备及存储介质 | |
TWI663960B (zh) | 足畸形檢測模型、足畸形檢測系統及足畸形檢測方法 | |
KR20220001177A (ko) | 심층신경망을 이용하여 초음파 영상으로부터 골격근의 상태를 진단하기 위한 장치 및 이를 위한 방법 | |
CN113222985B (zh) | 一种图像处理方法、装置、计算机设备和介质 | |
CN111445447A (zh) | 一种ct图像异常检测方法及装置 | |
Naeemi et al. | Application of big data analytics for automated estimation of CT image quality | |
KR102421765B1 (ko) | 인공지능을 이용한 비조영 컴퓨터 단층촬영 이미지를 바탕으로 폐색전증을 판단하는 시스템의 제어 방법, 장치 및 프로그램 | |
RU2813480C1 (ru) | Способ обработки изображений магнитно-резонансной томографии для формирования обучающих данных |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |