CN109712105A - 一种结合彩色和深度信息的图像显著目标检测方法 - Google Patents

一种结合彩色和深度信息的图像显著目标检测方法 Download PDF

Info

Publication number
CN109712105A
CN109712105A CN201811582701.5A CN201811582701A CN109712105A CN 109712105 A CN109712105 A CN 109712105A CN 201811582701 A CN201811582701 A CN 201811582701A CN 109712105 A CN109712105 A CN 109712105A
Authority
CN
China
Prior art keywords
rgb
well
marked target
notable
image
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.)
Granted
Application number
CN201811582701.5A
Other languages
English (en)
Other versions
CN109712105B (zh
Inventor
汪宁宁
龚小谨
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Zhejiang University ZJU
Original Assignee
Zhejiang University ZJU
Priority date (The priority date 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 date listed.)
Filing date
Publication date
Application filed by Zhejiang University ZJU filed Critical Zhejiang University ZJU
Priority to CN201811582701.5A priority Critical patent/CN109712105B/zh
Publication of CN109712105A publication Critical patent/CN109712105A/zh
Application granted granted Critical
Publication of CN109712105B publication Critical patent/CN109712105B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Landscapes

  • Image Analysis (AREA)

Abstract

本发明公开了一种结合彩色和深度信息的图像显著目标检测方法。采集多幅已知显著目标区域的彩色图像及其对应的深度图像,构建神经网络并输入彩色图像与深度图像,分别得到彩色和深度信息预测的显著图,神经网络设置一路分支预测权重图,彩色信息预测的显著图和深度信息预测的显著图根据权重图得到融合后的最终显著目标检测图;构建显著图损失函数、权重图损失函数和边缘保持损失函数进行训练,采用训练后的神经网络处理待测彩色图像及其深度图像,获得显著目标检测结果。本发明能够选择彩色及深度信息预测的显著图各自最有效的部分进行融合,同时保持边缘细节,与以往结合彩色和深度信息的图像显著目标检测方法相比取得了更好的效果。

Description

一种结合彩色和深度信息的图像显著目标检测方法
技术领域
本发明属于图像处理技术领域,具体涉及一种结合彩色和深度信息的图像显著目标检测方法。
背景技术
图像显著目标检测旨在从包含大量信息的图像场景中检测出重要的、引人注意的区域。研究发现人类的视觉机制总是能够快速从场景中找到这类区域并对其进行分析,而很少对其余并不显著的区域加以处理,显著目标检测任务目的即在于构造自动化的工具模拟此视觉机制,提取图像中的显著区域。对于后续处理来说,以显著区域作为处理单元,可以节省对全图进行处理的时间,提高图像的处理效率。因此,显著目标检测被广泛地应用在图像分割、图像检索和目标识别等领域,是当前计算机视觉领域的热点研究方向之一。
以往大部分的图像显著目标检测算法仅在彩色图像上实现,没有利用其他的补充信息。随着三维感知传感器技术的发展,物体的距离信息是人类可以获取并利用的重要线索。在彩色图像中,目标物体有时与背景或者周围区域拥有相似的纹理、颜色等特征,但与周围物体存在着距离上的差异,此时深度信息通常能够给出具有判别性的线索。这意味着,我们可以结合彩色和深度信息,使其相互补充,共同预测,从而使得显著目标检测任务能够获得表现更好、更具鲁棒性的预测结果。
然而如何对彩色信息与深度信息进行更好地融合仍是一个值得思考的问题,以往的方法对于两种不同模态结合方式的挖掘依然不够深入,只有充分发挥彩色信息与深度信息各自有效的内容,才能有益于最终的检测结果。另外,保持预测显著图的边缘细节也是帮助检测结果获得更好表现的一个关注点。
发明内容
本发明的目的在于克服现有技术的不足,提出新的融合方式,更充分地利用彩色信息和深度信息各自的有效部分,改进结合彩色和深度信息的显著目标检测效果,同时加入边缘损失函数,保持预测显著图的边缘细节。
本发明采用的技术方案是包括如下步骤:
步骤1)采集多幅已知显著目标区域的彩色图像RGB及其对应的深度图像D以及彩色图像RGB和深度图像D对应的显著目标真值图Y;显著目标真值图Y为显著目标所在区域和非显著目标所在区域的标签,实质上为二值化图。显著目标真值图是总共具有T个像素点的图像,其中yi表示在位置i上像素的显著值,yi=1表示显著的像素点,yi=0表示非显著的像素点。
步骤2)构建显著目标检测的神经网络,包括有:将彩色图像RGB和深度图像D输入神经网络得到彩色信息预测的显著图Prgb和深度信息预测的显著图Pd,神经网络设置一路分支并结合显著目标真值图Y预测获得一张权重图SW,权重图SW用于帮助彩色信息预测的显著图Prgb和深度信息预测的显著图Pd的加权融合,彩色信息预测的显著图Prgb和深度信息预测的显著图Pd根据权重图SW得到融合后的显著目标检测图Pfuse
步骤3)构建显著图损失函数Lsal、权重图损失函数Lsw和显著图边缘损失函数Ledge三个损失函数,将三个损失函数计算获得的损失函数值通过反向传播进行迭代计算,从而对神经网络进行训练;
步骤4)采用训练后的神经网络处理未知显著区域的彩色图像RGB及其对应的深度图像D,获得待测图像的显著目标检测结果。
本发明特别设计了权重图SW及其标签图,用于神经网络中彩色和深度信息的融合处理,能够帮助具有深度信息的彩色图像实现显著目标的准确检测。
所述步骤2)包括依次连接的特征提取模块、权重图预测模块和显著目标检测图融合模块,具体为:
步骤2.1):将彩色图像RGB和深度图像D输入特征提取模块,输出得到彩色图像特征图Frgb和深度图像特征图Fd,利用获得的彩色图像特征图Frgb和深度图像特征图Fd预测得到显著图,即分别得到彩色信息预测的显著图Prgb和深度信息预测的显著图Pd
步骤2.2):将彩色图像特征图Frgb与深度图像特征图Fd级联后作为新的分支输出,得到新特征图Frgbd,将新特征图Frgbd输入到权重图预测模块预测得到权重图SW;
步骤2.3):显著目标检测图融合模块利用权重图SW对彩色信息预测的显著图Prgb和深度信息预测的显著图Pd进行加权融合,得到融合后的显著目标检测图Pfuse
所述步骤(2.1)中,彩色图像RGB和深度图像D均采用相同方式进行处理得到图像特征图Frgb和深度图像特征图Fd,以下以彩色图像RGB为例说明:
所述的特征提取模块包括五个依次连接的小模块,每个小模块均由卷积层、激活层、池化层依次连接构成,小模块中的参数相同,每个小模块各自输出一张特征图,五张特征图分别为F1、F2、F3、F4和F5,然后采用一种往前递进的方式反向融合五张特征图,得到最终输出的特征图;
采用一种往前递进的方式反向融合五张特征图,具体为:在得到五张特征图后,采用以下公式从特征图F5开始进行计算得到递进融合后的特征图F4,再依次以此处理最后得到递进融合后第1个小模块的特征图
其中,表示递进融合后第m个小模块的特征图,g(·)表示融合函数,由一个卷积层和一个激活层组成,{·,·}表示对两张特征图的级联操作;
最后以递进融合后第1个小模块的特征图作为彩色图像特征图Frgb,即获得的彩色图像特征图Frgb再通过一个卷积层和一个归一化函数输出得到彩色信息预测的显著图Prgb
Prgb=h(W*Frgb+b)
其中,*表示卷积操作,W和b分别表示卷积层中卷积核的权重和偏置,h(·)表示Sigmoid函数。
所述步骤2.2)中,将彩色图像特征图Frgb和深度图像特征图Fd进行级联之后,通过融合函数输出得到新特征图Frgbd,融合函数由一个卷积层和一个激活层组成,利用新特征图Frgbd处理预测权重图SW,具体采用以下公式进行计算:
Frgbd=g({Frgb,Fd})
SW=h(W*Frgbd+b)
并且,为权重图SW构建一个标签图Ysw,标签图Ysw由彩色信息预测的显著图Prgb与显著目标真值图Y计算得到,具体通过以下公式进行计算:
Ysw=Prgb·Y+(1-Prgb)·(1-Y)
其中,·表示彩色信息预测的显著图Prgb与显著目标真值图Y之间的点乘运算。标签Ysw用于在训练神经网络的过程中监督权重图SW的学习训练过程。如果图像中位置i上的像素点在Prgb和Y中均显著或均不显著,则被设为高权重,反之则被设为低权重。
所述步骤2.3)中,利用权重图SW采用以下公式对彩色信息预测的显著图Prgb和深度信息预测的显著图Pd进行加权融合,得到融合后的显著目标检测图Pfuse
Pfuse=SW·Prgb+(1-SW)·Pd
所述步骤3)中,三个损失函数的计算过程具体为:
步骤3.1)将彩色信息预测的显著图Prgb、深度信息预测的显著图Pd和融合后的显著目标检测图Pfuse输入到显著图损失函数Lsal中计算预测显著图的损失函数值,具体采用以下公式计算:
Lsal=Lrgb+Ld+Lfuse
其中,θ表示神经网络中的所有参数,i表示像素点在显著目标真值图Y中的位置,yi表示显著目标真值图Y中在位置i上像素点的显著值,Prgb(yi=1|RGB,D;θ)表示彩色信息预测的显著图Prgb中位置i上的像素点属于显著像素点的概率,Pd(yi=1|RGB,D;θ)表示深度信息预测的显著图Pd中位置i上的像素点属于显著像素点的概率,Pfuse(yi=1|RGB,D;θ)表示融合后的显著目标检测图Pfuse中位置i上的像素点属于显著像素点的概率;Lrgb、Ld和Lfuse分别表示彩色信息预测的显著图Prgb、深度信息预测的显著图Pd和融合后的显著目标检测图Pfuse对应的损失函数,Lsal表示三张预测显著图对应的Lrgb、Ld和Lfuse损失函数之和;
步骤3.2):将权重图SW输入到权重图损失函数Lsw中计算预测权重图SW的损失函数值,权重图损失函数Lsw采用以下公式:
其中,yi sw表示标签图Ysw中在位置i上像素点的权重值,Psw(yi sw=1|RGB,D;θ)表示预测的权重图中位置i上的像素点属于高权重的概率;
步骤3.3):对显著目标检测图Pfuse提取边缘信息E,将边缘信息E输入到显著图边缘损失函数Ledge中计算预测边缘信息E的损失函数值,所述的边缘信息由图像的梯度得到,具体采用以下公式计算:
其中,分别表示显著目标检测图Pfuse或显著目标真值图Y在水平方向和竖直方向上的梯度,表示显著目标检测图Pfuse中位置i上的像素点在水平方向上的梯度值,表示显著目标检测图Pfuse中位置i上的像素点在竖直方向上的梯度值,表示显著目标真值图Y中位置i上的像素点在水平方向上的梯度值,表示显著目标检测图Pfuse中位置i上的像素点在水平方向上的梯度值,|Y|表示显著目标真值图Y中像素点的数目。
所述步骤(4)具体为:针对未知显著目标真值区域的待测彩色图像及其对应的深度图像,输入训练后且去掉所有损失函数的神经网络中,预测得到显著目标检测图Pfuse,显著目标检测图Pfuse上的值代表了图像每个像素点属于显著像素点的概率值,最后选择概率值大于阈值η的像素点作为显著像素点,其余作为非显著像素点。
优选的,神经网络的训练是利用Adam优化器训练。
本发明方法构造一个深度神经网络,在普通显著目标检测网络的基础上增加一路权重图预测的分支,同时构建一个权重图的标签监督权重图的学习,利用学习得到的权重图指导彩色信息预测的显著图和深度信息预测的显著图之间的融合,此外,本发明方法在普通显著图损失函数的基础上,加入边缘保持项从而保持显著目标区域的边缘细节。
本发明的有益效果是:
本发明方法利用权重图模块学习彩色和深度信息各部分的权重,以此指导两种模态信息各自预测的显著图的融合,更加充分地利用彩色信息和深度信息各自的有效部分,同时在损失函数中加入了边缘保持项改善边缘细节。
本发明和以往结合彩色和深度信息的图像显著目标检测方法相比,取得了更好的效果。
附图说明
图1是本发明方法的流程示意图。
图2-5是采用现有图像显著目标检测方法与本发明方法进行图像显著目标检测的结果对比图,其中:
图2-5中的(a)表示为待检测彩色图像RGB;
图2-5中的(b)表示为(a)的深度图像D;
图2-5中的(c)表示为GP算法(Ren J,Gong X,Yu L,et al.Exploiting globalpriors for RGB-D saliency detection[C]//Proceedings of the IEEE Conference onComputer Vision and Pattern Recognition Workshops.2015:25-32.)得到的显著目标图像;
图2-5中的(d)表示为LBE算法(Feng D,Barnes N,You S,et al.Localbackground enclosure for RGB-D salient object detection[C]//Proceedings ofthe IEEE Conference on Computer Vision and Pattern Recognition.2016:2343-2350.)得到的显著目标图像;
图2-5中的(e)表示为CTMF算法(Han J,Chen H,Liu N,et al.CNNs-based RGB-Dsaliency detection via cross-view transfer and multiview fusion[J].IEEETransactions on Cybernetics,2017.)得到的显著目标图像;
图2-5中的(f)表示为MPCI算法(Chen H,Li Y,Su D.Multi-modal fusionnetwork with multi-scale multi-path and cross-modal interactions for RGB-Dsalient object detection[J].Pattern Recognition,2019,86:376-385.)得到的显著目标图像;
图2-5中的(g)表示为PCA算法(Chen H,Li Y.Progressively Complementarity-Aware Fusion Network for RGB-D Salient Object Detection[C]//Proceedings ofthe IEEE Conference on Computer Vision and Pattern Recognition.2018:3051-3060.)得到的显著目标图像;
图2-5中的(h)表示通过本发明得到的显著目标检测图像;
图2-5中的(i)表示为人工标注的显著目标真值图像;
图6是采用现有的图像显著目标检测方法与本发明方法进行图像显著目标检测后的F-Measure柱状图的对比。
具体实施方式
下面结合附图与实施例对本发明进行进一步说明。
本发明所述的显著目标一般是指在图像中和周围对比强烈的引起人视觉注意的目标物体。按照本发明发明内容的完整方法实施的实施例及其实施过程是:
(1)采集一幅已知显著目标区域的彩色图像RGB及其对应的深度图像D,显著目标真值图为Y;其中显著目标真值图是总共具有T个像素点的图像,yi表示在位置i上像素的显著值,yi=1表示显著的像素点,yi=0表示非显著的像素点。
(2)构建显著目标检测的神经网络。
具体包括特征提取模块、权重图预测模块和显著目标检测图融合模块。彩色图像RGB和深度图像D输入特征提取模块,输出得到彩色图像特征图Frgb和深度图像特征图Fd,利用获得的彩色图像特征图Frgb和深度图像特征图Fd预测得到显著图,即分别得到彩色信息预测的显著图Prgb和深度信息预测的显著图Pd;将彩色图像特征图Frgb与深度图像特征图Fd级联后,作为神经网络一路新的分支,对其处理得到新的特征图Frgbd,通过特征图Frgbd预测权重图SW;利用获得的权重图SW对彩色信息预测的显著图Prgb和深度信息预测的显著图Pd的进行加权融合,得到融合后的显著目标检测图Pfuse
使用VGG16模型中的卷积层作为特征提取模块中的五个基本的小模块,得到不同感受野的五张特征图F1,F2,F3,F4和F5,然后采用一种往前递进的方式反向融合五张特征图,得到彩色图像RGB和深度图像D的特征图Frgb和Fd
具体实施中,VGG16模型采用Simonyan K,Zisserman A.Very deepconvolutional networks for large-scale image recognition文献中的计算方法。
(3)构建损失函数模块,利用Adam优化器训练神经网络。
具体包括显著图损失函数模块Lsal、权重图损失函数模块Lsw和边缘损失函数模块Ledge。将彩色信息预测的显著图Prgb、深度信息预测的显著图Pd和融合后的显著目标检测图Pfuse输入到显著图损失函数模块Lsal计算预测显著图的损失函数值;将权重图SW输入到权重图损失函数模块Lsw中计算预测权重图的损失函数值;对显著目标检测图提取边缘信息E,将边缘信息E输入到边缘损失函数模块Ledge中计算预测边缘信息的损失函数值;采用Adam优化器训练神经网络,获得神经网络的参数。
具体实施中,学习率设置为10-4,训练至模型收敛,保存神经网络的参数。
(4)训练结束后,针对未知显著目标真值区域的待测彩色图像RGB及其对应的深度图像D,输入训练后且去掉了所有损失函数模块的神经网络中,预测得到显著目标检测图Pfuse,显著目标检测图Pfuse上的值代表了图像每个像素点属于显著像素点的概率值,最后选择概率值大于阈值η的像素点作为显著像素点,其余作为非显著像素点。
本实施例最后在三个标准数据集NJUD(Ju R,Ge L,Geng W,et al.Depthsaliency based on anisotropic center-surround difference[C]//Image Processing(ICIP),2014IEEE International Conference on.IEEE,2014:1115-1119.)、NLPR(PengH,Li B,Xiong W,et al.Rgbd salient object detection:a benchmark and algorithms[C]//European conference on computer vision.Springer,Cham,2014:92-109.)和STEREO(Niu Y,Geng Y,Li X,et al.Leveraging stereopsis for saliency analysis[C]//Computer Vision and Pattern Recognition(CVPR),2012IEEE Conferenceon.IEEE,2012:454-461.)上评估,在NJUD中随机抽取1400个样本,NLPR中抽取650个样本组成训练集,另外在NJUD中随机抽取100个样本,NLPR中抽取50个样本组成验证集,剩余的样本与STEREO组成测试集。采用本发明方法对测试集中的数据进行显著目标检测,然后采用其他现有的图像显著目标检测方法进行检测,得出的检测结果如图2至图5所示。各幅图中(a)为待检测彩色图像,(b)为对应的深度信息,(c)为GP算法得到的显著图像,(d)为LBE算法得到的显著图像,(e)为CTMF算法得到的显著图像,(f)为MPCI算法得到的显著图像,(g)为PCA算法得到的显著图像,(h)为本发明算法得到的显著图像,(i)为人工标注的显著目标真值图像。
实施例一如图2所示,图2展示了处于背景中的一个人像,与图2的(i)相比,通过本发明方法得到的图2的(h)与(i)最为接近,并且在人像与图2的背景较为接近的情况下,本发明方法所得到的图像与(c)~(g)相比较为准确。
实施例二如图3所示,图3展示了在海景植物的背景下的一个手拿箱的潜水员,本发明在图3的背景图像较为丰富的情况下,图3(h)提取到的显著目标仍较为准确。
实施例三如图4所示,图4展示了一个置于展示座上的物品,与人工标注的显著目标真值图像的图4(i)相比,本发明的图4的(h)最为精确。
实施例四如图5所示,图5展示了处于草坪之上的一个带有三角架的天文镜,由图5的(f)、(g)、(h)的对比可以看出,本发明的图5(h)较为清晰、准确,更符合图5的(i)的图像特点。
为了对本发明方法的实施例一、二、三和四进行客观数据指标评价,选用F-Measure指标量化评价不同方法的实验结果,如图6所示。F-Measure是精确率和召回率的综合考量,其中精确率定义为被正确检测到的显著性像素点数目与检测到的显著性像素点数目之比,召回率定义为被正确检测到的显著性像素点数目与真正的显著性像素点数目之比,F-Measure的定义如下式:
本发明实施例中取值:β2=0.3。
由图2至图6可以得出结论:与其他现有显著性方法进行比较,采用本发明图像显著性检测方法的表现结果更好,能更充分地利用彩色信息和深度信息中的有效内容,更加完整和准确地从背景中提取显著物体。
本方法首先生成彩色图像和深度图像的特征图与各自预测的显著图,然后将两张特征图输入权重图预测模块,接着利用输出的权重图指导彩色信息预测的显著图与深度信息预测的显著图,得到融合后的最终检测结果。本发明能够选择彩色及深度信息预测的显著图各自最有效的部分进行融合,同时保持边缘细节,与以往结合彩色和深度信息的图像显著目标检测方法相比取得了更好的效果。

Claims (8)

1.一种结合彩色和深度信息的图像显著目标检测方法,其特征在于包括以下步骤:
步骤1)采集多幅已知显著目标区域的彩色图像RGB及其对应的深度图像D以及彩色图像RGB和深度图像D对应的显著目标真值图Y;
步骤2)构建显著目标检测的神经网络,包括有:将彩色图像RGB和深度图像D输入神经网络得到彩色信息预测的显著图Prgb和深度信息预测的显著图Pd,神经网络设置一路分支并结合显著目标真值图Y预测获得一张权重图SW,彩色信息预测的显著图Prgb和深度信息预测的显著图Pd根据权重图SW得到融合后的显著目标检测图Pfuse
步骤3)构建显著图损失函数Lsal、权重图损失函数Lsw和显著图边缘损失函数Ledge三个损失函数,将三个损失函数计算获得的损失函数值通过反向传播进行迭代计算,从而对神经网络进行训练;
步骤4)采用训练后的神经网络处理未知显著区域的彩色图像RGB及其对应的深度图像D,获得待测图像的显著目标检测结果。
2.根据权利要求1所述的一种结合彩色和深度信息的图像显著目标检测方法,其特征在于:所述步骤2)包括依次连接的特征提取模块、权重图预测模块和显著目标检测图融合模块,具体为:
步骤2.1):将彩色图像RGB和深度图像D输入特征提取模块,输出得到彩色图像特征图Frgb和深度图像特征图Fd,利用获得的彩色图像特征图Frgb和深度图像特征图Fd预测得到显著图,即分别得到彩色信息预测的显著图Prgb和深度信息预测的显著图Pd
步骤2.2):将彩色图像特征图Frgb与深度图像特征图Fd级联后输出,得到新特征图Frgbd,将新特征图Frgbd输入到权重图预测模块预测得到权重图SW;
步骤2.3):显著目标检测图融合模块利用权重图SW对彩色信息预测的显著图Prgb和深度信息预测的显著图Pd进行加权融合,得到融合后的显著目标检测图Pfuse
3.根据权利要求2所述的一种结合彩色和深度信息的图像显著目标检测方法,其特征在于:所述步骤(2.1)中,彩色图像RGB和深度图像D均采用相同方式进行处理得到图像特征图Frgb和深度图像特征图Fd,以下以彩色图像RGB为例说明:
所述的特征提取模块包括五个依次连接的小模块,每个小模块均由卷积层、激活层、池化层依次连接构成,每个小模块各自输出一张特征图,五张特征图分别为F1、F2、F3、F4和F5,然后采用一种往前递进的方式反向融合五张特征图,得到最终输出的特征图;
采用一种往前递进的方式反向融合五张特征图,具体为:在得到五张特征图后,采用以下公式从特征图F5开始进行计算得到递进融合后的特征图F4,最后得到递进融合后第1个小模块的特征图
其中,表示递进融合后第m个小模块的特征图,g(·)表示融合函数,由一个卷积层和一个激活层组成,{·,·}表示对两张特征图的级联操作;
最后以递进融合后第1个小模块的特征图作为彩色图像特征图Frgb,获得的彩色图像特征图Frgb再通过一个卷积层和一个归一化函数输出得到彩色信息预测的显著图Prgb
Prgb=h(W*Frgb+b)
其中,*表示卷积操作,W和b分别表示卷积层中卷积核的权重和偏置,h(·)表示Sigmoid函数。
4.根据权利要求2所述的一种结合彩色和深度信息的图像显著目标检测方法,其特征是:
所述步骤2.2)中,将彩色图像特征图Frgb和深度图像特征图Fd进行级联之后,通过融合函数输出得到新特征图Frgbd,利用新特征图Frgbd处理预测权重图SW,具体采用以下公式进行计算:
Frgbd=g({Frgb,Fd})
SW=h(W*Frgbd+b)
并且,为权重图SW构建一个标签图Ysw,标签图Ysw由彩色信息预测的显著图Prgb与显著目标真值图Y计算得到,具体通过以下公式进行计算:
Ysw=Prgb·Y+(1-Prgb)·(1-Y)
其中,·表示彩色信息预测的显著图Prgb与显著目标真值图Y之间的点乘运算。
5.根据权利要求2所述的一种结合彩色和深度信息的图像显著目标检测方法,其特征是:所述步骤2.3)中,利用权重图SW采用以下公式对彩色信息预测的显著图Prgb和深度信息预测的显著图Pd进行加权融合,得到融合后的显著目标检测图Pfuse
Pfuse=SW·Prgb+(1-SW)·Pd
6.根据权利要求1所述的一种结合彩色和深度信息的图像显著目标检测方法,其特征是:所述步骤3)中,三个损失函数的计算过程具体为:
步骤3.1)将彩色信息预测的显著图Prgb、深度信息预测的显著图Pd和融合后的显著目标检测图Pfuse输入到显著图损失函数Lsal中计算预测显著图的损失函数值,具体采用以下公式计算:
Lsal=Lrgb+Ld+Lfuse
其中,θ表示神经网络中的所有参数,i表示像素点在显著目标真值图Y中的位置,yi表示显著目标真值图Y中在位置i上像素点的显著值,
Prgb(yi=1|RGB,D;θ)表示彩色信息预测的显著图Prgb中位置i上的像素点属于显著像素点的概率,Pd(yi=1|RGB,D;θ)表示深度信息预测的显著图Pd中位置i上的像素点属于显著像素点的概率,Pfuse(yi=1|RGB,D;θ)表示融合后的显著目标检测图Pfuse中位置i上的像素点属于显著像素点的概率;Lrgb、Ld和Lfuse分别表示彩色信息预测的显著图Prgb、深度信息预测的显著图Pd和融合后的显著目标检测图Pfuse对应的损失函数,Lsal表示三张预测显著图对应的Lrgb、Ld和Lfuse损失函数之和;
步骤3.2):将权重图SW输入到权重图损失函数Lsw中计算预测权重图SW的损失函数值,权重图损失函数Lsw采用以下公式:
其中,yi sw表示标签图Ysw中在位置i上像素点的权重值,Psw(yi sw=1|RGB,D;θ)表示预测的权重图中位置i上的像素点属于高权重的概率;
步骤3.3):对显著目标检测图Pfuse提取边缘信息E,将边缘信息E输入到显著图边缘损失函数Ledge中计算预测边缘信息E的损失函数值,具体采用以下公式计算:
其中,分别表示显著目标检测图Pfuse或显著目标真值图Y在水平方向和竖直方向上的梯度,表示显著目标检测图Pfuse中位置i上的像素点在水平方向上的梯度值,表示显著目标检测图Pfuse中位置i上的像素点在竖直方向上的梯度值,表示显著目标真值图Y中位置i上的像素点在水平方向上的梯度值,表示显著目标检测图Pfuse中位置i上的像素点在水平方向上的梯度值,|Y|表示显著目标真值图Y中像素点的数目。
7.根据权利要求1所述的一种结合彩色和深度信息的图像显著目标检测方法,其特征在于:所述步骤(4)具体为:针对未知显著目标真值区域的待测彩色图像及其对应的深度图像,输入训练后且去掉所有损失函数的神经网络中,预测得到显著目标检测图Pfuse,显著目标检测图Pfuse上的值代表了图像每个像素点属于显著像素点的概率值,最后选择概率值大于阈值η的像素点作为显著像素点,其余作为非显著像素点。
8.根据权利要求1所述的一种结合彩色和深度信息的图像显著目标检测方法,其特征在于:所述的神经网络的训练是利用Adam优化器训练。
CN201811582701.5A 2018-12-24 2018-12-24 一种结合彩色和深度信息的图像显著目标检测方法 Active CN109712105B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN201811582701.5A CN109712105B (zh) 2018-12-24 2018-12-24 一种结合彩色和深度信息的图像显著目标检测方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN201811582701.5A CN109712105B (zh) 2018-12-24 2018-12-24 一种结合彩色和深度信息的图像显著目标检测方法

Publications (2)

Publication Number Publication Date
CN109712105A true CN109712105A (zh) 2019-05-03
CN109712105B CN109712105B (zh) 2020-10-27

Family

ID=66257451

Family Applications (1)

Application Number Title Priority Date Filing Date
CN201811582701.5A Active CN109712105B (zh) 2018-12-24 2018-12-24 一种结合彩色和深度信息的图像显著目标检测方法

Country Status (1)

Country Link
CN (1) CN109712105B (zh)

Cited By (13)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN110210539A (zh) * 2019-05-22 2019-09-06 西安电子科技大学 多级深度特征融合的rgb-t图像显著性目标检测方法
CN110276831A (zh) * 2019-06-28 2019-09-24 Oppo广东移动通信有限公司 三维模型的建构方法和装置、设备、计算机可读存储介质
CN110458797A (zh) * 2019-06-18 2019-11-15 南开大学 一种基于深度图过滤器的显著性物体检测方法
CN110458178A (zh) * 2019-08-12 2019-11-15 浙江科技学院 多模态多拼接的rgb-d显著性目标检测方法
CN110472639A (zh) * 2019-08-05 2019-11-19 山东工商学院 一种基于显著性先验信息的目标提取方法
CN111047571A (zh) * 2019-12-10 2020-04-21 安徽大学 一种具有自适应选择训练过程的图像显著目标检测方法
CN111209840A (zh) * 2019-12-31 2020-05-29 浙江大学 一种基于多传感器数据融合的3d目标检测方法
CN111402306A (zh) * 2020-03-13 2020-07-10 中国人民解放军32801部队 一种基于深度学习的微光/红外图像彩色融合方法及系统
CN111428602A (zh) * 2020-03-18 2020-07-17 浙江科技学院 卷积神经网络边缘辅助增强的双目显著性图像检测方法
CN112395922A (zh) * 2019-08-16 2021-02-23 杭州海康威视数字技术股份有限公司 面部动作检测方法、装置及系统
CN112651406A (zh) * 2020-12-18 2021-04-13 浙江大学 一种深度感知和多模态自动融合的rgb-d显著性目标检测方法
CN114419323A (zh) * 2022-03-31 2022-04-29 华东交通大学 基于跨模态学习与领域自适应rgbd图像语义分割方法
CN115496975A (zh) * 2022-08-29 2022-12-20 锋睿领创(珠海)科技有限公司 辅助加权数据融合方法、装置、设备及存储介质

Citations (9)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN104123529A (zh) * 2013-04-25 2014-10-29 株式会社理光 人手检测方法及系统
CN104574375A (zh) * 2014-12-23 2015-04-29 浙江大学 结合彩色和深度信息的图像显著性检测方法
CN104680546A (zh) * 2015-03-12 2015-06-03 安徽大学 一种图像显著目标检测方法
CN106296692A (zh) * 2016-08-11 2017-01-04 深圳市未来媒体技术研究院 基于对抗网络的图像显著性检测方法
CN106373162A (zh) * 2015-07-22 2017-02-01 南京大学 一种基于显著性融合与传播的显著物体检测方法
CN106446908A (zh) * 2016-08-31 2017-02-22 乐视控股(北京)有限公司 图像中目标检测方法及装置
CN107085848A (zh) * 2017-04-20 2017-08-22 安徽大学 一种rgb‑d图显著性的检测方法
CN108388901A (zh) * 2018-02-05 2018-08-10 西安电子科技大学 基于空间-语义通道的协同显著目标检测方法
CN108665455A (zh) * 2018-05-14 2018-10-16 北京航空航天大学 图像显著性预测结果的评价方法和装置

Patent Citations (9)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN104123529A (zh) * 2013-04-25 2014-10-29 株式会社理光 人手检测方法及系统
CN104574375A (zh) * 2014-12-23 2015-04-29 浙江大学 结合彩色和深度信息的图像显著性检测方法
CN104680546A (zh) * 2015-03-12 2015-06-03 安徽大学 一种图像显著目标检测方法
CN106373162A (zh) * 2015-07-22 2017-02-01 南京大学 一种基于显著性融合与传播的显著物体检测方法
CN106296692A (zh) * 2016-08-11 2017-01-04 深圳市未来媒体技术研究院 基于对抗网络的图像显著性检测方法
CN106446908A (zh) * 2016-08-31 2017-02-22 乐视控股(北京)有限公司 图像中目标检测方法及装置
CN107085848A (zh) * 2017-04-20 2017-08-22 安徽大学 一种rgb‑d图显著性的检测方法
CN108388901A (zh) * 2018-02-05 2018-08-10 西安电子科技大学 基于空间-语义通道的协同显著目标检测方法
CN108665455A (zh) * 2018-05-14 2018-10-16 北京航空航天大学 图像显著性预测结果的评价方法和装置

Non-Patent Citations (2)

* Cited by examiner, † Cited by third party
Title
JUNWEI HAN: "CNNs-Based RGB-D Saliency Detection via Cross-View Transfer and Multiview Fusion", 《IEEE TRANSACTIONS ON CYBERNETICS》 *
XIANG WANG: "Edge Preserving and Multi-Scale Contextual Neural Network for Salient Object Detection", 《JOURNAL OF LATEX CLASS FILES》 *

Cited By (21)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN110210539A (zh) * 2019-05-22 2019-09-06 西安电子科技大学 多级深度特征融合的rgb-t图像显著性目标检测方法
CN110210539B (zh) * 2019-05-22 2022-12-30 西安电子科技大学 多级深度特征融合的rgb-t图像显著性目标检测方法
CN110458797A (zh) * 2019-06-18 2019-11-15 南开大学 一种基于深度图过滤器的显著性物体检测方法
CN110276831A (zh) * 2019-06-28 2019-09-24 Oppo广东移动通信有限公司 三维模型的建构方法和装置、设备、计算机可读存储介质
CN110276831B (zh) * 2019-06-28 2022-03-18 Oppo广东移动通信有限公司 三维模型的建构方法和装置、设备、计算机可读存储介质
CN110472639A (zh) * 2019-08-05 2019-11-19 山东工商学院 一种基于显著性先验信息的目标提取方法
CN110472639B (zh) * 2019-08-05 2023-04-18 山东工商学院 一种基于显著性先验信息的目标提取方法
CN110458178A (zh) * 2019-08-12 2019-11-15 浙江科技学院 多模态多拼接的rgb-d显著性目标检测方法
CN110458178B (zh) * 2019-08-12 2023-09-22 浙江科技学院 多模态多拼接的rgb-d显著性目标检测方法
CN112395922A (zh) * 2019-08-16 2021-02-23 杭州海康威视数字技术股份有限公司 面部动作检测方法、装置及系统
CN111047571A (zh) * 2019-12-10 2020-04-21 安徽大学 一种具有自适应选择训练过程的图像显著目标检测方法
CN111047571B (zh) * 2019-12-10 2023-04-25 安徽大学 一种具有自适应选择训练过程的图像显著目标检测方法
CN111209840A (zh) * 2019-12-31 2020-05-29 浙江大学 一种基于多传感器数据融合的3d目标检测方法
CN111209840B (zh) * 2019-12-31 2022-02-18 浙江大学 一种基于多传感器数据融合的3d目标检测方法
CN111402306A (zh) * 2020-03-13 2020-07-10 中国人民解放军32801部队 一种基于深度学习的微光/红外图像彩色融合方法及系统
CN111428602A (zh) * 2020-03-18 2020-07-17 浙江科技学院 卷积神经网络边缘辅助增强的双目显著性图像检测方法
CN112651406A (zh) * 2020-12-18 2021-04-13 浙江大学 一种深度感知和多模态自动融合的rgb-d显著性目标检测方法
CN112651406B (zh) * 2020-12-18 2022-08-09 浙江大学 一种深度感知和多模态自动融合的rgb-d显著性目标检测方法
CN114419323A (zh) * 2022-03-31 2022-04-29 华东交通大学 基于跨模态学习与领域自适应rgbd图像语义分割方法
CN115496975A (zh) * 2022-08-29 2022-12-20 锋睿领创(珠海)科技有限公司 辅助加权数据融合方法、装置、设备及存储介质
CN115496975B (zh) * 2022-08-29 2023-08-18 锋睿领创(珠海)科技有限公司 辅助加权数据融合方法、装置、设备及存储介质

Also Published As

Publication number Publication date
CN109712105B (zh) 2020-10-27

Similar Documents

Publication Publication Date Title
CN109712105A (zh) 一种结合彩色和深度信息的图像显著目标检测方法
CN109543606B (zh) 一种加入注意力机制的人脸识别方法
CN107832672B (zh) 一种利用姿态信息设计多损失函数的行人重识别方法
CN104850850B (zh) 一种结合形状和颜色的双目立体视觉图像特征提取方法
Huang et al. Indoor depth completion with boundary consistency and self-attention
CN108520226B (zh) 一种基于躯体分解和显著性检测的行人重识别方法
CN104050471B (zh) 一种自然场景文字检测方法及系统
CN104601964B (zh) 非重叠视域跨摄像机室内行人目标跟踪方法及系统
CN111368815A (zh) 一种基于多部件自注意力机制的行人重识别方法
CN111814845B (zh) 一种基于多支流融合模型的行人重识别方法
CN108154102A (zh) 一种道路交通标志识别方法
CN105469111B (zh) 基于改进的mfa和迁移学习的小样本集的物体分类方法
CN102609724B (zh) 一种基于双摄像头的周围环境信息提示方法
CN110827312B (zh) 一种基于协同视觉注意力神经网络的学习方法
CN109635726B (zh) 一种基于对称式深度网络结合多尺度池化的滑坡识别方法
CN108960142B (zh) 基于全局特征损失函数的行人再识别方法
CN108549901A (zh) 一种基于深度学习的快速迭代目标检测方法
CN108073940B (zh) 一种非结构化环境中的3d目标实例物体检测的方法
CN112070010B (zh) 一种联合多损失动态训练策略增强局部特征学习的行人重识别方法
Estrela et al. Sign language recognition using partial least squares and RGB-D information
CN111882620A (zh) 一种基于多尺度信息道路可行驶区域分割方法
Hong et al. USOD10K: a new benchmark dataset for underwater salient object detection
CN111310720A (zh) 基于图度量学习的行人重识别方法及系统
CN110751005A (zh) 融合深度感知特征和核极限学习机的行人检测方法
Liu et al. Multi-cue fusion: Discriminative enhancing for person re-identification

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