一种图像显著性目标检测方法与流程

文档序号:20920319发布日期:2020-05-29 14:03阅读:453来源:国知局
一种图像显著性目标检测方法与流程

本发明的技术方案涉及图像分析的区域分割,具体地说是一种图像显著性目标检测方法。



背景技术:

图像显著性目标检测是指利用计算机模拟人眼的视觉注意力机制,从图像中提取人类感兴趣的区域,是计算机视觉领域的关键技术之一。

图像显著性目标检测的现有技术中,根据所提取图像特征类型的不同,图像显著性目标检测方法分为手工模型和深度学习模型。图像显著性目标检测的手工模型是指根据图像手工特征,例如颜色、纹理、位置和梯度,计算区域的显著值,但低级的图像手工特征无法描述对象的语义信息,在复杂场景中,无法精确描述对象边界;图像显著性目标检测的深度学习模型是指利用卷积神经网络提取图像的高级语义特征计算图像显著值,得到更丰富的深度特征,进而得到更准确的检测结果,但采用深度卷积神经网络会丢失图像显著性目标的位置信息和细节信息,在检测图像显著目标时可能会引入误导信息,导致检测到的图像显著性目标不完整。

2014年,zhu等人在“saliencyoptimizationfromrobustbackgrounddetection”一文中提出的基于边界连通性的背景检测方法,其主要思路是提取图像边界作为背景先验,通过计算每个区域的边界连通性扩大显著目标和背景区域的差异,从而得到更准确的图像显著区域,解决了图像显著目标处于边界时检测不准确的问题,但是当背景区域较复杂时,图像显著性检测目标会缺失。yu等人于2019年在“基于多图流形排序的图像显著性检测”一文中提出的基于多图流形排序的检测方法,其主要思路是在对每个图像构造基于超像素的knn图模型和k正则图模型,在两种模型上利用流形排序算法得到图像显著值,加权融合不同模型结果,得到最终的图像显著值,该方法通过构造不同的图模型得到更准确的显著区域,但在显著目标和背景区域相似或背景区域较复杂时,图像显著性检测结果引入较多的背景噪声。zhao等人于2019年在“pyramidfeatureattentionnetworkforsaliencydetection”一文中提出一种图像显著性目标检测的金字塔特征注意网络方法,该方法采用上下文感知金字塔模块提取多尺度高级特征,然后利用空间注意机制过滤低层空间结构特征,最后将两种特征融合在一起,该方法得到的结果显著目标边缘模糊,复杂环境下检测到的图像显著目标缺失。cn106780582a公开了基于纹理特征和颜色特征融合的图像显著性检测方法,该方法利用颜色特征计算初步对比度,利用纹理特征计算背景概率,融合初步对比度得到新的对比度,即得到图像显著图,其存在是当图像显著目标较大且位于图像边界位置时,图像显著目标往往会被误检测为背景区域的缺陷。cn110188763a公开了基于改进图模型的图像显著性检测方法,该方法在改进图模型的基础上,根据底层特征得到基于底层特征的显著图,利用底层特征和先验知识提取高层特征,并结合显著物体的紧凑性确定阈值,得到前景、背景种子节点,得到基于前景、背景的显著图并融合,最后融合基于底层特征的显著图,得到最终显著图,其存在是前景、背景种子选择部分很难找到一个合适阈值导致基于该种子节点的显著图不准确,导致最终显著图不准确且对噪声敏感的缺陷。cn106780430b公开了基于被包围状态和马尔科夫模型的图像显著性检测方法,该方法利用被包围状态得到显著性物体的大概区域,并将该区域作为前景先验,同时将距离该区域最远的两条边界作为背景,利用吸收马尔可夫链得到基于前景的显著图和基于背景的显著图,并融合两个显著图得到最终显著图,其存在是该方法没有考虑复杂背景的情况,导致显著性检测结果中引入过多的背景噪声的缺陷。cn108898136a公开了跨模态图像显著性检测方法,该方法利用多模态图像和流形排序算法,以图像边界为种子节点,计算节点的背景概率,其存在当显著目标触及图像边界时,显著目标会被误检测为背景的缺陷。cn106952251b公开了基于吸附模型的图像显著性检测方法,该方法利用模糊c均值聚类算法得到候选显著性区域和背景区域,利用基于漏电的积分放电模型的胜者得全网络确定场景中的关注点,使用基于关注点和背景区域的吸附过程筛选候选区域,利用区域的颜色差异和空间距离计算显著性值,通过显著性传播和高斯滤波优化最终结果,其存在是当显著图像较大且位于图像边界时,利用高斯滤波优化后的显著图存在主体缺失的情况的缺陷。cn106373126a公开了基于融合类测地线和边界对比的图像显著性检测方法,该方法通过融合边界对比图、有颜色对比的测地线图和无颜色对比的测地线图得到最终显著图,其存在是该方法在前景背景颜色相似情况下,基于颜色对比度的测地线图中存在过多的背景噪声的缺陷。cn107169954a公开了基于并行卷积神经网络的图像显著性检测方法,该方法设计全局检测模块cnn#g和局部检测模块cnn#l,并通过一个全连接层实现并行,实现对图像进行显著性检测,其存在是没有充分利用cnn模块得到的深度特征,显著目标边缘不精细的缺陷。

总之,图像显著性目标检测的现有技术中仍存在显著目标检测不完整、当前景背景颜色相似时算法检测不准确的缺陷。



技术实现要素:

本发明所要解决的技术问题是:提供一种图像显著性目标检测方法,是基于多图模型先验和短连接网络优化的图像显著性检测方法,该方法是对每张输入图像利用颜色和位置信息,计算knn图模型和k正则图模型,得到在knn图模型下的显著图s1和在k正则图模型下的显著图s2,再将knn图模型和k正则图模型进行像素级别的融合,得到原图像的初始显著图s3,利用短连接网络优化初始显著图s3,获得原图像的最终的显著图sfinal,完成图像显著性目标检测,克服了图像显著性目标检测的现有技术中存在的显著目标检测不完整、当前景背景颜色相似时算法检测不准确的缺陷。

本发明解决该技术问题所采用的技术方案是:一种图像显著性目标检测方法,是基于多图模型先验和短连接网络优化的图像显著性检测方法,具体步骤如下:

第一步,输入原图像i,进行预处理:

输入原图像i,使用slic超像素区域分割算法进行预分割,得到超像素集sp={spi,i=1,2,...,n},其中,i为超像素的排序数(下同),spi为原图像i的第i个超像素,n为图像i的预分割超像素的个数(下同),对每个超像素区域提取平均cie-lab颜色空间特征和空间位置特征其中,ci为原图像i的第i个超像素的颜色特征(下同),pi为原图像i的第i个超像素的空间位置特征(下同),至此完成输入原图像i的预处理;

第二步,获得原图像i在knn图模型下的显著图s1:

第(2.1)步,对原图像i构造knn图模型:

用如下公式(1)计算上述第一步中的原图像i的第i个超像素spi和原图像i的第j个超像素spj之间的颜色特征的欧式距离dist(ci,cj),

dist(ci,cj)=||ci,cj||2(1),

公式(1)中,cj为原图像i的第j个超像素的颜色特征,

用如下公式(2)计算上述第一步中的原图像i的第i个超像素spi和原图像i的第j个超像素spj之间的空间位置距离dist(pi,pj),

公式(2)中,pj为原图像i的第j个超像素的空间位置特征,

用如下公式(3)计算上述第一步中的原图像i的第i个超像素spi和原图像i的第j个超像素spj之间的特征距离dist(spi,spj),

dist(spi,spj)=λp×dist(pi,pj)+λc×dist(ci,cj)(3),

公式(3)中,λp为用来控制dist(pi,pj)对特征距离dist(spi,spj)的权重值,λc为用来控制dist(ci,cj)对特征距离dist(spi,spj)的权重值,设定为λp=0.5,λc=1,

对特征距离dist(spi,spj)按照从小到大顺序排序,选择前5个值,其对应的超像素设置为近邻超像素,然后对原图像i构造knn图模型为g1=<v1,e1>,其中v1为节点集合,每个超像素为一个节点,e1为边集合,e1=[e1,ij]n×n;

第(2.2)步,确定内边缘权重和外边缘权重:

根据上述第(2.1)中提及的超像素的颜色特征和空间位置特征,为每个超像素选择5个近邻超像素,每个超像素不仅连接它的相邻超像素,还和与它的相邻超像素共享一个超像素边界的超像素相连,

当一个超像素在图像内部时,与它的5个近邻超像素构成的边称为内边缘e1,bondrary_i,内边缘权重由权重矩阵w1=[w1]n×n定义,如下公式(4)所示,

w1,boundary_i(e1,ij)=exp(-dist(ci,cj)/2σ2)(4),

公式(4)中,σ为一个常数(下同),设定σ=10,

当一个超像素在图像边界时,与它的5个邻接节点构成的边称为外边缘e1,bondrary_o,外边缘权重定义如下公式(5)所示,

w1,boundary_o(e1,ij)=exp(-dist(pi,pj)/2σ2)(5),

公式(5)中,设定σ=10;

第(2.3)步,筛选边界背景节点:

利用边界连通性筛选图像i的边界背景节点,具体步骤如下:

设定原图像i的四条边界超像素为背景集合,记为bnd,使用如下公式(6)计算原图像i的超像素sp与背景边界的连通程度,即原图像i超像素spi的边界连通性值,

公式(6)中,lenbnd(spi)为原图像i超像素spi的边界周长,area(spi)为原图像i超像素spi的区域面积,

lenbnd(sp)的定义如下公式(7)所示,

公式(7)中,dist(spi,spj)为原图像i超像素spi与原图像i超像素spj之间的特征距离,当原图像i超像素spj被归为背景集合bnd时,δ(spj∈bnd)=1,否则为0,

area(sp)的定义如下公式(8)所示,

公式(8)中,σclr为常数,σclr=10,dgeo(spi,spj)为原图像i第i个超像素spi与原图像i第j个超像素spj之间沿knn图模型最短路径的边缘权重的测地线距离,

dgeo(spi,spj)的定义如下公式(9)所示,

公式(9)中,dapp(spl,spl+1)为原图像i超像素spl和原图像i超像素spl+1在cie-lab颜色空间中的欧氏距离,p1为超像素spi,p2为原图像i第i个超像素spi到原图像i第j个超像素spj中可以连接到的所有超像素、pn为超像素spj,

然后,根据原图像i超像素sp的边界连通性值bndcon(·),利用k-means方法将背景集合bnd聚为两类,计算每个聚类中原图像i超像素的边界连通性值的平均值,选择边界连通性平均值最大的一类背景集合bnd作为背景节点,

由此完成筛选边界背景节点;

第(2.4)步,获得原图像i在knn图模型下的显著图s1:

设置上述第(2.3)步筛选得到的边界背景节点为查询节点,利用流形排序算法计算超像素与边界背景节点的相关性,并进行排序,得到原图像i在knn图模型下的显著图s1,计算公式如下公式(10)所示,

s1=(d-αw1)-1y(10),

公式(10)中,d为原图像i对应的度矩阵,w1为权重矩阵,y为指示向量,

当超像素spi是查询节点时,则yi=1,否则yi=0,

原图像i对应的度矩阵d的定义如下公式(11)所示,

公式(11)中,dii为度矩阵d中的元素,w1,ij为knn图模型下超像素spi和超像素spj之间的边缘权重值,

由此获得原图像i在knn图模型下的显著图s1;

第三步:获得原图像i在k正则图模型下的显著图s2:

定义超像素的空间接近度和外观相似度,结合影响因子、空间接近度和外观相似度计算图像显著值,具体操作如下:

第(3.1)步,对上述第一步中的原图像i构造k正则图模型:

对上述第一步的原图像i,以超像素为节点构造其k正则图模型为g2=<v2,e2>,其中,v2为节点集合,e2为边集合,e2=[e2,ij]n×n;

第(3.2)步,获得每个超像素的影响因子inf(spi):

每个超像素节点不仅与它的相邻节点相连,而且还和与它共享同一超像素边界的超像素节点相连,由此定义边缘权重矩阵w2=[w2,ij]n×n如下公式(12)所示,

公式(12)中,设定σ=10,e2,ij为k正则图模型下超像素spi和超像素spj之间的边缘(下同),

然后,根据上述定义边缘权重矩阵计算超像素spi受到超像素spj的影响即spt(spi,spj),如下公式(13)所示,

公式(13)中,w2,ij为k正则图模型下超像素spi和超像素spj之间的边缘权重值,w2,it为原图像i超像素spi与原图像i超像素spt之间的边缘权重,

进而由如下公式(14)计算每个超像素的影响因子inf(spi),

由此获得每个超像素的影响因子inf(spi);

第(3.3)步,获得超像素之间的外观相似性distc(spi,spj):

超像素之间的外观相似性distc(spi,spj)由如下公式(15)计算而获得,

公式(15)中,为超像素spi对应的颜色直方图,为超像素spj对应的颜色直方图,由此获得超像素之间的外观相似性distc(spi,spj);

第(3.4)步,获得超像素之间的空间接近度distp(spi,spj):

超像素之间的空间接近度distp(spi,spj)由如下公式(16)计算而获得,

由此获得超像素之间的空间接近度distp(spi,spj)

第(3.5)步,获得原图像i在k正则图模型下的显著图s2:

根据上述第(3.3)步所计算超像素之间的外观相似性和上述第(3.4)步所计算超像素之间的空间接近度,用如下公式(17)计算而获得原图像i在k正则图模型下的显著图s2,

由此获得原图像i在k正则图模型下的显著图s2;

第四步:获得原图像i的初始显著图s3:

对上述第二步获得的原图像i在knn图模型下的显著图s1和第三步获得原图像i在k正则图模型下的显著图s2进行自适应融合,具体操作如下所示,

通过公式(18)计算显著图s1和显著图s2的融合权重ε,

ε=max(s1)-median(s2)(18),

公式(18)中,max(s1)为显著图s1中的最大显著值对应的超像素的显著值,median(s2)为显著图s2中的显著值中值对应的超像素的显著值;

根据如下自适应融合公式(19)计算而得到融合后的原图像i的初始显著图s3,

s3=ρ1((1-ε)s1+εs2)+ρ2(s1·s2)(19),

公式(19)中,ρ1和ρ2为控制该公式中前一项((1-ε)s1+εs2)和后一项(s1·s2)融合时的参数,设定为ρ1=ρ2=0.5,

由此获得原图像i的初始显著图s3;

第五步:获得原图像i的最终的显著图sfinal:

通过短连接网络优化上述第四步得到的原图像i的初始显著图s3,即将上述第四步得到的原图像i的初始显著图s3作为先验知识映射到显著性先验中,与原图i一起输入短连接网络,输出的结果为最终显著图sfinal,具体操作如下:

第(5.1)步,提取原图像i的深度特征:

选用vgg16卷积部分提取原图像i的深度特征,即用如下公式(20)提取原图像i的深度特征f(i),

f(i)=ω1*hi*i+ω2*hs*s3+b(20),

公式(20)中,hi为原图像i的卷积核,hs为初始显著图s3的卷积核,ω1为原图像i卷积的融合权重,ω2为初始显著图s3卷积的融合权重,b为偏置量,

第(5.2)步,得到侧输出层的深度特征图:

在vgg16卷积部分中的卷积层conv1_2、conv2_2、conv3_2、conv4_2和conv5_2后面添加使用卷积核通道和空间大小不同的卷积层的侧输出层,使用双线性插值法在这些侧输出层上采样,得到侧输出层的深度特征图,分别为第1侧输出层深度特征图,第2侧输出层深度特征图,第3侧输出层深度特征图,第4侧输出层深度特征图,第5侧输出层深度特征图,

第(5.3)步,将各侧输出层深度特征图进行连接:

利用短连接方式将上述第(5.2)步得到的各侧输出层深度特征图进行连接,具体操作如下:

对第5侧输出层深度特征图引入一个1×1的卷积层作为第5侧输出层深度特征图的预测图z5,

然后,对第4侧输出层深度特征图引入一个1×1的卷积层,然后与第5侧输出层深度特征图的预测图z5进行连接,得到第4侧输出层深度特征图的预测图z4,

然后,对第3侧输出层深度特征图引入一个1×1的卷积层,然后与第4侧输出层深度特征图的预测图z4和第5侧输出层深度特征图的预测图z5进行连接,得到第3侧输出层深度特征图的预测图z3,

然后,对第2侧输出层深度特征图引入一个1×1的卷积层,然后与第3侧输出层深度特征图的预测图z3、第4侧输出层深度特征图的预测图z4和第5侧输出层深度特征图的预测图z5进行连接,得到第2侧输出层深度特征图的预测图z2,

然后,对第1侧输出层深度特征图引入一个1×1的卷积层,然后与第2侧输出层深度特征图的预测图z2、第3侧输出层深度特征图的预测图z3、第4侧输出层深度特征图的预测图z4和第5侧输出层深度特征图的预测图z5进行连接,得到第1侧输出层深度特征图的预测图z1,

然后,用下公式(21)融合预测图z5、预测图z4、预测图z3、预测图z2和预测图z1,得到所有侧输出层深度特征图的融合预测图zfusion,

公式(21)中,fm为各侧输出层深度特征图所对应的融合权重,zm为第m个侧输出层深度特征图,

第(5.4)步,获得最终输出预测图zfinal:

在上述第(5.3)步得到的所有侧输出层深度特征图的融合预测图zfusion中添加预测图z2、预测图z3、预测图z4,以补充原图像i的细节特征,得到所有侧输出层深度特征图的最终输出预测图zfinal如下公式(22)所示,

zfinal=mean(zfusion,z2,z3,z4)(22),

公式(22)中,mean(·)为均值,

上述所有侧输出层深度特征图的最终输出预测图zfinal即为原图像i的最终显著图sfinal,

由此获得原图像i的最终的显著图sfinal;

至此,完成基于多图模型先验和短连接网络优化的图像显著性检测。

上述一种图像显著性目标检测方法,所述“slic超像素区域分割算法”、“knn图模型”、“k近邻节点”、“k正则图模型”、“流形排序算法”、“cie-lab颜色空间”、“k-means方法”、“双线性插值”、“短连接网络”都是本技术领域的公知技术。

本发明的有益效果是:与现有技术相比,本发明的突出的实质性特点和显著性进步如下:

(1)与cn106780582a一种基于纹理特征和颜色特征融合的图像显著性检测方法相比,本发明具有的突出的实质性特点和显著进步是:cn106780582a在利用颜色特征和纹理特征计算相似度和背景概率时使用同一种图结构,本发明针对颜色特征构造knn图模型,针对空间位置特征构造k正则图模型,分别在这两个模型上得到不同的显著图,通过自适应融合两个不同图结构,本发明的所具有的优点是获得的显著性图中的显著目标一致高亮。

(2)与cn110188763a一种基于改进图模型的图像显著性检测方法相比,本发明具有的突出的实质性特点和显著进步是:cn110188763a利用底层特征和多种先验知识得到高层特征,仅利用底层特征没有充分利用图像中的语义信息,得到的检测结果不能得到完整的图像显著目标,本发明通过神经网络提取高级特征,得到更加丰富的语义信息,最终得到的显著目标更加准确。

(3)与cn106780430b一种基于被包围状态和马尔科夫模型的图像显著性检测方法相比,本发明具有的突出的实质性特点和显著进步是:cn106780430b利用被包围状态得到显著性物体的大概区域,将距离该区域最远的两条边界作为背景,此方法适用图像显著目标触及图像边界的情况,当显著图目标处于图像中心且背景复杂的情况下无法准确的检测图像显著目标,本发明对边界超像素聚类,利用边界连通性剔除边界中属于前景的超像素,将剩余边界超像素作为背景种子节点,克服了当前景目标触及图像边界时导致的显著性目标检测存在缺失的缺陷。

(4)与cn108898136a一种跨模态的图像显著性检测方法相比,本发明具有的突出的实质性特点和显著进步是:cn108898136a在利用流形排序算法计算显著性值时,直接将图像边界作为背景种子节点,图像边界中会存在一定的前景,直接将图像边界作为背景种子节点得到的显著图得到的图像显著目标缺失,本发明利用边界连通性剔除边界中的前景目标超像素,将优化后的边界超像素作为种子节点,克服了显著目标位于边界时所导致的显著目标检测存在缺失的缺陷。

(5)与cn106952251b一种基于吸附模型的图像显著性检测方法相比,本发明具有的突出的实质性特点和显著进步是:cn106952251b利用模糊c均值聚类算法得到显著区域和背景区域,利用区域的颜色差异和空间距离计算显著值,当处理低对比度图像时利用聚类方法得到显著区域和背景区域不准确,本发明通过超像素的颜色特征计算超像素的影响因子,同时结合外观相似性和空间接近度得到基于k正则图的显著值,明显减少了k正则图模型中的背景噪声。

(6)与本发明人团队早先的cn106373126a基于融合类测地线和边界对比的图像显著性检测方法相比,本发明具有的突出的实质性特点和显著进步是:首先,cn106373126a只将图像边界区域作为背景,当显著目标位于边界时,往往会被检测为背景,本发明方法对边界超像素节点进行改进,利用边界连通性剔除边界中的前景目标超像素,将优化后的边界超像素作为种子节点,本发明的优点是克服了显著目标位于边界时所导致的显著目标检测存在缺失的缺陷;其次,cn106373126a只是构造了k正则图模型结构,本发明方法对图像构造knn图模型和k正则图模型,从不同角度刻画节点之间的连接性,使获得的显著性图中显著目标一致高亮;cn106373126a还存在采用直接相乘的方式融合边界对比图、有颜色对比度地测地线图和无颜色对比地测地线图,最终的显著性检测效果往往不能一致高亮显著目标的缺陷,而本发明方法通过自适应方法融合不同图模型检测结果,融合权重是根据两个模型的检测结果共同决定,本发明的优点是通过融合knn图模型的检测结果和k正则图模型的检测结果,避免当前景目标和背景颜色相似时所导致的显著性目标检测缺失的情况。

总之,在cn106373126a的基础上结合本领域的公知常识要得出本发明的技术方案,对本领域技术人员来说绝非显而易见的。

(7)本发明方法将knn图模型和k正则图模型进行融合,对融合后结果作为先验输入深度网络进行优化,克服了单一图模型下生成的显著图中噪声较多和显著目标不完整的问题。

(8)本发明方法利用边界连通性和k-means方法优化图像边界超像素,剔除边界中的显著目标,克服了显著目标触及图像边界时,直接利用图像边界区域作为背景而导致的部分前景目标误当作背景的缺陷。

(9)本发明方法利用图结构中节点与节点之间的边权重,定义影响因子,结合超像素之间的外观相似性和空间接近度,克服了现有技术中图内显著性线索利用不够充分,导致显著性检测结果中包含过多的背景噪声和目标缺失的缺陷。

(10)本发明将传统模型检测结果为先验映射并以此作为网络输入,并通过一个短连接网络得到最终预测图,实现了传统模型与深度模型的融合,克服了在复杂背景下显著目标检测存在的目标缺失情况的缺陷。

附图说明

下面结合附图和实施例对本发明进一步说明。

图1是本发明方法的流程示意框图。

图2是原图像i的显著性图sfinal。

具体实施方式

图1所示实施例表明,本发明方法的流程如下:

输入图像i,进行预处理→获得原图像i在knn图模型下的显著图s1→获得原图像i在k正则图模型下的显著图s2→获得原图像i的初始显著图s3→获得最终的显著图sfinal→完成基于多图模型先验和短连接网络优化的图像显著性检测。

图2显示原图像i一个女人和一头大象的最终的显著性图sfinal,图中黑色线条是原图像中显著目标区域与非显著区域的分界结果,边界线内部是检测到的图像显著目标,边界线外部是图像背景。

实施例

本实施例中图像显著性目标为一个女人和一头大象,本实施例基于多图模型先验和短连接网络优化的图像显著性检测方法,具体步骤如下:

第一步,输入原图像i,进行预处理:

输入原图像i,使用slic超像素区域分割算法进行预分割,得到超像素集sp={spi,i=1,2,...,n},其中,i为超像素的排序数(下同),spi为原图像i的第i个超像素,n为图像i的预分割超像素的个数(下同),对每个超像素区域提取平均cie-lab颜色空间特征和空间位置特征其中,ci为原图像i的第i个超像素的颜色特征(下同),pi为原图像i的第i个超像素的空间位置特征(下同),至此完成输入原图像i的预处理;

第二步,获得原图像i在knn图模型下的显著图s1:

第(2.1)步,对原图像i构造knn图模型:

用如下公式(1)计算上述第一步中的原图像i的第i个超像素spi和原图像i的第j个超像素spj之间的颜色特征的欧式距离dist(ci,cj),

dist(ci,cj)=||ci,cj||2(1),

公式(1)中,cj为原图像i的第j个超像素的颜色特征,

用如下公式(2)计算上述第一步中的原图像i的第i个超像素spi和原图像i的第j个超像素spj之间的空间位置距离dist(pi,pj),

公式(2)中,pj为原图像i的第j个超像素的空间位置特征,

用如下公式(3)计算上述第一步中的原图像i的第i个超像素spi和原图像i的第j个超像素spj之间的特征距离dist(spi,spj),

dist(spi,spj)=λp×dist(pi,pj)+λc×dist(ci,cj)(3),

公式(3)中,λp为用来控制dist(pi,pj)对特征距离dist(spi,spj)的权重值,λc为用来控制dist(ci,cj)对特征距离dist(spi,spj)的权重值,设定为λp=0.5,λc=1,

对特征距离dist(spi,spj)按照从小到大顺序排序,选择前5个值,其对应的超像素设置为近邻超像素,然后对原图像i构造knn图模型为g1=<v1,e1>,其中v1为节点集合,每个超像素为一个节点,e1为边集合,e1=[e1,ij]n×n;

第(2.2)步,确定内边缘权重和外边缘权重:

根据上述第(2.1)中提及的超像素的颜色特征和空间位置特征,为每个超像素选择5个近邻超像素,每个超像素不仅连接它的相邻超像素,还和与它的相邻超像素共享一个超像素边界的超像素相连,

当一个超像素在图像内部时,与它的5个近邻超像素构成的边称为内边缘e1,bondrary_i,内边缘权重由权重矩阵w1=[w1]n×n定义,如下公式(4)所示,

w1,boundary_i(e1,ij)=exp(-dist(ci,cj)/2σ2)(4),

公式(4)中,σ为一个常数(下同),设定σ=10,

当一个超像素在图像边界时,与它的5个邻接节点构成的边称为外边缘e1,bondrary_o,外边缘权重定义如下公式(5)所示,

w1,boundary_o(e1,ij)=exp(-dist(pi,pj)/2σ2)(5),

公式(5)中,设定σ=10;

第(2.3)步,筛选边界背景节点:

利用边界连通性筛选图像i的边界背景节点,具体步骤如下:

设定原图像i的四条边界超像素为背景集合,记为bnd,使用如下公式(6)计算原图像i的超像素sp与背景边界的连通程度,即原图像i超像素spi的边界连通性值,

公式(6)中,lenbnd(spi)为原图像i超像素spi的边界周长,area(spi)为原图像i超像素spi的区域面积,

lenbnd(sp)的定义如下公式(7)所示,

公式(7)中,dist(spi,spj)为原图像i超像素spi与原图像i超像素spj之间的特征距离,当原图像i超像素spj被归为背景集合bnd时,δ(spj∈bnd)=1,否则为0,area(sp)的定义如下公式(8)所示,

公式(8)中,σclr为常数,σclr=10,dgeo(spi,spj)为原图像i第i个超像素spi与原图像i第j个超像素spj之间沿knn图模型最短路径的边缘权重的测地线距离,

dgeo(spi,spj)的定义如下公式(9)所示,

公式(9)中,dapp(spl,spl+1)为原图像i超像素spl和原图像i超像素spl+1在cie-lab颜色空间中的欧氏距离,p1为超像素spi,p2为原图像i第i个超像素spi到原图像i第j个超像素spj中可以连接到的所有超像素、pn为超像素spj,

然后,根据原图像i超像素sp的边界连通性值bndcon(·),利用k-means方法将背景集合bnd聚为两类,计算每个聚类中原图像i超像素的边界连通性值的平均值,选择边界连通性平均值最大的一类背景集合bnd作为背景节点,

由此完成筛选边界背景节点;

第(2.4)步,获得原图像i在knn图模型下的显著图s1:

设置上述第(2.3)步筛选得到的边界背景节点为查询节点,利用流形排序算法计算超像素与边界背景节点的相关性,并进行排序,得到原图像i在knn图模型下的显著图s1,计算公式如下公式(10)所示,

s1=(d-αw1)-1y(10),

公式(10)中,d为原图像i对应的度矩阵,w1为权重矩阵,y为指示向量,

当超像素spi是查询节点时,则yi=1,否则yi=0,

原图像i对应的度矩阵d的定义如下公式(11)所示,

公式(11)中,dii为度矩阵d中的元素,w1,ij为knn图模型下超像素spi和超像素spj之间的边缘权重值,

由此获得原图像i在knn图模型下的显著图s1;

第三步:获得原图像i在k正则图模型下的显著图s2:

定义超像素的空间接近度和外观相似度,结合影响因子、空间接近度和外观相似度计算图像显著值,具体操作如下:

第(3.1)步,对上述第一步中的原图像i构造k正则图模型:

对上述第一步的原图像i,以超像素为节点构造其k正则图模型为g2=〈v2,e2〉,其中,v2为节点集合,e2为边集合,e2=[e2,ij]n×n;

第(3.2)步,获得每个超像素的影响因子inf(spi):

每个超像素节点不仅与它的相邻节点相连,而且还和与它共享同一超像素边界的超像素节点相连,由此定义边缘权重矩阵w2=[w2,ij]n×n如下公式(12)所示,

公式(12)中,设定σ=10,e2,ij为k正则图模型下超像素spi和超像素spj之间的边缘(下同),

然后,根据上述定义边缘权重矩阵计算超像素spi受到超像素spj的影响即spt(spi,spj),如下公式(13)所示,

公式(13)中,w2,ij为k正则图模型下超像素spi和超像素spj之间的边缘权重值,w2,it为原图像i超像素spi与原图像i超像素spt之间的边缘权重,

进而由如下公式(14)计算每个超像素的影响因子inf(spi),

由此获得每个超像素的影响因子inf(spi);

第(3.3)步,获得超像素之间的外观相似性distc(spi,spj):

超像素之间的外观相似性distc(spi,spj)由如下公式(15)计算而获得,

公式(15)中,为超像素spi对应的颜色直方图,为超像素spj对应的颜色直方图,由此获得超像素之间的外观相似性distc(spi,spj);

第(3.4)步,获得超像素之间的空间接近度distp(spi,spj):

超像素之间的空间接近度distp(spi,spj)由如下公式(16)计算而获得,

由此获得超像素之间的空间接近度distp(spi,spj)

第(3.5)步,获得原图像i在k正则图模型下的显著图s2:

根据上述第(3.3)步所计算超像素之间的外观相似性和上述第(3.4)步所计算超像素之间的空间接近度,用如下公式(17)计算而获得原图像i在k正则图模型下的显著图s2,

由此获得原图像i在k正则图模型下的显著图s2;

第四步:获得原图像i的初始显著图s3:

对上述第二步获得的原图像i在knn图模型下的显著图s1和第三步获得原图像i在k正则图模型下的显著图s2进行自适应融合,具体操作如下所示,

通过公式(18)计算显著图s1和显著图s2的融合权重ε,

ε=max(s1)-median(s2)(18),

公式(18)中,max(s1)为显著图s1中的最大显著值对应的超像素的显著值,median(s2)为显著图s2中的显著值中值对应的超像素的显著值;

根据如下自适应融合公式(19)计算而得到融合后的原图像i的初始显著图s3,

s3=ρ1((1-ε)s1+εs2)+ρ2(s1·s2)(19),

公式(19)中,ρ1和ρ2为控制该公式中前一项((1-ε)s1+εs2)和后一项(s1·s2)融合时的参数,设定为ρ1=ρ2=0.5,

由此获得原图像i的初始显著图s3;

第五步:获得最终的显著图sfinal:

通过短连接网络优化上述第四步得到的初始显著图s3,即将上述第四步得到的原图像i的初始显著图s3作为先验知识映射到显著性先验中,与原图i一起输入短连接网络,输出的结果为最终显著图sfinal,具体操作如下:

第(5.1)步,提取原图像i的深度特征:

选用vgg16卷积部分提取原图像i的深度特征,即用如下公式(20)提取原图像i的深度特征f(i),

f(i)=ω1*hi*i+ω2*hs*s3+b(20),

公式(20)中,hi为原图像i的卷积核,hs为初始显著图s3的卷积核,ω1为原图像i卷积的融合权重,ω2为初始显著图s3卷积的融合权重,b为偏置量,

第(5.2)步,得到侧输出层的深度特征图:

在vgg16卷积部分中的卷积层conv1_2、conv2_2、conv3_2、conv4_2和conv5_2后面添加使用卷积核通道和空间大小不同的卷积层的侧输出层,使用双线性插值法在这些侧输出层上采样,得到侧输出层的深度特征图,分别为第1侧输出层深度特征图,第2侧输出层深度特征图,第3侧输出层深度特征图,第4侧输出层深度特征图,第5侧输出层深度特征图,

第(5.3)步,将各侧输出层深度特征图进行连接:

利用短连接方式将上述第(5.2)步得到的各侧输出层深度特征图进行连接,具体操作如下:

对第5侧输出层深度特征图引入一个1×1的卷积层作为第5侧输出层深度特征图的预测图z5,

然后,对第4侧输出层深度特征图引入一个1×1的卷积层,然后与第5侧输出层深度特征图的预测图z5进行连接,得到第4侧输出层深度特征图的预测图z4,

然后,对第3侧输出层深度特征图引入一个1×1的卷积层,然后与第4侧输出层深度特征图的预测图z4和第5侧输出层深度特征图的预测图z5进行连接,得到第3侧输出层深度特征图的预测图z3,

然后,对第2侧输出层深度特征图引入一个1×1的卷积层,然后与第3侧输出层深度特征图的预测图z3、第4侧输出层深度特征图的预测图z4和第5侧输出层深度特征图的预测图z5进行连接,得到第2侧输出层深度特征图的预测图z2,

然后,对第1侧输出层深度特征图引入一个1×1的卷积层,然后与第2侧输出层深度特征图的预测图z2、第3侧输出层深度特征图的预测图z3、第4侧输出层深度特征图的预测图z4和第5侧输出层深度特征图的预测图z5进行连接,得到第1侧输出层深度特征图的预测图z1,

然后,用下公式(21)融合预测图z5、预测图z4、预测图z3、预测图z2和预测图z1,得到所有侧输出层深度特征图的融合预测图zfusion,

公式(21)中,fm为各侧输出层深度特征图所对应的融合权重,zm为第m个侧输出层深度特征图,

第(5.4)步,获得最终输出预测图zfinal:

为了减少背景噪声,提高最终预测结果的准确率,在上述第(5.3)步得到的所有侧输出层深度特征图的融合预测图zfusion中添加预测图z2、预测图z3、预测图z4,以补充原图像i的细节特征,得到所有侧输出层深度特征图的最终输出预测图zfinal如下公式(22)所示,

zfinal=mean(zfusion,z2,z3,z4)(22),

公式(22)中,mean(·)为均值,

上述所有侧输出层深度特征图的最终输出预测图zfinal即为原图像i的最终显著图sfinal,

由此获得最终的显著图sfinal;

至此,完成基于多图模型先验和短连接网络优化的图像显著性检测。

上述实施例中,所述“slic超像素区域分割算法”、“knn图模型”、“k近邻节点”、“k正则图模型”、“流形排序算法”、“cie-lab颜色空间”、“k-means方法”、“双线性插值”、“短连接网络”都是本技术领域的公知技术。

当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1