CN113139549B - 一种基于多任务学习的参数自适应全景分割方法 - Google Patents
一种基于多任务学习的参数自适应全景分割方法 Download PDFInfo
- Publication number
- CN113139549B CN113139549B CN202110318993.7A CN202110318993A CN113139549B CN 113139549 B CN113139549 B CN 113139549B CN 202110318993 A CN202110318993 A CN 202110318993A CN 113139549 B CN113139549 B CN 113139549B
- Authority
- CN
- China
- Prior art keywords
- segmentation
- information
- branch
- loss function
- 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
- 230000011218 segmentation Effects 0.000 title claims abstract description 106
- 238000000034 method Methods 0.000 title claims abstract description 38
- 230000006870 function Effects 0.000 claims abstract description 74
- 238000012549 training Methods 0.000 claims abstract description 43
- 238000001514 detection method Methods 0.000 claims abstract description 20
- 238000013461 design Methods 0.000 claims abstract description 8
- 230000002776 aggregation Effects 0.000 claims abstract description 7
- 238000004220 aggregation Methods 0.000 claims abstract description 7
- 230000008859 change Effects 0.000 claims abstract description 4
- 230000003044 adaptive effect Effects 0.000 claims description 19
- 238000000605 extraction Methods 0.000 claims description 11
- 230000004927 fusion Effects 0.000 claims description 10
- 230000008569 process Effects 0.000 claims description 7
- 238000011156 evaluation Methods 0.000 claims description 6
- 230000002708 enhancing effect Effects 0.000 claims description 4
- 238000006116 polymerization reaction Methods 0.000 claims description 3
- 238000000638 solvent extraction Methods 0.000 claims description 3
- 230000005764 inhibitory process Effects 0.000 claims description 2
- 238000005070 sampling Methods 0.000 claims description 2
- 238000005728 strengthening Methods 0.000 claims description 2
- 239000013589 supplement Substances 0.000 claims description 2
- 238000013528 artificial neural network Methods 0.000 abstract description 2
- 238000010586 diagram Methods 0.000 description 12
- 238000012545 processing Methods 0.000 description 10
- 238000002474 experimental method Methods 0.000 description 6
- 230000001965 increasing effect Effects 0.000 description 3
- 238000004891 communication Methods 0.000 description 2
- 238000012986 modification Methods 0.000 description 2
- 230000004048 modification Effects 0.000 description 2
- 238000011160 research Methods 0.000 description 2
- 230000004044 response Effects 0.000 description 2
- 238000006467 substitution reaction Methods 0.000 description 2
- NAWXUBYGYWOOIX-SFHVURJKSA-N (2s)-2-[[4-[2-(2,4-diaminoquinazolin-6-yl)ethyl]benzoyl]amino]-4-methylidenepentanedioic acid Chemical compound C1=CC2=NC(N)=NC(N)=C2C=C1CCC1=CC=C(C(=O)N[C@@H](CC(=C)C(O)=O)C(O)=O)C=C1 NAWXUBYGYWOOIX-SFHVURJKSA-N 0.000 description 1
- 239000003086 colorant Substances 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 238000003709 image segmentation Methods 0.000 description 1
- 238000010801 machine learning Methods 0.000 description 1
- 230000007246 mechanism Effects 0.000 description 1
- 238000010606 normalization Methods 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/20—Image preprocessing
- G06V10/26—Segmentation of patterns in the image field; Cutting or merging of image elements to establish the pattern region, e.g. clustering-based techniques; Detection of occlusion
- G06V10/267—Segmentation of patterns in the image field; Cutting or merging of image elements to establish the pattern region, e.g. clustering-based techniques; Detection of occlusion by performing operations on regions, e.g. growing, shrinking or watersheds
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
- G06F18/2415—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches based on parametric or probabilistic models, e.g. based on likelihood ratio or false acceptance rate versus a false rejection rate
- G06F18/24155—Bayesian classification
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
-
- 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
-
- 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
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02T—CLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
- Y02T10/00—Road transport of goods or passengers
- Y02T10/10—Internal combustion engine [ICE] based vehicles
- Y02T10/40—Engine management systems
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- General Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Molecular Biology (AREA)
- Biomedical Technology (AREA)
- Health & Medical Sciences (AREA)
- Evolutionary Biology (AREA)
- General Health & Medical Sciences (AREA)
- Biophysics (AREA)
- Computing Systems (AREA)
- Computational Linguistics (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Probability & Statistics with Applications (AREA)
- Multimedia (AREA)
- Image Analysis (AREA)
Abstract
本发明涉及一种基于多任务学习的参数自适应调节的全景分割方法,包括:可提取图像前景对象信息的基于单阶段检测网络的实例分割分支网络设计;可提取图像背景语义信息的基于全卷积神经网络的语义分割分支设计;以两分支结果为先验信息,通过启发式方法进行结果聚合的端到端全景分割网络设计;以及基于多任务学习思想,设计可根据训练情况自适应调节参数改变训练权重的损失函数。本发明可以根据训练中损失函数的变化情况作为先验知识对全景分割子分支训练权重进行自适应调节,提高模型表现,完成全景分割任务。
Description
技术领域
本发明涉及图像分割技术领域,尤其涉及一种基于多任务学习的参数自适应全景分割方法。
背景技术
在计算机视觉领域,全景分割是一个新颖且重要的研究问题,深度学习作为目前主流的机器学习研究方法,在全景分割领域得到了广泛应用。全景分割的基本原则是将被分割的图像中每一个像素点的类别信息识别出来,以让计算机理解图像。这就意味着全景分割问题所带来的挑战:前景对象的定位与分类、背景像素信息的分类以及边界信息冲突解决。要让计算机理解一张所给定的图像,首先要做的便是将前景对象的位置给确定出来,再根据所检测出的热点区域像素的深层语义去确定前景对象的类别,这样一方面是将降低训练神经网络时的计算资源,另一方面可以提升处理稠密对象时的网络表现;其次是对背景信息的分类,不同于前景对象,背景信息一般缺少明确边界,且缺少规则外形等先验信息,所以并不适用于先检测再分割的方法;最后是对于冲突信息的处理,前景对象与背景信息在边界处常会存在分类置信度相近的问题,此时如何处理好信息的冲突问题往往决定着网络最终的准确度。
我们基于全景分割网络的基本结构将全景分割网络分为单阶段全景分割网络与两阶段全景分割网络这两类。基于单阶段分割网络的全景分割网络处理速度更快,实时响应能力强,更能满足部署在智能终端以及实时处理复杂现实情况的要求。基于单阶段分割网络的全景分割网络在准确性上更好,能够达到更高的准确性要求,但是响应速度少要比前者慢,更适合理想情况下的需求。
现有技术存在以下问题。第一,没有利用分支之间的任务相似性加强信息交流,网络表现力低。第二,训练过程中各分支权重设置为固定值,不能根据训练情况自适应调节。第三,在处理信息冲突时对小物体以及被覆盖物体不友好。
发明内容
为此,本发明提供一种基于多任务学习的参数自适应全景分割方法,用以克服现有技术中分支之间的任务相似性加强信息交流不强导致网络表现力低的问题。
为实现上述目的,本发明提供一种基于多任务学习的参数自适应全景分割方法,所述方法包括:
对输入的图像进行特征提取以获取其深层语义信息与特征图并通过多尺度融合增强全景分割网络对图像信息的提取能力;
利用单阶段检测网络以获取图像的前景对象位置与类别信息,通过基于检测网络实现的实例分割分支对图像进行多尺度融合,实现前景对象逐像素分割的信息获取;
以所述获取的前景对象信息为先验信息,通过全景分割网络中的语义分割分支确定背景的后验概率,通过基于先验信息的分割结果引导将两分支结果进行聚合,得到图像信息的全景分割结果;
以基于多任务学习的思想,对全景分割的目标损失函数进行设计,使训练中各分支的训练情况作为先验信息,动态调整各模块在训练中所占的权重,实现参数自适应调节的动态调整训练,增强网络准确度和鲁棒性。
进一步地,所述将输入图像通过实例分割分支获取前景对象先验信息包括:
使用特征提取网络VGG-16获取图像深层语义信息及特征图;
将所述获取的特征信息输入单阶段检测网络,获取候选框,并通过非最大值抑制,获取前景对象高置信度位置与类别;
将所述特征图通过多尺度融合,对图像中对象信息进行增强,并通过反卷积操作上采样,生成前景对象掩码,并覆盖于所述检测网络所生成的候选框中;
将所述特征图通过深度卷积网络提取图像信息,并连接全卷积层,给出逐像素分类信息,完成背景信息获取,构造语义分割分支,并以上述所获取前景对象为先验信息进行两分支之间的结果聚合。
进一步地,所述分支之间前景对象分割结果为先验信息,指导结果聚合,再以背景信息进行辅助,采用贝叶斯概率进行像素置信度结果计算;
根据下式进行结果聚合指导:
其中,P(Bi)为在实例分割分支结果中像素点分类结果置信度;P(A|Bi)为该像素点在以前景对象分割结果作为先验情况下的像素点类别置信度;P(Bi|A)为通过语义分割分支结果作为先验信息的置信度。
进一步地,所述以训练情况为先验信息进行自适应参数调整损失函数设计特征包括:
根据训练中各分支训练情况进行自适应调整,对于全景分割问题,由于其存在两个强关联性任务分支,对于整个网络的目标损失函数设置如下:
Lpan=αLins+βLsem
α+β=1
其中,Lpan为整个全景分割网络的总体目标损失函数;Lins为实例分割分支损失函数;Lsem为语义分割分支函数;α与β为相加为1的两个常数项,是对两个分支训练调整的权重。
进一步地,根据多任务学习的思想,以实时训练的情况来对损失函数中代表权重的参数进行实时调节,增强模型表现。对此,需要以损失函数的变化来对训练情况进行评估,调整后的模型自适应调整参数设计如下:
其中,D(x)为训练过程中分支调整情况的评价指标,用来实时对当前网络分支训练情况进行评估;为第k个分支在第i轮次迭代时的损失函数值;/>为第k个分支第i轮次迭代与第i-1次迭代损失函数的差值;F(x)为距离函数,用来强化分支之间损失函数的差距;最后,权值α与β通过对评价指标的归一化后确定。
进一步地,α与β初始设定为0.5。权重的自适应调整从第二个迭代轮次开始,每一轮迭代轮次数设为200。
进一步地,所述对于实例分支,其损失函数包括以下几个部分:
检测阶段候选框目标损失函数:
其中,Ldet为检测阶段损失函数;N为锚点数量;Lconf(x,c)为分类阶段的损失函数,Lloc(x,l,g)为位置误差函数。
进一步地,Lconf(x,c)具体选用softmax多类别的损失函数,Lloc(x,l,g)具体选择为smooth L1损失函数,Lconf(x,c)与Lloc(x,l,g)的函数表示如下:
其中,对于SmoothL1函数有:
进一步地,对于添加掩码部分的目标损失函数,公式如下:
其为binary cross-entropy公式。
进一步地,语义分割的目标函数如下:
所述对于语义分割分支,选择交叉熵作为目标函数,进行背景信息的分割,作为前景对象信息的后验信息补充。
与现有技术相比,本发明的有益效果在于,利用所述特征图进行多尺度的前景对象检测,获取高置信度的前景对象检测框,并对检测框内对象进行多尺度的反卷积操作,以生成前景对象掩码,以所生成的前景对象信息为先验知识,对图像添加通过语义分割分支所生成的背景信息;设计可根据训练中目标损失函数变化情况进行参数自适应调节的损失函数,能以训练中上一轮次各分支在训练中的表现情况自适应的调整各分支在训练中所占权重,提升训练的稳定性与最终的模型表现。
进一步地,本发明通过基于多任务学习的参数自适应调整全景分割方法,可以根据训练时损失函数下降情况作为先验知识,进行参数的自适应调整,从而进行高准确性的全景分割任务。
附图说明
图1为本发明实施例提供的基于多任务学习的参数自适应调整全景分割方法流程图;
图2为本发明实施例提供的输入图像示意图;
图3为本发明实施例提供的全景分割网络处理流程示意图;
图4为本发明实施例提供的实例分割分支处理流程图;
图5为本发明实施例提供的参数自适应调整函数图像;
图6为本发明实施例提供的前景对象检测与分类先期实验结果图;
图7为本发明实施例提供的前景对象分支分割先期实验结果图;
图8为本发明实施例提供的前景和背景的分割先期实验结果图。
具体实施方式
为了使本发明的目的和优点更加清楚明白,下面结合实施例对本发明作进一步描述;应当理解,此处所描述的具体实施例仅仅用于解释本发明,并不用于限定本发明。
下面参照附图来描述本发明的优选实施方式。本领域技术人员应当理解的是,这些实施方式仅仅用于解释本发明的技术原理,并非在限制本发明的保护范围。
需要说明的是,在本发明的描述中,术语“上”、“下”、“左”、“右”、“内”、“外”等指示的方向或位置关系的术语是基于附图所示的方向或位置关系,这仅仅是为了便于描述,而不是指示或暗示所述装置或元件必须具有特定的方位、以特定的方位构造和操作,因此不能理解为对本发明的限制。
此外,还需要说明的是,在本发明的描述中,除非另有明确的规定和限定,术语“安装”、“相连”、“连接”应做广义理解,例如,可以是固定连接,也可以是可拆卸连接,或一体地连接;可以是机械连接,也可以是电连接;可以是直接相连,也可以通过中间媒介间接相连,可以是两个元件内部的连通。对于本领域技术人员而言,可根据具体情况理解上述术语在本发明中的具体含义。
请参阅图1-图3所示,图1为本发明实施例提供的基于多任务学习的参数自适应调整全景分割方法流程图,图2为本发明实施例提供的输入图像示意图,图3为本发明实施例提供的全景分割网络处理流程示意图,所述方法包括:
步骤S101,将输入图像通过特征提取网络获得深层语义信息,并通过多尺度融合进行信息增强,再由实例分割分支生成前景对象的检测与分割结果。
步骤S102,利用所述实例分支所分割得到的前景信息作为先验信息,通过语义分支处理图像中剩余像素点的分割,并以前景对象的分割作为高置信度信息引导两分支结果聚合,形成全景分割结果。
步骤S103,通过所述全景分割模型设计,修改损失函数中两分支的训练权重,以训练中损失函数下降情况为评估标准,对训练较为困难的分支提升下一轮迭代中分支的权重,提升训练过程的稳定性,最终达到提升训练后模型的稳定性的结果。
具体而言,所述将输入图像通过特征提取网络获得深层语义信息,并通过多尺度融合进行信息增强,再由实例分割分支生成前景对象的检测与分割结果包括:
从数据集读取所述输入图像;
将所述图像通过由VGG-16的特征提取网络获取图像的特征信息,获取特征图。
这里,需要将VGG-16分类网络做出修改,VGG-16网络的后五层为softmax层与四个全连接层,这是网络为了处理分类任务时使用的,本发明只需求VGG-16网络的特征提取功能,所以将之后的用于分类工作的网络层去掉,分别接上实例分割分支与语义分割分支,用于图像深层语义的获取。
所述分支之间前景对象分割结果为先验信息,指导结果聚合,再以背景信息进行辅助,采用贝叶斯概率进行像素置信度结果计算。
根据下式进行结果聚合指导:
其中,P(Bi)为在实例分割分支结果中像素点分类结果置信度;P(A|Bi)为该像素点在以前景对象分割结果作为先验情况下的像素点类别置信度;P(Bi|A)为通过语义分割分支结果作为先验信息的置信度。
请继续参阅图4,其为本发明实施例提供的实例分割分支处理流程图,实例分割分支处理流程如下,所述处理图像前景对象位置、类别以及逐像素掩码分割的实例分支设计包括:
在改进后的特征提取网络后接入单阶段检测器,以6层卷积层对特征图进行下采样操作,获取不同尺度下的特征。由于卷积层的感受野问题,浅层的特征对较小面积前景对象敏感,深层的特征对于较大面积的前景对象敏感,进行特征图的多尺度融合对前景对象的特征信息。
这里,单阶段检测网络通过候选框的回归与框内对象的分类操作,获取前景对象的位置与类别信息。进一步地,将多尺度融合后得到的特征图根据候选框得到的信息进行逐层上采样操作,为前景对象覆盖上掩码,得到前景对象的逐像素分割结果。
请继续参阅图6,其为本发明实施例提供的前景对象检测与分类先期实验结果图,从图中可以看出,在检测步骤,本发明实施例可以将前景对象的位置与类别进行确认。
请继续参阅图7,其为本发明实施例提供的前景对象分支分割先期实验结果图,这里给出了本发明实施例在实例分支对前景对象进行逐像素分类后并添加掩码的结果图。从该图中可以看出,对前景对象可以进行高置信度的分割操作。
这里,根据公式(1)计算实例分支的损失函数:
Lins=Ldet+Lmask (1)
其中,Ldet包括了检测框的回归损失与检测对象的分类损失,具体由公式(2)、(3)、(4)可知:
具体而言,对于前景对象的掩码由公式(6)来进行计算:
将所述多尺度融合后的特征图通过基于全卷积网络的背景提取分支,通过对不同尺度的特征图进行卷积操作,再经过全连接层对图像进行逐像素的分类,以获取背景信息。
这里,语义分支的损失函数如公式(6)所示:
为了处理由实例分支与语义分支所带来的冲突问题,本发明选择使用以前景对象分割结果为先验指导的结果聚合方法。给予前景对象分割结果以高置信度的权重,在此基础上对结果添加背景掩码。
请继续参阅图7,对每一个前景对象都给出了其类别信息的置信度以及包含该前景对象的检测框,并对框内的前景对象进行了逐像素的分类,且用掩码覆盖了所待分割的前景对象。
这里,对于整个全景分割网络,其损失函数由实例分支的损失函数与语义分支共同组成,如公式(7)所示:
Lpan=αLins+βLsem (7)
其中,Lpan为整个全景分割网络的总体目标损失函数;Lins为实例分割分支损失函数;Lsem为语义分割分支函数;α与β为相加为1的两个常数项,是对两个分支训练调整的权重。
所述利用所设计的基于多任务学习的参数自适应调节的损失函数对网络训练进行调整。
请继续参阅图5,其为本发明实施例提供的参数自适应调节损失函数的图像。当某分支两轮迭代的损失函数差值较小时,说明上一个迭代轮次中该分支训练不够充分,需要增大该分支在下一轮次中所占的权重。
具体而言,根据公式(8)来度量两轮次之间的损失函数训练情况:
其中,函数的定义域为两轮次损失函数的差值,图5为该函数的图像,如图5所示,对于损失函数下降不明显的情况,两迭代轮次损失函数的差值会落在[0,1]区间之中,此时说明在上一个迭代轮次中该分支训练不够充分,则应该对该分支在下一轮次增加其训练中的权重。
本发明实施例增大权重的方法如公式(9)、公式(10)以及公式(11)所示:
其中,公式(9)为计算出两轮次迭代中第i个分支的损失函数下降差值,以D(x)作为下降的评价度量,公式(10)与(11)做归一化处理,并计算出实例分支与语义分支的各自在这一轮次中各自分支所占的权重。
具体而言,α与β初始设定为0.5,权重的自适应调整从第二个迭代轮次开始,每一轮迭代轮次数设为200。
请参阅图8所示,其为为本发明实施例提供的前景和背景的分割先期实验结果图。
如图8所示,对输入图像进行了逐像素的分割处理,包括了对不同前景对象的分割以及给予不同个体对象独特的类别ID,以及对于不同类别的背景信息的分割结果,本发明使用不同颜色的掩码对。
至此,已经结合附图所示的优选实施方式描述了本发明的技术方案,但是,本领域技术人员容易理解的是,本发明的保护范围显然不局限于这些具体实施方式。在不偏离本发明的原理的前提下,本领域技术人员可以对相关技术特征做出等同的更改或替换,这些更改或替换之后的技术方案都将落入本发明的保护范围之内。
Claims (10)
1.一种基于多任务学习的参数自适应全景分割方法,其特征在于,包括:
对输入的图像进行特征提取以获取其深层语义信息与特征图并通过多尺度融合增强全景分割网络对图像信息的提取能力;
利用单阶段检测网络以获取图像的前景对象位置与类别信息,通过基于检测网络实现的实例分割分支对图像进行多尺度融合,实现前景对象逐像素分割的信息获取;
以所述获取的前景对象信息为先验信息,通过全景分割网络中的语义分割分支确定背景的后验概率,通过基于先验信息的分割结果引导将两分支结果进行聚合,得到图像信息的全景分割结果;
以基于多任务学习的思想,对全景分割的目标损失函数进行设计,使训练中各分支的训练情况作为先验信息,动态调整各模块在训练中所占的权重,实现参数自适应调节的动态调整训练,增强网络准确度和鲁棒性。
2.根据权利要求1所述的基于多任务学习的参数自适应全景分割方法,其特征在于,将输入图像通过实例分割分支获取前景对象先验信息包括:
使用特征提取网络VGG-16获取图像深层语义信息及特征图;
将所述获取的特征信息输入单阶段检测网络,获取候选框,并通过非最大值抑制,获取前景对象高置信度位置与类别;
将所述特征图通过多尺度融合,对图像中对象信息进行增强,并通过反卷积操作上采样,生成前景对象掩码,并覆盖于所述检测网络所生成的候选框中;
将所述特征图通过深度卷积网络提取图像信息,并连接全卷积层,给出逐像素分类信息,完成背景信息获取,构造语义分割分支,并以上述所获取前景对象为先验信息进行两分支之间的结果聚合。
3.根据权利要求2所述的基于多任务学习的参数自适应全景分割方法,其特征在于,所述分支之间前景对象分割结果为先验信息,指导结果聚合,再以背景信息进行辅助,采用贝叶斯概率进行像素置信度结果计算;
根据下式进行结果聚合指导:
,
其中,为在实例分割分支结果中像素点分类结果置信度;/>为该像素点在以前景对象分割结果作为先验情况下的像素点类别置信度;/>为通过语义分割分支结果作为先验信息的置信度。
4.根据权利要求3所述的基于多任务学习的参数自适应全景分割方法,其特征在于,以训练情况为先验信息进行自适应参数调整损失函数设计特征包括:
根据训练中各分支训练情况进行自适应调整,对于全景分割问题,由于其存在两个强关联性任务分支,对于整个网络的目标损失函数设置如下:
,
其中,为整个全景分割网络的总体目标损失函数;/>为实例分割分支损失函数;为语义分割分支函数;α与β为相加为1的两个常数项,是对两个分支训练调整的权重。
5.根据权利要求4所述的基于多任务学习的参数自适应全景分割方法,其特征在于,根据多任务学习的思想,以实时训练的情况来对损失函数中代表权重的参数进行实时调节,增强模型表现,对此,需要以损失函数的变化来对训练情况进行评估,调整后的模型自适应调整参数设计如下:
,
其中,为训练过程中分支调整情况的评价指标,用来实时对当前网络分支训练情况进行评估;/>为第k个分支在第i轮次迭代时的损失函数值;/>为第k个分支第i轮次迭代与第i-1次迭代损失函数的差值;F(x)为距离函数,用来强化分支之间损失函数的差距;最后,权值α与β通过对评价指标的归一化后确定。
6.根据权利要求5所述的基于多任务学习的参数自适应全景分割方法,其特征在于,α与β初始设定为0.5,权重的自适应调整从第二个迭代轮次开始,每一轮迭代轮次数设为200。
7.根据权利要求6所述的基于多任务学习的参数自适应全景分割方法,其特征在于,对于实例分支,其损失函数为:
,
其中,为添加掩码部分的目标损失函数,/>为检测阶段损失函数;,
为分类阶段的损失函数,/>为位置误差函数。
8.根据权利要求7所述的基于多任务学习的参数自适应全景分割方法,其特征在于,具体选用softmax多类别的损失函数,/> 具体选择为smooth L1损失函数,/>与/>的函数表示如下:
,
其中,对于 函数有:/>。
9.根据权利要求8所述的基于多任务学习的参数自适应全景分割方法,其特征在于,对于添加掩码部分的目标损失函数,公式如下:
,
其为binary cross-entropy公式。
10.根据权利要求9所述的基于多任务学习的参数自适应全景分割方法,其特征在于,语义分割的目标函数如下:
,
对于语义分割分支,选择交叉熵作为目标函数,进行背景信息的分割,作为前景对象信息的后验信息补充。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110318993.7A CN113139549B (zh) | 2021-03-25 | 2021-03-25 | 一种基于多任务学习的参数自适应全景分割方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110318993.7A CN113139549B (zh) | 2021-03-25 | 2021-03-25 | 一种基于多任务学习的参数自适应全景分割方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN113139549A CN113139549A (zh) | 2021-07-20 |
CN113139549B true CN113139549B (zh) | 2024-03-15 |
Family
ID=76810091
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110318993.7A Active CN113139549B (zh) | 2021-03-25 | 2021-03-25 | 一种基于多任务学习的参数自适应全景分割方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113139549B (zh) |
Families Citing this family (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114972231B (zh) * | 2022-05-17 | 2024-09-06 | 华中科技大学 | 一种基于先验-后验概率编码器的多模态mr图像分割方法 |
CN115239954B (zh) * | 2022-06-21 | 2024-06-14 | 上海人工智能创新中心 | 一种用于点云全景分割的系统及方法 |
CN115914495B (zh) * | 2022-11-15 | 2024-09-13 | 大连海事大学 | 一种用于车载自动驾驶系统的目标与背景分离方法 |
Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108446662A (zh) * | 2018-04-02 | 2018-08-24 | 电子科技大学 | 一种基于语义分割信息的行人检测方法 |
CN109801297A (zh) * | 2019-01-14 | 2019-05-24 | 浙江大学 | 一种基于卷积实现的图像全景分割预测优化方法 |
CN110276765A (zh) * | 2019-06-21 | 2019-09-24 | 北京交通大学 | 基于多任务学习深度神经网络的图像全景分割方法 |
CN110287837A (zh) * | 2019-06-17 | 2019-09-27 | 上海大学 | 基于先验估计网络和空间约束混合模型的海面障碍物检测方法 |
CN110533048A (zh) * | 2018-05-23 | 2019-12-03 | 上海交通大学 | 基于全景区域场景感知的组合语义层次连接模型的实现方法及系统 |
WO2020125495A1 (zh) * | 2018-12-17 | 2020-06-25 | 中国科学院深圳先进技术研究院 | 一种全景分割方法、装置及设备 |
Family Cites Families (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US10986325B2 (en) * | 2018-09-12 | 2021-04-20 | Nvidia Corporation | Scene flow estimation using shared features |
-
2021
- 2021-03-25 CN CN202110318993.7A patent/CN113139549B/zh active Active
Patent Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108446662A (zh) * | 2018-04-02 | 2018-08-24 | 电子科技大学 | 一种基于语义分割信息的行人检测方法 |
CN110533048A (zh) * | 2018-05-23 | 2019-12-03 | 上海交通大学 | 基于全景区域场景感知的组合语义层次连接模型的实现方法及系统 |
WO2020125495A1 (zh) * | 2018-12-17 | 2020-06-25 | 中国科学院深圳先进技术研究院 | 一种全景分割方法、装置及设备 |
CN109801297A (zh) * | 2019-01-14 | 2019-05-24 | 浙江大学 | 一种基于卷积实现的图像全景分割预测优化方法 |
CN110287837A (zh) * | 2019-06-17 | 2019-09-27 | 上海大学 | 基于先验估计网络和空间约束混合模型的海面障碍物检测方法 |
CN110276765A (zh) * | 2019-06-21 | 2019-09-24 | 北京交通大学 | 基于多任务学习深度神经网络的图像全景分割方法 |
Non-Patent Citations (1)
Title |
---|
全景分割研究综述;徐鹏斌;瞿安国;王坤峰;李大字;自动化学报;第47卷(第003期);全文 * |
Also Published As
Publication number | Publication date |
---|---|
CN113139549A (zh) | 2021-07-20 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN113139549B (zh) | 一种基于多任务学习的参数自适应全景分割方法 | |
US11321593B2 (en) | Method and apparatus for detecting object, method and apparatus for training neural network, and electronic device | |
CN107230218B (zh) | 用于生成对从安装在运载工具上的摄像机捕捉的图像导出的估计的置信度测量的方法和设备 | |
CN109671070B (zh) | 一种基于特征加权和特征相关性融合的目标检测方法 | |
EP3389011B1 (en) | Image analysis system for detecting edges of a nucleus | |
CN108280455B (zh) | 人体关键点检测方法和装置、电子设备、程序和介质 | |
CN111160249A (zh) | 基于跨尺度特征融合的光学遥感图像多类目标检测方法 | |
CN111461213B (zh) | 一种目标检测模型的训练方法、目标快速检测方法 | |
JP2017004480A (ja) | 顕著性情報取得装置及び顕著性情報取得方法 | |
CN108305260B (zh) | 一种图像中角点的检测方法、装置及设备 | |
CN110443258B (zh) | 文字检测方法、装置、电子设备及存储介质 | |
CN109409288B (zh) | 图像处理方法、装置、电子设备和存储介质 | |
CN114067186B (zh) | 一种行人检测方法、装置、电子设备及存储介质 | |
CN116012291A (zh) | 工业零件图像缺陷检测方法及系统、电子设备和存储介质 | |
CN112861915A (zh) | 一种基于高级语义特征无锚框非合作目标检测方法 | |
CN114331946B (zh) | 一种图像数据处理方法、设备以及介质 | |
CN113870254B (zh) | 目标对象的检测方法、装置、电子设备及存储介质 | |
CN113139906B (zh) | 一种生成器的训练方法、装置与存储介质 | |
CN114332133A (zh) | 基于改进CE-Net的新冠肺炎CT图像感染区分割方法及系统 | |
CN112926667B (zh) | 深度融合边缘与高层特征的显著性目标检测方法及装置 | |
CN117853942A (zh) | 用于云雾的识别方法、云雾识别装置及云雾识别系统 | |
CN116363136B (zh) | 一种机动车部件自动化生产在线筛选方法及系统 | |
CN111310839A (zh) | 一种钼靶图像中乳头位置的检测方法及系统 | |
CN116579985A (zh) | 一种基于RefineDet模型的晶圆缺陷检测方法 | |
CN116503763A (zh) | 基于二元协同反馈的无人机巡航森林火灾检测方法 |
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 |