CN113468996A - 一种基于边缘细化的伪装物体检测方法 - Google Patents

一种基于边缘细化的伪装物体检测方法 Download PDF

Info

Publication number
CN113468996A
CN113468996A CN202110691277.3A CN202110691277A CN113468996A CN 113468996 A CN113468996 A CN 113468996A CN 202110691277 A CN202110691277 A CN 202110691277A CN 113468996 A CN113468996 A CN 113468996A
Authority
CN
China
Prior art keywords
camouflage
feature
features
object detection
module
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
CN202110691277.3A
Other languages
English (en)
Other versions
CN113468996B (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.)
Guangzhou University
Original Assignee
Guangzhou University
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 Guangzhou University filed Critical Guangzhou University
Priority to CN202110691277.3A priority Critical patent/CN113468996B/zh
Publication of CN113468996A publication Critical patent/CN113468996A/zh
Application granted granted Critical
Publication of CN113468996B publication Critical patent/CN113468996B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/213Feature extraction, e.g. by transforming the feature space; Summarisation; Mappings, e.g. subspace methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/214Generating training patterns; Bootstrap methods, e.g. bagging or boosting
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/24Classification techniques
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/25Fusion techniques
    • G06F18/253Fusion techniques of extracted features
    • YGENERAL 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
    • Y02TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
    • Y02TCLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
    • Y02T10/00Road transport of goods or passengers
    • Y02T10/10Internal combustion engine [ICE] based vehicles
    • Y02T10/40Engine management systems

Abstract

本发明公开了一种基于边缘细化的伪装物体检测方法,首先,利用各类伪装物体的原始图像来构建伪装物体图像数据集,然后构建伪装物体检测网络,接着使用伪装物体图像数据集对构建的伪装物体检测网络进行迭代训练,该伪装物体检测网络将伪装物体检测任务分成两个阶段,第一阶段负责伪装物体检测,第二阶段负责细化伪装物体边缘;最后,以待预测的伪装物体原始图像作为训练完成后的伪装物体检测网络模型的输入,利用该伪装物体检测网络模型输出对应的有区分出图像背景和伪装物体的伪装图。本发明不仅能准确定位伪装物体,而且能保证伪装物体边缘的清晰,有效提高了伪装物体图像整体的检测精度。

Description

一种基于边缘细化的伪装物体检测方法
技术领域
本发明涉及伪装物体检测技术领域,特别是一种基于边缘细化的伪装物体检测方法。
背景技术
伪装是指物体根据环境的纹理,通过使自身的形态、质地和颜色等接近周围的环境来达到隐藏自己的目的。伪装物体的视觉特征与背景非常相似,它的强度或颜色接近其周围环境、纹理被破坏以与背景融合且边界模糊,这也使得伪装的物体不能被人类视觉系统充分地看见。随着人类社会的不断进步,伪装物体检测逐渐走进人们的生活中,它在保护野生动物、战场上探测敌人、医学图像分析等上都存在潜在应用。伪装物体检测就是从背景中提取目标,它能够区分前景物体和伪装物体。
近年来,基于各种视觉特征(例如,强度或颜色、纹理、梯度等)开发了各种算法,以实现在周围环境中检测伪装物体。强度或颜色特征可以检测出与环境背景纹理相似的伪装物体,但却无法检测到与环境背景颜色相似的伪装物体;而当物体的颜色与环境背景颜色相似时,纹理则被认为是区分物体与其周围环境的特征。如果物体和环境背景都包含相似的颜色和纹理,那么梯度信息则有助于从背景区域提取伪装目标。但是,所有特征,例如颜色、纹理和渐变等,都是手工制作的,可能不适用于所有类型的伪装物体。由于这种视觉特征的复杂性,这给伪装物体检测的突破带来了很大的阻碍,在这种背景下,基于深度学习的方法被提出来分析伪装物体。
在大多数情况下,人类仍然能够找到伪装对象,这是由于人类大脑中的语义信息有助于人类发现伪装对象,即使是在对象的底层特征被破坏的情况下。受人类视觉系统的启发,利用卷积神经网络从广泛的训练图像中学习得到的特征来检测伪装物体,这种学习到的深层特征与手工制作的特征相比更具普适性,整体检测的效果会更好。但是由于卷积神经网络的复杂性以及伪装物体的特殊性,现有的方法都有一定的局限性,比如现有的网络框架往往会存在边缘模糊、目标丢失等问题,这会给检测精度带来很大的影响,尤其是在医学影像中检测病变区域以及在军事领域检测敌人时发生的检测失误可能会产生不可估计的后果,因此伪装检测领域还存在巨大的进步空间。
发明内容
本发明的目的在于解决现有技术的不足,提出一种基于边缘细化的伪装物体检测方法,可以在准确定位伪装物体的情况下,保证边缘的清晰,提高整体的检测精度,提供应用于实际场景的更多可能性,给相关工作人员提供更可靠的帮助。
本发明的目的通过下述技术方案实现:
一种基于边缘细化的伪装物体检测方法,包括如下步骤:
S1、利用各类伪装物体的原始图像来构建伪装物体图像数据集;
S2、构建伪装物体检测网络;
S3、训练模型:使用伪装物体图像数据集对构建的伪装物体检测网络进行迭代训练,该伪装物体检测网络将伪装物体检测任务分成两个阶段,第一阶段负责伪装物体检测,第二阶段负责细化伪装物体边缘,训练完成后得到伪装物体检测模型;
S4、测试模型:以待预测的伪装物体原始图像作为训练完成后的伪装物体检测模型的输入,利用该伪装物体检测模型输出对应的有区分出图像背景和伪装物体的伪装图。
优选的,伪装物体检测网络包括特征提取模块、感受野模块、特征编码模块、特征解码模块以及边缘细化模块,其中,特征提取模块、感受野模块、特征编码模块、特征解码模块和边缘细化模块依次连接,感受野模块连接至边缘细化模块;
对于第一阶段:特征提取模块以原始图像作为输入,输出对应的一组特征,该组特征经感受野扩大搜索范围,再经特征编码模块和特征解码模块进行融合,特征解码模块输出初始的伪装图;
对于第二阶段:特征提取模块所获取到的特征中的较低层特征经感受野扩大搜索范围后,和初始伪装图同时作为边缘细化模块的输入,边缘细化模块输出最终的伪装图。
更进一步的,特征提取模块的处理过程如下:
对于输入图像I∈RH×W×3,利用ResNet50网络从输入图像提取出一组多尺度特征{x1,x2,x3,x4},该组多尺度特征中的低级特征保留了用于构建物体边缘的空间信息,高级特征保留了用于定位目标的语义信息;
然后,将多尺度特征按层次划分为低级特征组L:{x1,x2,x3}和高级特征组H:{x2,x3,x4},其中,x1为低级特征,x2,x3为中级特征,x4为高级特征。
更进一步的,多尺度特征{x1,x2,x3,x4}中的每层特征分别经感受野模块扩大感受野的范围,过程具体如下:
感受野模块含有多个分支,对于前4个分支,每个分支首先应用1×1卷积来减小输入特征的通道维数,第k(k∈{2,3,4})个分支再应用核大小为2k-1、膨胀率为1的卷积运算,接着再应用另一个不同应用核大小、膨胀率为2k-1的卷积运算;
将该4个分支的输出拼接在一起,然后应用1×1卷积运算来减小拼接后的特征的通道数,再将处理后的特征与原输入特征形成一个残差连接,最终得到的特征表示为:
Figure BDA0003126261160000041
其中,j∈{1,2,3,4},C∈{c,e},c表示伪装物体检测阶段,e表示细化伪装物体边缘阶段;FRF(·)表示扩大感受野操作;
对于低级特征组L:{x1,x2,x3}中的每层特征,按照上述过程完成扩大感受野操作,最终得到特征:
Figure BDA0003126261160000042
同理,对于高级特征组H:{x2,x3,x4}中的每层特征,按照上述过程完成扩大感受野操作,最终得到特征:
Figure BDA0003126261160000043
更进一步的,特征编码模块的处理过程如下:
首先,对扩大感受野范围的低层特征组
Figure BDA0003126261160000044
进行双线性上采样,再进行卷积操作,表示为:
Figure BDA0003126261160000045
其中,UP(·)表示如果输入特征尺寸不一致,则进行双线性上采样操作,使尺寸为输入的最大尺寸;BConvN(·)是结合了标准的N×N卷积运算和批量归一化的顺序操作;
Figure BDA0003126261160000046
为处理后的特征,每个特征也相应是一张特征图像;
然后,采用逐像素相乘的方法来增强特征图像中的公共像素,并且消除部分模糊像素,再将增强后的特征
Figure BDA0003126261160000047
和特征
Figure BDA0003126261160000048
拼接起来,得到携带了较丰富的上下文信息的特征
Figure BDA0003126261160000049
Figure BDA00031262611600000410
其中,∏表示逐元素累乘符号,
Figure BDA00031262611600000411
表示通道间的拼接操作;
最后,对拼接后的特征
Figure BDA0003126261160000051
采用残差的思想进行卷积操作,从而获得编码后的低层特征图
Figure BDA0003126261160000052
Figure BDA0003126261160000053
其中,ConvN表示标准的N×N卷积操作;
Figure BDA0003126261160000054
表示逐元素加法操作,
Figure BDA0003126261160000055
表示ReLU激活函数;
同理,对于扩大感受野范围的高层特征组
Figure BDA0003126261160000056
按照上述过程完成特征编码操作,最终获得高层特征图
Figure BDA0003126261160000057
更进一步的,特征解码模块的处理过程如下:
首先,将高层特征图上采样到与低层特征图一样的尺寸,并采用卷积操作分别对低层特征图和高层特征图进行初步变换:
Figure BDA0003126261160000058
其中,l∈[low,high];
然后,对变换后的高、低层特征图进行逐像素相乘操作来增强公共像素,并将增强后的特征图与变换后的低、高层特征图拼接起来:
Figure BDA0003126261160000059
其中,
Figure BDA00031262611600000510
表示逐元素相乘操作;
接着,采用注意力机制来消除
Figure BDA00031262611600000511
中无关特征的干扰,通过执行全局平均池化来压缩特征,再采用两个全连接层得到每个通道的权值,再对输入特征进行加权,来突显伪装物体目标区域,具体如下:
经注意力机制操作后得到的特征表示为:
Figure BDA00031262611600000512
其中,Fatt(·)表示通道注意力机制操作;
然后,对获得的特征进行卷积变换降维,再分别与两个原初步变换后的特征进行逐像素相加,得到两组特征
Figure BDA0003126261160000061
最后,对两组特征经卷积操作后再拼接,使得具有区分性的特征组合到一起,再将通道降到1维输出,即得到初始的伪装图C1
Figure BDA0003126261160000062
其中,T表示连续的BConv和Conv操作,采用渐进降维的方式输出伪装图C1
更进一步的,边缘细化模块利用初始的伪装图C1来细化边缘特征,即L:{x1,x2,x3},以获得更全面的特征,过程具体如下:
在边缘特征通过感受野模块进行增强来得到特征
Figure BDA0003126261160000063
后,将特征
Figure BDA0003126261160000064
上采样到与特征{rf1 e,C1}相同的尺寸;
然后将初始伪装图C1与增强后且同尺寸的各个边缘特征进行逐像素相乘,再经卷积操作归一化像素值,得到处理后的特征cei
Figure BDA0003126261160000065
其中,i∈{1,2,3};
接着,采用拼接的方式将处理后的特征聚合起来,再对其降维以及与初始伪装图C1逐像素相加,再利用1×1卷积降维得到最终的伪装图C2
Figure BDA0003126261160000066
更进一步的,在步骤S3训练模型的过程中,边缘细化模块还生成训练用的边缘图E1
将处理后的特征{ce1,ce2,ce3}和增强后且同尺寸的边缘特征
Figure BDA0003126261160000067
逐像素相加,并采用卷积操作进行变换:
Figure BDA0003126261160000068
然后,采用从高到低逐级拼接的方式将特征ei进行融合,最后,利用卷积操作逐步将通道数降至1维,以获得边缘图E1
Figure BDA0003126261160000071
训练模型所用的损失函数如下:
Figure BDA0003126261160000072
λce(P,G)=GlogP+(1-G)log(1-P) (14)
其中,Ltotal表示总体伪装损失;λce表示二进制交叉熵损失;α∈[0,1]为控制因子,用于控制两部分损失之间的权衡;
C、E分别是C2和E1上采样后获得的伪装物体映射图;GC为二元真值伪装图,GE为二元真值边缘图,采用传统的边缘检测算法来获得;
G={GC,GE},P为伪装物体检测网络预测的图,P={C,E},也就是说,当G为GC时,P对应为C,当G为GE时,P对应为E。
优选的,使用随机梯度下降法训练伪装物体检测网络。
优选的,在步骤S1中,构建伪装物体图像数据集的步骤为:
先收集各种类别伪装物体的原始图像,然后将收集的原始图像按比例随机划分为用于训练模型的训练数据集和用于检测模型的测试数据集,最后对数据集中的图像进行分类,并对每张图像标注用于包围伪装物体的包围框。
本发明相对于现有技术具有如下的优点及效果:
1、本发明设计了伪装物体检测网络,能在准确定位伪装物体的情况下,保证边缘的清晰,提高整体的检测精度,由于该网络是在大规模的伪装物体数据集上进行训练而得到,因此,不仅仅是检测伪装的动物,它在检测人类等其他的物体时也可以取得很显著的效果,具有很好的鲁棒性和通用性,因此,本发明检测方法在计算机、军事、医学、农业等多个领域都存在明显的应用价值。
2、本发明充分利用了输入图像的高低级信息,由于低级特征保留了用于构建物体边缘的空间信息,高级特征保留了用于定位目标的语义信息,因此最终获得的伪装图可以很好地区分图像中背景和伪装物体,有利于定位及区分出伪装物体。
3、本发明伪装物体检测模型可基于深度学习技术来处理特征聚合和伪装物体边缘,而基于深度学习的方法学习到的深层特征具有更好的通用性,整体检测的精度也会更佳,这点是其他方法很难做到的。
4、本发明伪装物体检测网络是端对端训练的,比起现有的双流网络模型,其运行速度更快,在实时性方面存在着优势。
附图说明
图1为本发明实施例1中的伪装物体检测网络的整体结构图。
图2为本发明实施例1中的特征解码模块的结构图。
图3为本发明实施例1中的边缘细化模块的结构图。
具体实施方式
下面结合实施例及附图对本发明作进一步详细的描述,但本发明的实施方式不限于此。
实施例1
本实施例提供了一种基于边缘细化的伪装物体检测方法,包括如下步骤:
S1、构建伪装物体图像数据集:
首先,从各个摄影学网站收集各种类别伪装物体的原始图像,伪装物体例如可以是昆虫、动植物、违禁物品等。
然后,将收集的原始图像按比例如(6:4)、(7:3)随机划分为用于训练模型的训练数据集和用于检测模型的测试数据集;
最后,对数据集中的图像进行分类,并对每张图像标注用于包围伪装物体的包围框。标注包围框有利于提高检测模型的检测能力。
S2、构建伪装物体检测网络,如图1所示,具体包括特征提取模块、感受野(Receptive Field,RF)模块、特征编码模块(Feature Encoder Module,FEM)、特征解码模块(Feature Decoder Module,FDM)以及边缘细化模块(Edge Refinement Moudel,ERM),特征提取模块、感受野模块、特征编码模块、特征解码模块和边缘细化模块依次连接,感受野模块连接至边缘细化模块。
S3、训练模型:使用伪装物体图像数据集对构建的伪装物体检测网络进行迭代训练,该伪装物体检测网络将伪装物体检测任务分成两个阶段,第一阶段负责伪装物体检测,第二阶段负责细化伪装物体边缘,训练完成后可得到伪装物体检测模型。
对于第一阶段:特征提取模块以原始图像作为输入,输出对应的一组特征,该组特征经感受野扩大搜索范围,再经特征编码模块和特征解码模块进行融合,特征解码模块输出初始的伪装图;
对于第二阶段:特征提取模块所获取到的特征中的较低层特征经感受野扩大搜索范围后,和初始伪装图同时作为边缘细化模块的输入,边缘细化模块输出最终的伪装图。
其中,(1)特征提取模块采用ResNet50网络,其处理过程如下:
ResNet50网络从输入图像I∈RH×W×3提取出一组多尺度特征{x1,x2,x3,x4},其浅层网络可提取低级特征(也可称为低层特征),该低级特征保留了用于构建物体边缘的空间信息,深层网络可提取高级特征(也可称为高层特征),该高级特征保留了用于定位目标的语义信息。
由于伪装物体自身的复杂性,直接聚集多层次的特征通常会导致细节缺失或引入模糊的特征,这两者都会导致网络无法优化。因此,为了能够充分利用低级特征中的细节信息和高级特征中的语义信息的同时,又降低多层次的特征融合造成的信息混淆,这里还将多尺度特征按层次划分为低级特征组L:{x1,x2,x3}和高级特征组H:{x2,x3,x4},从而实现分组保存相邻特征层的更多信息的同时,缩小不同层特征之间的差距,之后可再采用聚合策略来有效融合不同层特征。其中,x1为低级特征,x2,x3为中级特征,x4为高级特征。
(2)特征组{x1,x2,x3,x4}中的每层特征分别通过感受野模块来扩大感受野的范围,过程具体如下:
感受野模块含有多个分支,如图1所示,对于前4个分支,每个分支首先应用1×1卷积来减小输入特征的通道维数,第k(k∈{2,3,4})个分支再应用核大小为2k-1、膨胀率为1的卷积运算,接着再应用另一个3×3卷积运算,膨胀率为2k-1,以从多尺度特征中挖掘出全局上下文信息。
接下来,将该4个分支的输出拼接在一起,然后应用1×1卷积运算来将拼接后的特征的通道数减少到32,再将处理后的特征与原输入特征形成一个残差连接,最终得到的特征表示为:
Figure BDA0003126261160000101
其中,j∈{1,2,3,4},C∈{c,e},c表示伪装物体检测阶段,e表示细化伪装物体边缘阶段;FRF(·)表示扩大感受野操作。
对于低级特征组L:{x1,x2,x3}中的每层特征,按照上述过程完成扩大感受野操作,最终得到特征:
Figure BDA0003126261160000102
同理,对于高级特征组H:{x2,x3,x4}中的每层特征,按照上述过程完成扩大感受野操作,最终得到特征:
Figure BDA0003126261160000103
(3)特征编码模块可有效地对上述两组多层次特征即低级特征组和高级特征组中的多尺度多层次信息分别进行整合。本实施例引入了一种轻量化的特征编码模块,利用相乘和拼接等操作来分别整合这两组多尺度多层次的特征,处理过程具体如下:
如图1所示,首先,对扩大感受野范围的低层特征组
Figure BDA0003126261160000111
进行双线性上采样,再进行卷积操作,可表示为:
Figure BDA0003126261160000112
其中,UP(·)表示如果输入特征尺寸不一致,则进行双线性上采样操作,使尺寸为输入的最大尺寸;BConvN(·)是结合了标准的N×N卷积运算和批量归一化的顺序操作;
Figure BDA0003126261160000113
为处理后的特征,每个特征也相应是一张特征图像。
然后,采用逐像素相乘的方法来增强特征图像中的公共像素,并且消除部分模糊像素,再将增强后的特征
Figure BDA0003126261160000114
和特征
Figure BDA0003126261160000115
拼接起来,得到携带了较丰富的上下文信息的特征
Figure BDA0003126261160000116
Figure BDA0003126261160000117
其中,∏表示逐元素累乘符号,
Figure BDA0003126261160000118
表示通道间的拼接操作;
最后,对拼接后的特征
Figure BDA0003126261160000119
采用残差的思想进行卷积操作,从而获得编码后的低层特征图
Figure BDA00031262611600001110
Figure BDA00031262611600001111
其中,ConvN表示标准的N×N卷积操作;
Figure BDA00031262611600001112
表示逐元素加法操作,
Figure BDA00031262611600001113
表示ReLU激活函数;
同理,对于扩大感受野范围的高层特征组
Figure BDA00031262611600001114
按照上述过程完成特征编码操作,最终获得高层特征图
Figure BDA00031262611600001115
(4)特征解码模块可针对两个不同层次的特征,自适应地突显目标物体本身,其模型结构可参见图2,解码的过程如下:
首先,将高层特征图上采样到与低层特征图一样的尺寸,并采用卷积操作分别对低层特征图和高层特征图进行初步变换:
Figure BDA0003126261160000121
其中,l∈[low,high]。
然后,对变换后的高、低层特征图进行逐像素相乘操作来增强公共像素,并将增强后的特征图与变换后的低、高层特征图拼接起来:
Figure BDA0003126261160000122
其中,
Figure BDA0003126261160000123
表示逐元素相乘操作。
在整合丰富的多层特征之后,采用一个有效的轻量级的注意力机制来消除
Figure BDA0003126261160000124
中无关特征的干扰,通过执行全局平均池化来压缩特征,再采用两个全连接层得到每个通道的权值,再对输入特征进行加权,来突显伪装目标区域,具体如下:
经注意力机制操作后得到的特征表示为:
Figure BDA0003126261160000125
其中,Fatt(·)表示通道注意力机制操作。
然后,对获得的特征进行卷积变换降维,再分别与两个原初步变换后的特征进行逐像素相加,得到两组特征
Figure BDA0003126261160000126
如此可使与伪装目标相关的特征在每一组源特征中突出。
最后,对两组特征经卷积操作后再拼接,使得具有区分性的特征组合到一起,再将通道降到1维输出,即得到初始的伪装图C1
Figure BDA0003126261160000127
其中,T表示连续的BConv和Conv操作,卷积核大小分别为3×3和1×1,采用渐进降维的方式输出88×88×1的伪装图C1
(5)由于伪装物体往往存在边界模糊这一现象,因此,本实施例以此为出发点,从主干网络(即第一阶段的网络)引出边缘特征分支,边缘细化模块利用初始伪装图C1来细化低层边缘细节特征,即L:{x1,x2,x3},以获得更全面的特征,如图3所示,边缘细化模块的处理过程具体如下:
51)在边缘特征通过感受野模块进行增强得到特征
Figure BDA0003126261160000131
后,将特征
Figure BDA0003126261160000132
上采样到与特征{rf1 e,C1}相同的尺寸。
52)然后将初始伪装图C1与增强后且同尺寸的各个边缘特征进行逐像素相乘,再经卷积操作归一化像素值,得到处理后的特征cei
Figure BDA0003126261160000133
其中,i∈{1,2,3}。
53)接着,采用拼接的方式将处理后的特征聚合起来,再对其降维以及与初始伪装图C1逐像素相加,达到细化边缘的目的,再利用1×1卷积降维得到最终的有区分出图像背景和伪装物体的伪装图C2
Figure BDA0003126261160000134
54)同时,为了实现监督边缘分支达到优化网络架构的效果,边缘细化模块还将几个边缘分支聚合为一个边缘特征输出,并生成训练用的边缘图E1
将处理后的特征{ce1,ce2,ce3}和增强后且同尺寸的边缘特征{rf1 e,rf2 e,rf3 e}逐像素相加,并采用卷积操作进行变换:
Figure BDA0003126261160000135
然后,采用从高到低逐级拼接的方式将特征ei进行融合,最后,利用卷积操作逐步将通道数降至1维,以获得边缘图E1
Figure BDA0003126261160000136
基于伪装图C2和边缘图E1,即可相应地计算损失函数,进而根据损失函数来调整伪装物体检测网络。
损失函数如下:
Figure BDA0003126261160000141
λce(P,G)=GlogP+(1-G)log(1-P)(14)
其中,Ltotal表示总体伪装损失;λce表示二进制交叉熵损失;α∈[0,1]为控制因子,用于控制两部分损失之间的权衡;
C、E分别是C2和E1上采样后获得的伪装物体映射图;GC为二元真值伪装图,GE为二元真值边缘图,通常采用传统的边缘检测算法来获得,图像的分辨率均为:352×352;
G={GC,GE},P为伪装物体检测网络预测的图,P={C,E},也就是说,当G为GC时,P对应为C,当G为GE时,P对应为E。
在本实施例中,使用步骤S1划分好的训练数据集对步骤S2构建的伪装物体检测网络进行训练的实验细节具体如下:
1)训练时,将原始图像调整为352×352,使用随机梯度下降法(Stochasticgradient descent)以0.9的动量和0.0005的权重衰减训练网络,epoch为30,batch size为8,初始学习率为0.002,采用学习计划策略,在epoch达到20后学习率下降10%。
2)使用多尺度输入图像的大小来增加数据量,通常按照[0.75,1,1.25]的比例变换。
3)每10个epoch保存一次模型。
上述训练参数可根据实际情况设置。
S4、测试模型:使用步骤S1划分好的测试数据集对步骤S3)训练后的伪装物体检测模型进行检测,即以待预测的伪装物体原始图像作为训练完成后的伪装物体检测模型的输入,利用该伪装物体检测模型输出对应的有区分出图像背景和伪装物体的伪装图。检测人员根据该伪装图,可有效定位伪装物体及判断伪装物体类型。
可通过各种手段实施本发明描述的技术。举例来说,这些技术可实施在硬件、固件、软件或其组合中。对于硬件实施方案,处理模块可实施在一个或一个以上专用集成电路(ASIC)、数字信号处理器(DSP)、可编程逻辑装置(PLD)、现场可编辑逻辑门阵列(FPGA)、处理器、控制器、微控制器、电子装置、其他经设计以执行本发明所描述的功能的电子单元或其组合内。
对于固件和/或软件实施方案,可用执行本文描述的功能的模块(例如,过程、步骤、流程等)来实施所述技术。固件和/或软件代码可存储在存储器中并由处理器执行。存储器可实施在处理器内或处理器外部。本领域普通技术人员可以理解:实现上述方法实施例的全部或部分步骤可以通过程序指令相关的硬件来完成,前述的程序可以存储在一计算机可读取存储介质中,该程序在执行时,执行包括上述方法实施例的步骤;而前述的存储介质包括:ROM、RAM、磁碟或者光盘等各种可以存储程序代码的介质。
以上仅为本发明创造的较佳实施例而已,并不用以限制本发明创造,凡在本发明创造的精神和原则之内所作的任何修改、等同替换和改进等,均应包含在本发明创造的保护范围之内。

Claims (10)

1.一种基于边缘细化的伪装物体检测方法,其特征在于,包括如下步骤:
S1、利用各类伪装物体的原始图像来构建伪装物体图像数据集;
S2、构建伪装物体检测网络;
S3、训练模型:使用伪装物体图像数据集对构建的伪装物体检测网络进行迭代训练,该伪装物体检测网络将伪装物体检测任务分成两个阶段,第一阶段负责伪装物体检测,第二阶段负责细化伪装物体边缘,训练完成后得到伪装物体检测模型;
S4、测试模型:以待预测的伪装物体原始图像作为训练完成后的伪装物体检测模型的输入,利用该伪装物体检测模型输出对应的有区分出图像背景和伪装物体的伪装图。
2.根据权利要求1所述的伪装物体检测方法,其特征在于,伪装物体检测网络包括特征提取模块、感受野模块、特征编码模块、特征解码模块以及边缘细化模块,其中,特征提取模块、感受野模块、特征编码模块、特征解码模块和边缘细化模块依次连接,感受野模块连接至边缘细化模块;
对于第一阶段:特征提取模块以原始图像作为输入,输出对应的一组特征,该组特征经感受野扩大搜索范围,再经特征编码模块和特征解码模块进行融合,特征解码模块输出初始的伪装图;
对于第二阶段:特征提取模块所获取到的特征中的较低层特征经感受野扩大搜索范围后,和初始伪装图同时作为边缘细化模块的输入,边缘细化模块输出最终的伪装图。
3.根据权利要求2所述的伪装物体检测方法,其特征在于,特征提取模块的处理过程如下:
对于输入图像I∈RH×W×3,利用ResNet50网络从输入图像提取出一组多尺度特征{x1,x2,x3,x4},该组多尺度特征中的低级特征保留了用于构建物体边缘的空间信息,高级特征保留了用于定位目标的语义信息;
然后,将多尺度特征按层次划分为低级特征组L:{x1,x2,x3}和高级特征组H:{x2,x3,x4},其中,x1为低级特征,x2,x3为中级特征,x4为高级特征。
4.根据权利要求3所述的伪装物体检测方法,其特征在于,多尺度特征{x1,x2,x3,x4}中的每层特征分别经感受野模块扩大感受野的范围,过程具体如下:
感受野模块含有多个分支,对于前4个分支,每个分支首先应用1×1卷积来减小输入特征的通道维数,第k(k∈{2,3,4})个分支再应用核大小为2k-1、膨胀率为1的卷积运算,接着再应用另一个不同应用核大小、膨胀率为2k-1的卷积运算;
将该4个分支的输出拼接在一起,然后应用1×1卷积运算来减小拼接后的特征的通道数,再将处理后的特征与原输入特征形成一个残差连接,最终得到的特征表示为:
Figure FDA0003126261150000021
其中,j∈{1,2,3,4},C∈{c,e},c表示伪装物体检测阶段,e表示细化伪装物体边缘阶段;FRF(·)表示扩大感受野操作;
对于低级特征组L:{x1,x2,x3}中的每层特征,按照上述过程完成扩大感受野操作,最终得到特征:
Figure FDA0003126261150000022
同理,对于高级特征组H:{x2,x3,x4}中的每层特征,按照上述过程完成扩大感受野操作,最终得到特征:
Figure FDA0003126261150000023
5.根据权利要求4所述的伪装物体检测方法,其特征在于,特征编码模块的处理过程如下:
首先,对扩大感受野范围的低层特征组
Figure FDA0003126261150000024
进行双线性上采样,再进行卷积操作,表示为:
Figure FDA0003126261150000031
其中,UP(·)表示如果输入特征尺寸不一致,则进行双线性上采样操作,使尺寸为输入的最大尺寸;BConvN(·)是结合了标准的N×N卷积运算和批量归一化的顺序操作;
Figure FDA0003126261150000032
为处理后的特征,每个特征也相应是一张特征图像;
然后,采用逐像素相乘的方法来增强特征图像中的公共像素,并且消除部分模糊像素,再将增强后的特征
Figure FDA0003126261150000033
和特征
Figure FDA0003126261150000034
拼接起来,得到携带了较丰富的上下文信息的特征
Figure FDA0003126261150000035
Figure FDA0003126261150000036
其中,∏表示逐元素累乘符号,
Figure FDA0003126261150000037
表示通道间的拼接操作;
最后,对拼接后的特征
Figure FDA0003126261150000038
采用残差的思想进行卷积操作,从而获得编码后的低层特征图
Figure FDA0003126261150000039
Figure FDA00031262611500000310
其中,ConvN表示标准的N×N卷积操作;
Figure FDA00031262611500000311
表示逐元素加法操作,
Figure FDA00031262611500000312
表示ReLU激活函数;
同理,对于扩大感受野范围的高层特征组
Figure FDA00031262611500000313
按照上述过程完成特征编码操作,最终获得高层特征图
Figure FDA00031262611500000314
6.根据权利要求5所述的伪装物体检测方法,其特征在于,特征解码模块的处理过程如下:
首先,将高层特征图上采样到与低层特征图一样的尺寸,并采用卷积操作分别对低层特征图和高层特征图进行初步变换:
Fl fem'=BConv3(UP(Fl fem)) (5)
其中,l∈[low,high];
然后,对变换后的高、低层特征图进行逐像素相乘操作来增强公共像素,并将增强后的特征图与变换后的低、高层特征图拼接起来:
Figure FDA0003126261150000041
其中,
Figure FDA0003126261150000042
表示逐元素相乘操作;
接着,采用注意力机制来消除
Figure FDA0003126261150000043
中无关特征的干扰,通过执行全局平均池化来压缩特征,再采用两个全连接层得到每个通道的权值,再对输入特征进行加权,来突显伪装物体目标区域,具体如下:
经注意力机制操作后得到的特征表示为:
Figure FDA0003126261150000044
其中,Fatt(·)表示通道注意力机制操作;
然后,对获得的特征进行卷积变换降维,再分别与两个原初步变换后的特征进行逐像素相加,得到两组特征
Figure FDA0003126261150000045
最后,对两组特征经卷积操作后再拼接,使得具有区分性的特征组合到一起,再将通道降到1维输出,即得到初始的伪装图C1
Figure FDA0003126261150000046
其中,T表示连续的BConv和Conv操作,采用渐进降维的方式输出伪装图C1
7.根据权利要求4所述的伪装物体检测方法,其特征在于,边缘细化模块利用初始的伪装图C1来细化边缘特征,即L:{x1,x2,x3},以获得更全面的特征,过程具体如下:
在边缘特征通过感受野模块进行增强来得到特征
Figure FDA0003126261150000047
后,将特征
Figure FDA0003126261150000048
上采样到与特征{rf1 e,C1}相同的尺寸;
然后将初始伪装图C1与增强后且同尺寸的各个边缘特征进行逐像素相乘,再经卷积操作归一化像素值,得到处理后的特征cei
Figure FDA0003126261150000051
其中,i∈{1,2,3};
接着,采用拼接的方式将处理后的特征聚合起来,再对其降维以及与初始伪装图C1逐像素相加,再利用1×1卷积降维得到最终的伪装图C2
Figure FDA0003126261150000052
8.根据权利要求7所述的伪装物体检测方法,其特征在于,在步骤S3训练模型的过程中,边缘细化模块还生成训练用的边缘图E1
将处理后的特征{ce1,ce2,ce3}和增强后且同尺寸的边缘特征
Figure FDA0003126261150000053
逐像素相加,并采用卷积操作进行变换:
Figure FDA0003126261150000054
然后,采用从高到低逐级拼接的方式将特征ei进行融合,最后,利用卷积操作逐步将通道数降至1维,以获得边缘图E1
Figure FDA0003126261150000055
训练模型所用的损失函数如下:
Ltotal=αlce(C,GC)+(1-α)lce(E,GE) (13)
λce(P,G)=GlogP+(1-G)log(1-P) (14)
其中,Ltotal表示总体伪装损失;λce表示二进制交叉熵损失;α∈[0,1]为控制因子,用于控制两部分损失之间的权衡;
C、E分别是C2和E1上采样后获得的伪装物体映射图;GC为二元真值伪装图,GE为二元真值边缘图,采用传统的边缘检测算法来获得;
G={GC,GE},P为伪装物体检测网络预测的图,P={C,E},也就是说,当G为GC时,P对应为C,当G为GE时,P对应为E。
9.根据权利要求1所述的伪装物体检测方法,其特征在于,使用随机梯度下降法训练伪装物体检测网络。
10.根据权利要求1所述的伪装物体检测方法,其特征在于,在步骤S1中,构建伪装物体图像数据集的步骤为:
先收集各种类别伪装物体的原始图像,然后将收集的原始图像按比例随机划分为用于训练模型的训练数据集和用于检测模型的测试数据集,最后对数据集中的图像进行分类,并对每张图像标注用于包围伪装物体的包围框。
CN202110691277.3A 2021-06-22 2021-06-22 一种基于边缘细化的伪装物体检测方法 Active CN113468996B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202110691277.3A CN113468996B (zh) 2021-06-22 2021-06-22 一种基于边缘细化的伪装物体检测方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202110691277.3A CN113468996B (zh) 2021-06-22 2021-06-22 一种基于边缘细化的伪装物体检测方法

Publications (2)

Publication Number Publication Date
CN113468996A true CN113468996A (zh) 2021-10-01
CN113468996B CN113468996B (zh) 2023-07-11

Family

ID=77869131

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202110691277.3A Active CN113468996B (zh) 2021-06-22 2021-06-22 一种基于边缘细化的伪装物体检测方法

Country Status (1)

Country Link
CN (1) CN113468996B (zh)

Cited By (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN114241277A (zh) * 2021-12-22 2022-03-25 中国人民解放军国防科技大学 注意力引导的多特征融合伪装目标检测方法、装置、设备及介质
CN114581752A (zh) * 2022-05-09 2022-06-03 华北理工大学 一种基于上下文感知和边界细化的伪装目标检测方法
CN114627299A (zh) * 2022-04-21 2022-06-14 杭州电子科技大学 一种模仿人类视觉系统对伪装目标检测与分割方法
CN115223018A (zh) * 2022-06-08 2022-10-21 东北石油大学 伪装对象协同检测方法及装置、电子设备和存储介质
CN115631346A (zh) * 2022-11-11 2023-01-20 南京航空航天大学 一种基于不确定性建模的伪装物体检测方法和系统

Citations (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN101272450A (zh) * 2008-05-13 2008-09-24 浙江大学 Sprite编码中的全局运动估计外点去除及运动参数细化方法
CN101431606A (zh) * 2008-12-01 2009-05-13 上海广电(集团)有限公司中央研究院 一种基于边缘检测的自适应降噪处理方法
CN101901477A (zh) * 2010-07-27 2010-12-01 中国农业大学 植物叶片田间图像边缘提取方法及其系统
CN112288008A (zh) * 2020-10-29 2021-01-29 四川九洲电器集团有限责任公司 一种基于深度学习的马赛克多光谱图像伪装目标检测方法
CN112733744A (zh) * 2021-01-14 2021-04-30 北京航空航天大学 一种基于边缘协同监督与多级约束的伪装物体检测模型

Patent Citations (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN101272450A (zh) * 2008-05-13 2008-09-24 浙江大学 Sprite编码中的全局运动估计外点去除及运动参数细化方法
CN101431606A (zh) * 2008-12-01 2009-05-13 上海广电(集团)有限公司中央研究院 一种基于边缘检测的自适应降噪处理方法
CN101901477A (zh) * 2010-07-27 2010-12-01 中国农业大学 植物叶片田间图像边缘提取方法及其系统
CN112288008A (zh) * 2020-10-29 2021-01-29 四川九洲电器集团有限责任公司 一种基于深度学习的马赛克多光谱图像伪装目标检测方法
CN112733744A (zh) * 2021-01-14 2021-04-30 北京航空航天大学 一种基于边缘协同监督与多级约束的伪装物体检测模型

Non-Patent Citations (1)

* Cited by examiner, † Cited by third party
Title
张润生;贺超;况朝青;: "基于深度神经网络的伪装人脸识别", 电子技术应用, no. 05, pages 1 - 4 *

Cited By (8)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN114241277A (zh) * 2021-12-22 2022-03-25 中国人民解放军国防科技大学 注意力引导的多特征融合伪装目标检测方法、装置、设备及介质
CN114627299A (zh) * 2022-04-21 2022-06-14 杭州电子科技大学 一种模仿人类视觉系统对伪装目标检测与分割方法
CN114627299B (zh) * 2022-04-21 2023-10-27 杭州电子科技大学 一种模仿人类视觉系统对伪装目标检测与分割方法
CN114581752A (zh) * 2022-05-09 2022-06-03 华北理工大学 一种基于上下文感知和边界细化的伪装目标检测方法
CN114581752B (zh) * 2022-05-09 2022-07-15 华北理工大学 一种基于上下文感知和边界细化的伪装目标检测方法
CN115223018A (zh) * 2022-06-08 2022-10-21 东北石油大学 伪装对象协同检测方法及装置、电子设备和存储介质
CN115223018B (zh) * 2022-06-08 2023-07-04 东北石油大学 伪装对象协同检测方法及装置、电子设备和存储介质
CN115631346A (zh) * 2022-11-11 2023-01-20 南京航空航天大学 一种基于不确定性建模的伪装物体检测方法和系统

Also Published As

Publication number Publication date
CN113468996B (zh) 2023-07-11

Similar Documents

Publication Publication Date Title
CN113468996A (zh) 一种基于边缘细化的伪装物体检测方法
CN109685819B (zh) 一种基于特征增强的三维医学图像分割方法
CN110378381A (zh) 物体检测方法、装置和计算机存储介质
CN108596258A (zh) 一种基于卷积神经网络随机池化的图像分类方法
CN111369440A (zh) 模型训练、图像超分辨处理方法、装置、终端及存储介质
CN110838125B (zh) 医学图像的目标检测方法、装置、设备、存储介质
CN111860398B (zh) 遥感图像目标检测方法、系统及终端设备
Wang et al. FE-YOLOv5: Feature enhancement network based on YOLOv5 for small object detection
CN110619638A (zh) 一种基于卷积块注意模块的多模态融合显著性检测方法
CN112750140A (zh) 基于信息挖掘的伪装目标图像分割方法
CN113591968A (zh) 一种基于非对称注意力特征融合的红外弱小目标检测方法
CN113191489B (zh) 二值神经网络模型的训练方法、图像处理方法和装置
CN112149526B (zh) 一种基于长距离信息融合的车道线检测方法及系统
CN115841629A (zh) 一种基于卷积神经网络的sar图像舰船检测方法
CN117237808A (zh) 基于odc-yolo网络的遥感图像目标检测方法及系统
CN113221731B (zh) 一种多尺度遥感图像目标检测方法及系统
CN109543744B (zh) 一种基于龙芯派的多类别深度学习图像识别方法及其应用
CN114882278A (zh) 一种基于注意力机制和迁移学习的轮胎花纹分类方法和装置
CN111881803B (zh) 一种基于改进YOLOv3的畜脸识别方法
CN117115632A (zh) 一种水下目标检测方法、装置、设备及介质
CN116740516A (zh) 基于多尺度融合特征提取的目标检测方法及系统
CN114663654B (zh) 一种改进YOLOv4网络模型及小目标检测方法
CN115578624A (zh) 农业病虫害模型构建方法、检测方法及装置
CN115049901A (zh) 一种基于特征图加权注意力融合的小目标检测方法及设备
CN114724023A (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