CN112836615B - 基于深度学习与全局推理的遥感影像多尺度固废检测方法 - Google Patents
基于深度学习与全局推理的遥感影像多尺度固废检测方法 Download PDFInfo
- Publication number
- CN112836615B CN112836615B CN202110114848.7A CN202110114848A CN112836615B CN 112836615 B CN112836615 B CN 112836615B CN 202110114848 A CN202110114848 A CN 202110114848A CN 112836615 B CN112836615 B CN 112836615B
- Authority
- CN
- China
- Prior art keywords
- detection
- scale
- image
- solid waste
- remote sensing
- 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
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/10—Terrestrial scenes
- G06V20/13—Satellite images
-
- 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
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N5/00—Computing arrangements using knowledge-based models
- G06N5/04—Inference or reasoning models
-
- 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/25—Determination of region of interest [ROI] or a volume of interest [VOI]
-
- 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/40—Extraction of image or video features
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V2201/00—Indexing scheme relating to image or video recognition or understanding
- G06V2201/07—Target detection
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Artificial Intelligence (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Multimedia (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Health & Medical Sciences (AREA)
- Life Sciences & Earth Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Astronomy & Astrophysics (AREA)
- Remote Sensing (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种基于深度学习与全局推理的遥感影像多尺度固废检测方法,包括:S1、获取待检测区域的原始遥感影像,并对其进行多尺度裁剪处理,获得各尺度下的影像块集合;S2、通过全局推理多尺度检测模型对各尺度下的影像块集合中的影像进行检测,输出各尺度下的检测框;S3、将输出检测框整合到原始遥感图像对应的大范围影像中,并对其合并,得到对应的固废检测结果。本发明该方法基于SSD模型,在原始的卷积层之间嵌入GloRe单元,使模型在进行多尺度检测时中充分利用影像的全局信息;针对大范围的固废堆场检测任务提出一种多尺度优化策略,提升对尺度差异巨大的固废堆场目标的检测效果,并有效解决完整堆场被滑动窗口切割的问题。
Description
技术领域
本发明属于固废堆场检测技术领域,具体涉及一种基于深度学习与全局推理的遥感影像多尺度固废检测方法。
背景技术
随着社会经济的飞速发展,城市化进程逐渐加快,诸多城市问题也随之而来,严重影响居民的生活舒适度与便利性,其中固体废物引发的城市环境问题尤为严峻,推动高效可靠的固废排查工作势在必行。快速精准地定位固废堆场对开展高效的固废排查工作具有重要意义,最初确定固废堆场位置主要依靠人工实地勘察,再有相关部门逐级上报的方式进行,然而由于固废堆场分布范围广、数量大等特点,人工方法不仅费时费力,还存在主观性强、作用范围局限等问题,难以满足环境治理的迫切需要。近年来,遥感成像技术的发展丰富了高分辨率遥感影像数据源,为固废堆场检测任务提供了新的手段,利用高分辨率遥感影像能够实现大范围、高效率、更为客观的固废堆场检测,极大地弥补了人工方法的不足。
在早期基于遥感影像的固废堆场研究中,人们大多通过设计影像特征,并训练特定分类器的方法完成固废堆场识别或提取任务,但由于针对特定应用场景的特征设计模式在面对情况复杂的实际任务时无法表现出足够的泛化能力,此类方法往往难以推广应用。随着深度学习方法在计算机视觉领域的迅速发展,卷积神经网络(Convolutional NeuralNetwork,CNN)以自动学习图像特征的方式弥补了人工设计特征的局限性,由此衍生出的众多目标检测算法在针对自然图像的目标检测应用中取得优秀表现,比如采用可自动学习候选框的区域建议网络(Region Proposal Network,RPN)的Faster-RCNN(Ren S等,2016)以及利用多层级特征提高多尺度目标检测效果的SSD(Liu W等,2016)等,近些年来,越来越多的研究将深度学习方法迁移至遥感影像的目标检测应用中,Chen X等(Chen X等,2014)采用一种基于滑动窗口与神经网络的方法对遥感影像上的车辆进行高效检测;Cheng G等(Cheng G等,2016)提出一种新颖的旋转不变CNN模型(RICNN)有效处理遥感影像上目标旋转变化的问题。然而,相较于建筑、车辆等一般地物,固废堆场在遥感影像上的特征表现更加复杂,将应用于一般地物的目标检测算法直接迁移至固废堆场检测任务中或难以达到同样的优秀表现,往往需要根据固废堆场检测任务的特性对深度学习模型进行针对性优化,Sun X等(Sun X等,2020)针对固废堆场形状不规则、边界模糊等特性提出一种形状鲁棒的无锚框检测模型——SRAF-Net,并采用一种基于上下文的可变形模块提高特征提取性能;Wu T等(Wu T等,2020,Xiao S,2018)采用一种针对固废堆场的样本合成方法进行训练样本更新,结合深度学习模型提升了固废堆场检测效果。
然而,固废堆场检测任务的复杂特性远不止如此,具体而言:1)目标尺度多变且差异巨大,小型堆场面积仅有几十平方米,大型堆场面积可达几千平方米,固定大小的滑动窗口难以同时顾及尺度差异如此巨大的多个目标,此外影像窗口在大范围目标区域中滑动时往往会将完整堆场切割,因此需要针对固废堆场检测的任务特性合理优化多尺度检测策略;2)固废堆场的形成与其他地物关联紧密,比如生活固废往往伴随住宅楼房出现,工业固废和建筑废物多堆积于空旷裸地、草地等地物之上,因此对影像全局信息充分推理挖掘能够有效增强特征表达,然而一般的局部卷积过程受限于卷积核,难以全面概览影像全局信息。
发明内容
针对现有技术中的上述不足,本发明提供的基于深度学习与全局推理的遥感影像多尺度固废检测方法解决了现有的固废堆场检测过程,尺度差异巨大的固废堆场对目标检测效果不理想,存在完整场堆被滑动窗口切割的问题。
为了达到上述发明目的,本发明采用的技术方案为:基于深度学习与全局推理的遥感影像多尺度固废检测方法,包括以下步骤:
S1、获取待检测区域的原始遥感影像,并对其进行多尺度裁剪处理,获得各尺度下的影像块集合;
S2、通过全局推理多尺度检测模型对各尺度下的影像块集合中的影像进行检测,输出各尺度下的所有检测框;
S3、将全局推理多尺度检测模型输出的检测框整合到原始遥感图像对应的大范围影像中,并对各检测框进行合并,得到对应的固废检测结果。
进一步地,所述步骤S1具体为:
S11、采用若干个尺寸的滑动窗口和步长对待检测区域的原始遥感影像进行裁剪;
S12、将裁剪后的各影像块缩放至全局推理多尺度检测模型的输入大小,得到各尺度下的影像块集合。
进一步地,所述步骤S2中的全局推理多尺度检测模型包括SSD网络及在嵌合在SSD网络中各卷积层之间的GloRe单元;
所述SSD网络用于实现对输入影像的特征提取及多尺度目标检测;
所述GloRe单元用于对SSD网络中的特征进行全局推理,提高SSD网络对影像全局信息的表达能力。
进一步地,所述SSD网络保留VGG-16网络的第一~第五卷积层,并将VGG-16网络第五层卷积层之后的全连接层修改为依次连接的第六卷积层、第七卷积层,同时在第七卷积层后依次添加依次连接的第八~第十一卷积层;
对于嵌合在SSD网络中的GloRe单元,根据SSD网络的特征计算方向和检测计算方向之后自由嵌合。
进一步地,在SSD网络中,所述第四卷积层及第七卷积层~第十一卷积层除进行特征向前计算外,其输出的六个不同分辨率的特征图还用于计算目标的边框、位置及类别置信度,并基于计算结果通过非极大值抑制操作,筛选出与输入图像中重叠度大于设定阈值的影像,作为全局推理多尺度检测模型的输出结果。
进一步地,对于前一卷积层输入到GloRe单元的栅格形式的特征图X,GloRe单元对特征图X进行处理的方法为:
A1、采用卷积运算φ(X;Wφ)和卷积运算θ(X;Wθ)将特征图X映射到交互空间中的一个全连通图结构;
A2、采用图卷积运算在整个全连通图结构中传递每个节点的特征,并更新节点状态,进而在关系密切的节点之间构建强关联;
A3、采用θ(X;Wθ)的转置将节点之间具有强关联的全连通图结构从交互空间反向映射到栅格形式,得到栅格特征图Y;
A4、将栅格特征图Y的维度扩展至与特征图X大小相同后与特征图X叠加,作为GloRe单元的输出图像输入到下一层卷积层。
进一步地,所述步骤A1中,全连通图结构中节点的特征矩阵V为:
式中,Wφ为降维卷积运算的权重,Wθ为映射卷积运算的权重;
所述步骤A2中,采用图卷积运算在整个全连通图结构V中传递每个节点的特征Z为:
Z=g(V;Wg)=((I-Ag)V)Wg
式中,g(·)为图卷积运算函数,Wg为图卷积的权重,I为单位矩阵,Ag为全连通图结构的邻接矩阵;
所述步骤A3中,所述栅格特征图Y为:
式中,θT(X;Wθ)为卷积运算θ(X;Wθ)的转置。
进一步地,所述步骤S3具体为:
S31、将全局推理多尺度检测模型输出的检测框叠加整合至原始遥感图像对应的大范围影像中,并以置信度阈值Confthr对所有检测框进行筛选;
S32、基于筛选结果,确定属于同一目标的检测框b的检测框集合gi,并构建目标集合G={g1,...,gm};
S33、将集合G={g1,...,gm}中包含检测框b的个数不小于2的检测框集合筛选出来,并计算其最小外接矩形,得到的各尺度下的影像块组成影像块检测集合,进而得到对应的固废检测结果。。
进一步地,在所述步骤S32中,所述目标检测框集合G={g1,...,gm}中每个元素gi={b1,b2,...,bn}表示被判定为属于同一目标的检测框b的影像集合,i=1,2,...,m,m为集合G中元素总数;
元素gi的最小外包矩形与检测框b之间的重叠区域满足:
式中,B为检测框集合,Sg、Sb、Sin、Sun分别表示gi的最小外包矩形、检测框b、最小外包矩形与检测框b的重叠区域和并集的面积,Mergethr为合并阈值。
本发明的有益效果为:
(1)考虑到遥感影像上的固废堆场具有尺度多变且与全局关联密切等特点,提出一种全局推理多尺度检测模型,该模型充分结合多尺度检测深度学习模型SSD与全局推理单元GloRe的优势,使模型在进行多尺度检测的过程中充分利用影像的全局信息,实验证明该模型能有效处理尺度多变、场景复杂的固废堆场检测任务,与其他方法相比,本发明中的全局推理多尺度检测模型具有明显优势;
(2)针对固废堆场尺度差异过大、在大幅影像中完整堆场会被切割的问题,本文提出一种多尺度优化策略,采用多尺度滑动窗口对目标影像裁剪,并采用一种多尺度合并方法对检测结果进行处理,经过与传统的NMS处理方法对比分析,该多尺度合并方法能够更好地应对大幅影像的固废堆场检测。
附图说明
图1为本发明提供的基于深度学习与全局推理的遥感影像多尺度固废检测方法流程图。
图2为本发明提供的完整固废堆场被切割示意图。
图3为本发明提供的多尺度裁剪示意图。
图4为本发明提供的全局推理多尺度模型结构示意图。
图5为本发明提供的检测结果合并示意图。
图6为本发明提供的实施例中不同方法的检测结果示意图。
图7为本发明提供的不同方法的PR曲线示意图。
图8为本发明提供的不同方法的大范围检测结果对比示意图
图9为本发明提供的不同合并阈值时的置信度阈值变化示意图。
图10为本发明提供的实施例中置信度阈值为0.99时的合并阈值变化示意图。
图11为本发明提供的实施例中不同合并方法的可视化结果示意图。
具体实施方式
下面对本发明的具体实施方式进行描述,以便于本技术领域的技术人员理解本发明,但应该清楚,本发明不限于具体实施方式的范围,对本技术领域的普通技术人员来讲,只要各种变化在所附的权利要求限定和确定的本发明的精神和范围内,这些变化是显而易见的,一切利用本发明构思的发明创造均在保护之列。
实施例1:
如图1所示,基于深度学习与全局推理的遥感影像多尺度固废检测方法,包括以下步骤:
S1、获取待检测区域的原始遥感影像,并对其进行多尺度裁剪处理,获得各尺度下的影像块集合;
S2、通过全局推理多尺度检测模型对各尺度下的影像块集合中的影像进行检测,输出各尺度下的所有检测框;
S3、将全局推理多尺度检测模型输出的检测框整合到原始遥感图像对应的大范围影像中,并对各检测框进行合并,得到对应的固废检测结果。
在本实施例的步骤S1中,对于该步骤中构造后续全局推理多尺度检测模型的输入图像,尽管后续的SSD网络内部设计了多尺度检测结构,但由于固废堆场的尺度差异过大,小型堆场面积仅几十平方米,大型堆场面积可达几千甚至几万平方米,而模型的输入大小固定,难以同时将较大型堆场和小型堆场特征表达完整;此外,在实际应用中,遥感影像往往覆盖范围较大的研究区域,需要以滑动窗口遍历整幅图像,此时极易出现完整固废堆场被切割的问题,如图2所示,因此,本实施例针对上述问题提出一种多尺度优化策略,以满足在大幅影像的实际中检测不同尺度固废堆场的需求。
本实施例中的多尺度优化过程主要分为多尺度剪裁和多尺度合并,如图3所示,多尺度剪裁对应于上述步骤S1,而多尺度合并在后续步骤S3中实现,为了尽可能地覆盖遥感影像上所有尺度的固废堆场,上述步骤S1实现多尺度剪裁的方法具体为:
S11、采用若干个尺寸的滑动窗口和步长对待检测区域的原始遥感影像进行裁剪;
S12、将裁剪后的各影像块缩放至全局推理多尺度检测模型的输入大小,得到各尺度下的影像块集合。
如图4所示,本实施例的步骤S2中的全局推理多尺度检测模型包括改进的SSD网络及在嵌合在SSD网络中各卷积层之间的GloRe单元;
SSD网络用于实现对输入影像的特征提取及多尺度目标检测;
GloRe单元用于对SSD网络中的特征进行全局推理,提高SSD网络对影像全局信息的表达能力。
对于改进的SSD模型,深度神经网络在进行层层卷积和池化的过程中,特征图分辨率逐渐降低同时特征表达也逐渐抽象,以至于深层特征图仅对大目标敏感,而浅层特征图保留的丰富细节信息则有助于对小目标的检测,因此充分利用深层抽象信息与浅层细节信息对于实现固废堆场这种尺度多变的目标检测具有重要意义。SSD(Single Shot MultiboxDetector)网络结合了来自多个不同分辨率特征图的检测结果,能够自然地处理不同尺度的目标,其结构如图3所示,该网络基于一种高质量的标准分类模型——VGG-16网络改进而来,其具体结构为:
SSD网络保留VGG-16网络的第一~第五卷积层,用于基础的特征提取,并将VGG-16网络第五层卷积层之后的全连接层修改为依次连接的第六卷积层、第七卷积层,同时在第七卷积层后依次添加依次连接的第八~第十一卷积层;
对于嵌合在SSD网络中的GloRe单元,根据SSD网络的特征计算方向和检测计算方向之后自由嵌合;如图4所示的“SSD的卷积层”即SSD网络进行的特征计算,特征会一层一层地向前计算(即特征计算方向,多边形箭头),在这个过程中层的尺寸会逐渐变小,越靠前层的特征(称为浅层特征)对小目标的关注度越高,而越靠后层的特征(称为深层特征)对大目标的关注度越高,因此为了能综合考虑多尺度目标的检测,SSD网络分别拿出以上几个不同尺寸的层的特征计算结果来进行目标边框检测计算(即检测计算方向,线形箭头)。GloRe可以嵌入在卷积层的上述两种方向之后,如果在特征计算方向嵌入GloRe单元(图4(a)),则此处进行的全局推理将影响该层的边框检测计算和后续几层的特征计算;如果在检测计算方向嵌入GloRe单元(图4(b)),则此处进行的全局推理将只影响该层的边框检测计算。为了探索何种嵌入方式是合理的,本文进行了不同层的嵌入实验,发现只在第四层的特征计算方向嵌入效果是最好的(如后面表3所示)。
在上述SSD网络中,第四卷积层及第七卷积层~第十一卷积层除进行特征向前计算外,其输出的六个不同分辨率的特征图还用于计算目标的边框、位置及类别置信度,并基于计算结果通过非极大值抑制操作(Non-Maximum Suppression,NMS),筛选出与输入图像中重叠度大于设定阈值的影像,作为全局推理多尺度检测模型的输出结果。
改进的SSD网络中的卷积操作主要关注图像的局部特征,对全局特征的利用需要经过更深层次的网络传递,此举不仅会加重模型的训练负担,同时过于复杂的网络结构会容易出现过拟合的现象。此外,SSD网络采用了类似于FasterR-CNN模型中的锚框进行目标搜索,即根据多个尺寸的窗口内的特征来回归计算目标的坐标和类别置信度,该操作使得模型对特征的利用更加局限,为了实现对特征图全局信息的充分利用,本实施例提出了一种可嵌入CNN模型中的全局推理(Global Reasoning,GloRe)单元,结构如图4所示,基于该结构,对于前一卷积层输入到GloRe单元的栅格形式的特征图X,GloRe单元对特征图X进行处理的方法为:
A1、采用卷积运算φ(X;Wφ)和卷积运算θ(X;Wθ)将特征图X映射到交互空间中的一个全连通图结构;
其中,全连通图结构中的每个节点即一个特征向量;
全连通图结构中节点的特征矩阵V为:
式中,Wφ为降维卷积运算的权重,Wθ为映射卷积运算的权重;
A2、采用图卷积运算在整个全连通图结构中传递每个节点的特征,并更新节点状态,进而在关系密切的节点之间构建强关联;
采用图卷积运算在整个全连通图结构V中传递每个节点的特征Z为:
Z=g(V;Wg)=((I-Ag)V)Wg
式中,g(·)为图卷积运算函数,Wg为图卷积的权重,I为单位矩阵,Ag为全连通图结构的邻接矩阵;
A3、采用θ(X;Wθ)的转置将节点之间具有强关联的全连通图结构从交互空间反向映射到栅格形式,得到栅格特征图Y;
A4、将栅格特征图Y的维度扩展至与特征图X大小相同后与特征图X叠加,作为GloRe单元的输出图像输入到下一层卷积层。
其中,栅格特征图Y为:
式中,θT(X;Wθ)为卷积运算θ(X;Wθ)的转置。
本实施例中的全局推理多尺度检测模型中的GloRe单元能够保证输出结果与输入大小匹配,因此该单元能够方便嵌入到改进的SSD网络中的任意卷积层之后,而不必对SSD网络的其他地方进行改动,SSD网络结构中的第四卷积层及第七卷积层~第十一卷积层除进行特征向前计算外,其输出的六个不同分辨率的特征图还用于计算目标的边框、位置及类别置信度,该机制决定了模型在以上几层卷积层之后分为特征计算与检测计算两个方向,同时考虑到第十、第十一卷积层的特征图分辨率大小分别为3×3和1×1,已不存在全局推理的必要。
在本申请的步骤S3中,如图5所示,在进行全局推理多尺度检测模型输出的影像多尺度合并时,其输出的影像中因窗口滑动而被切割的堆场目标与完整堆场目标并存,区别于一般目标在仅包含部件组件时识别置信度会降低的性质,固废堆场不存在固定的形状,即便是被切割的部分也能得到很高的置信度,因此基于置信度的排序的非极大值抑制(Non-Maximum Suppression,NMS),本实施例提供了一种基于模型输出影像的检测框尺寸的合并策略,基于此,本实施例的步骤S3具体为:
S31、将全局推理多尺度检测模型输出的检测框叠加整合至原始遥感图像对应的大范围影像中,并以置信度阈值Confthr对所有检测框进行筛选;
S32、基于筛选结果,确定属于同一目标的检测框b的检测框集合gi,并构建目标集合G={g1,...,gm};
S33、将集合G={g1,...,gm}中包含检测框b的个数不小于2的检测框集合筛选出来,并计算其最小外接矩形,得到的各尺度下的影像块组成影像块检测集合,进而得到对应的固废检测结果。
在步骤S32中,目标检测框集合G={g1,...,gm}中每个元素gi={b1,b2,...,bn}表示被判定为属于同一目标的检测框b的影像集合,i=1,2,...,m,m为集合G中元素总数;
元素gi的最小外包矩形与检测框b之间的重叠区域满足:
式中,B为检测框集合,Sg、Sb、Sin、Sun分别表示gi的最小外包矩形、检测框b、最小外包矩形与检测框b的重叠区域和并集的面积,Mergethr为合并阈值。
实施例2:
本实施例提供了基于上述方法进行固废检测的具体实验实例:
数据集:
基于生态环境部(Ministry of Ecology and Environment of the People’sRepublic of China,MEE)公布的固废堆场点位信息,获取对应的谷歌影像数据。从固废堆场点位清单中选择位于安徽省一处固废堆场数量较多、尺度丰富、场景复杂的区域作为研究区,大小为6400×6400像素,空间分辨率为0.26米。
本实施例所用训练和验证样本选自前述固废堆场点位清单,包括400×400、600×600、800×800像素三种大小,共计7099张影像,其中4970张用于训练,2129张用于验证,如表1所示。此外,为测试本文方法的泛化能力,本文另外从北京、河南等多地搜索固废堆场,制作了用于测试模型效果的测试集,包含400×400、600×600和800×800像素三种不同的尺寸。
表1样本数量统计
评价指标:
本文采用F1分数(F1-Score)和平均准确率(Average Precision,AP)两个指标对实验结果进行评估,具体计算方法如下:
其中,R、P、F1分别表示召回率、准确率和F1分数;TP表示被正确检测为固废堆场的检测框个数;FP表示被错误检测成固废堆场的检测框个数;FN表示未被检测到的固废堆场个数。随着置信度阈值的改变,召回率R与准确率P也会随之变化,二者形成的曲线被称为PR曲线,AP即在取特定的交并比阈值时PR曲线下方的面积,其值越高代表检测效果越好,通常取交并比阈值为0.5时的AP值作为评价指标,即AP50。
与经典方法对比:
本实施例将一层GloRe单元分别嵌入到Conv4的特征计算方向与检测计算方向之后,分别记为SSD+GloRe_Conv_1和SSD+GloRe_1,并与目标检测领域经典算法原始SSD300、FasterRCNN+FPN和YOLOv5对比,结果示例如图6所示,PR曲线如图7所示,评价指标对比如表2所示。可见,嵌入GloRe单元的模型在面对周边环境复杂的堆场时检测效果明显优于其他模型,而在特征计算方向嵌入GloRe单元(图6(a))要比在检测计算方向嵌入的模型(图6(b))表现更好。相比于原始SSD模型(图6(c))的检测效果,添加了GloRe单元的SSD模型(图6(a)、图6(b))的检测结果的漏检率更低,说明了GloRe单元的引入提高了模型对影像全局信息的利用,对固废堆场的精准检测起到了关键作用。此外,相比于本文方法,YOLOv5模型与Faster-RCNN+FPN模型均表现较差,其中YOLOv5检测结果中存在较多漏检目标(图6(e)),而Faster-RCNN+FPN模型误检问题严重(图6(d))。(图6中,(a)为SSD+GloRe_Conv_1、(b)为SSD+GloRe_1、(c)为SSD300、(d)为Faster-RCNN+FPN、(e)为YOLOv5)。
表2不同方法的评价指标
从大范围的检测效果(图8)来看,在面对固废堆场稀疏分布的场景(场景1)时,几种方法均表现较好,然而在面对复杂场景(场景2)时,SSD+GloRe_Conv_1检测效果最好,而其他几种方法存在比较严重的漏检问题,说明全局推理单元在处理场景复杂的情况时能够表现出更好的适应能力;此外,对于尺度变化较大的场景(场景3),尽管所有方法均存在一定的漏检问题,SSD+GloRe_Conv_1依然能够保持相对较好的检测效果,而FasterRCNN与YOLOv5则误检问题严重。(图8中,(a)为SSD+GloRe_Conv_1、(b)为SSD+GloRe_1、(c)为SSD300、(d)为Faster-RCNN+FPN、(e)为YOLOv5)。
嵌入不同数量的GloRe单元:
如图4模型结构所示,GloRe单元可以自由地嵌入在SSD模型的多个卷积层之后,具有很强的灵活性,本文通过在SSD模型的不同位置嵌入不同数量的GloRe单元,探讨其影响,评价指标如表3所示。可见,在特征计算方向嵌入GloRe单元整体表现优于在检测方向嵌入的模型,而二者在嵌入数量上均表现出随着嵌入GloRe单元的层数增多,检测效果有所下降的现象,该现象与Chen等(Chen et al,2019)在一般性深度学习任务实验中的测试表现相吻合,说明进行过多次的全局推理反而会影响特征提取表现。
表3嵌入不同数量GloRe单元的评价指标
本实施例针对研究区大幅影像上的固废堆场检测任务设计了多尺度优化策略,涉及多个实验参数,包括用于初步筛选检测框的置信度阈值Confthr和合并阈值Mergethr,为了确定合适的Confthr和Mergethr,本文采用SSD+GloRe_Conv_1模型通过设置不同参数进行敏感性分析实验,如图9所示,几乎取所有合并阈值时均是在置信度阈值为0.99时达到最大值,因此本文实验取Confthr=0.99;由图10可知,当以0.99作为置信度阈值时,取合并阈值为0.75时F1分数达到最大值,因此本文实验取Mergethr=0.75。
采用模型SSD+GloRe_Conv_1对经过多尺度裁剪后的研究区影像进行检测并按照多尺度合并策略及上文Confthr和Mergethr对检测结果处理后,最终结果如图11(b)所示;采用NMS方法对检测结果进行处理后,结果如图11(c)所示;可见与传统的NMS处理方法相比,多尺度合并将大量因滑动窗口裁剪而被分割的固废堆场合并,使检测结果与真值(图11(a))更加接近。
Claims (3)
1.基于深度学习与全局推理的遥感影像多尺度固废检测方法,其特征在于,包括以下步骤:
S1、获取待检测区域的原始遥感影像,并对其进行多尺度裁剪处理,获得各尺度下的影像块集合;
S2、通过全局推理多尺度检测模型对各尺度下的影像块集合中的影像进行检测,输出各尺度下的所有检测框;
S3、将全局推理多尺度检测模型输出的检测框整合到原始遥感图像对应的大范围影像中,并对各检测框进行合并,得到对应的固废检测结果;
所述步骤S1具体为:
S11、采用若干个尺寸的滑动窗口和步长对待检测区域的原始遥感影像进行裁剪;
其中,滑动窗口尺寸包括400×400、600×600、800×800;
S12、将裁剪后的各影像块缩放至全局推理多尺度检测模型的输入大小,得到各尺度下的影像块集合;
所述步骤S2中的全局推理多尺度检测模型包括SSD网络及在嵌合在SSD网络中的GloRe单元;
所述SSD网络用于实现对输入影像的特征提取及多尺度目标检测;
所述GloRe单元用于对SSD网络中的特征进行全局推理,提高SSD网络对影像全局信息的表达能力;
所述SSD网络保留VGG-16网络的第一~第五卷积层,并将VGG-16网络第五层卷积层之后的全连接层修改为依次连接的第六卷积层、第七卷积层,同时在第七卷积层后依次添加依次连接的第八~第十一卷积层;
在所述SSD网络中,对于输入到GloRe单元的栅格形式的特征图X,GloRe单元对特征图X进行处理的方法为:
A1、采用卷积运算φ(X;Wφ)和卷积运算θ(X;Wθ)将特征图X映射到交互空间中的一个全连通图结构;
A2、采用图卷积运算在整个全连通图结构中传递每个节点的特征,并更新节点状态,进而在关系密切的节点之间构建强关联;
A3、采用θ(X;Wθ)的转置将节点之间具有强关联的全连通图结构从交互空间反向映射到栅格形式,得到栅格特征图Y;
A4、将栅格特征图Y的维度扩展至与特征图X大小相同后与特征图X叠加,作为GloRe单元的输出图像;
所述步骤S3具体为:
S31、将全局推理多尺度检测模型输出的检测框叠加整合至原始遥感图像对应的大范围影像中,并以置信度阈值Confthr对所有检测框进行筛选;
S32、基于筛选结果,确定属于同一目标的检测框b的检测框集合gi,并构建目标集合G={g1,...,gm};
S33、将集合G={g1,...,gm}中包含检测框b的个数不小于2的检测框集合筛选出来,并计算其最小外接矩形,进而得到对应的固废检测结果;
在所述步骤S32中,所述目标检测框集合G={g1,...,gm}中每个元素gi={b1,b2,...,bn}表示被判定为属于同一目标的检测框b的影像集合,i=1,2,...,m,m为集合G中元素总数;
元素gi的最小外包矩形与检测框b之间的重叠区域满足:
式中,B为检测框集合,Sg、Sb、Sin、Sun分别表示gi的最小外包矩形、检测框b、最小外包矩形与检测框b的重叠区域和并集的面积,Mergethr为合并阈值。
2.根据权利要求1所述的基于深度学习与全局推理的遥感影像多尺度固废检测方法,其特征在于,在SSD网络中,所述第四卷积层及第七卷积层~第十一卷积层除进行特征向前计算外,其输出的六个不同分辨率的特征图还用于计算目标的边框、位置及类别置信度,并基于计算结果通过非极大值抑制操作,筛选出与输入图像中重叠度大于设定阈值的影像,作为全局推理多尺度检测模型的输出结果。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110114848.7A CN112836615B (zh) | 2021-01-26 | 2021-01-26 | 基于深度学习与全局推理的遥感影像多尺度固废检测方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110114848.7A CN112836615B (zh) | 2021-01-26 | 2021-01-26 | 基于深度学习与全局推理的遥感影像多尺度固废检测方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112836615A CN112836615A (zh) | 2021-05-25 |
CN112836615B true CN112836615B (zh) | 2021-11-09 |
Family
ID=75931994
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110114848.7A Active CN112836615B (zh) | 2021-01-26 | 2021-01-26 | 基于深度学习与全局推理的遥感影像多尺度固废检测方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112836615B (zh) |
Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110648364A (zh) * | 2019-09-17 | 2020-01-03 | 华侨大学 | 一种多维度空间固废视觉检测定位及识别方法与系统 |
CN110751075A (zh) * | 2019-10-12 | 2020-02-04 | 海南长光卫星信息技术有限公司 | 一种基于实例分割的遥感影像养殖塘检测方法 |
Family Cites Families (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2014179667A2 (en) * | 2013-05-03 | 2014-11-06 | Ecowastehub Corp. | Solid waste identification & segregation system |
CN108535201B (zh) * | 2018-02-12 | 2019-09-20 | 浙江大学 | 一种焚烧炉内生活垃圾组分实时检测装置和方法 |
CN110222638B (zh) * | 2019-06-05 | 2021-01-01 | 西安电子科技大学 | 一种基于素描图候选框策略和Fast R-CNN的飞机目标检测方法 |
CN110472627B (zh) * | 2019-07-02 | 2022-11-08 | 五邑大学 | 一种端到端的sar图像识别方法、装置及存储介质 |
CN111461127B (zh) * | 2020-03-30 | 2023-06-06 | 华南理工大学 | 基于一阶段目标检测框架的实例分割方法 |
-
2021
- 2021-01-26 CN CN202110114848.7A patent/CN112836615B/zh active Active
Patent Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110648364A (zh) * | 2019-09-17 | 2020-01-03 | 华侨大学 | 一种多维度空间固废视觉检测定位及识别方法与系统 |
CN110751075A (zh) * | 2019-10-12 | 2020-02-04 | 海南长光卫星信息技术有限公司 | 一种基于实例分割的遥感影像养殖塘检测方法 |
Also Published As
Publication number | Publication date |
---|---|
CN112836615A (zh) | 2021-05-25 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN110111345B (zh) | 一种基于注意力网络的3d点云分割方法 | |
CN110796186A (zh) | 基于改进的YOLOv3网络的干湿垃圾识别分类方法 | |
Liu et al. | Remote sensing image change detection based on information transmission and attention mechanism | |
CN106408030A (zh) | 基于中层语义属性和卷积神经网络的sar图像分类方法 | |
CN113505670B (zh) | 基于多尺度cam和超像素的遥感图像弱监督建筑提取方法 | |
CN109035300B (zh) | 一种基于深度特征与平均峰值相关能量的目标跟踪方法 | |
CN101276420A (zh) | 一种融合光谱信息和多点模拟空间信息的分类方法 | |
CN111160111B (zh) | 一种基于深度学习的人体关键点检测方法 | |
Despotovic et al. | Prediction and analysis of heating energy demand for detached houses by computer vision | |
CN115409069A (zh) | 村镇建筑的识别方法、分类方法、装置、电子设备和介质 | |
CN113487610B (zh) | 疱疹图像识别方法、装置、计算机设备和存储介质 | |
CN113869412B (zh) | 一种联合轻量级注意力机制和YOLOv3网络的图像目标检测方法 | |
CN113361496B (zh) | 一种基于U-Net的城市建成区统计方法 | |
CN112836615B (zh) | 基于深度学习与全局推理的遥感影像多尺度固废检测方法 | |
CN116012702A (zh) | 一种遥感影像场景级变化检测方法 | |
Wang et al. | FPA-DNN: a forward propagation acceleration based deep neural network for ship detection | |
CN114494999A (zh) | 一种双分支联合型目标密集预测方法及系统 | |
CN114283280A (zh) | 一种基于改进卷积神经网络的水面漂浮垃圾识别方法 | |
CN112949384A (zh) | 一种基于对抗性特征提取的遥感图像场景分类方法 | |
Busheska et al. | Machine Learning and Thermography Applied to the Detection and Classification of Cracks in Buildings | |
Liu et al. | Peaks fusion assisted early-stopping strategy for overhead imagery segmentation with noisy labels | |
CN117437234B (zh) | 基于图神经网络的航片地物分类与变化检测方法 | |
Swetha et al. | Visual Weather Analytics-Leveraging Image Recognition for Weather Prediction | |
CN117765361B (zh) | 一种基于残差神经网络检测双时相遥感影像中建筑物变化区域的方法 | |
Gao et al. | Evaluation on Computer Region Location Algorithm Based on Fully Convolutional Network |
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 |