CN115240078A - 一种基于轻量化元学习的sar图像小样本目标检测方法 - Google Patents

一种基于轻量化元学习的sar图像小样本目标检测方法 Download PDF

Info

Publication number
CN115240078A
CN115240078A CN202210723547.9A CN202210723547A CN115240078A CN 115240078 A CN115240078 A CN 115240078A CN 202210723547 A CN202210723547 A CN 202210723547A CN 115240078 A CN115240078 A CN 115240078A
Authority
CN
China
Prior art keywords
meta
module
lightweight
target detection
feature
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
CN202210723547.9A
Other languages
English (en)
Other versions
CN115240078B (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.)
CETC 38 Research Institute
Anhui University
Original Assignee
CETC 38 Research Institute
Anhui 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 CETC 38 Research Institute, Anhui University filed Critical CETC 38 Research Institute
Priority to CN202210723547.9A priority Critical patent/CN115240078B/zh
Publication of CN115240078A publication Critical patent/CN115240078A/zh
Application granted granted Critical
Publication of CN115240078B publication Critical patent/CN115240078B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V20/00Scenes; Scene-specific elements
    • G06V20/10Terrestrial scenes
    • G06V20/13Satellite images
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • G06N3/082Learning methods modifying the architecture, e.g. adding, deleting or silencing nodes or connections
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/764Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/77Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
    • G06V10/7715Feature extraction, e.g. by transforming the feature space, e.g. multi-dimensional scaling [MDS]; Mappings, e.g. subspace methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/77Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
    • G06V10/774Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/82Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V2201/00Indexing scheme relating to image or video recognition or understanding
    • G06V2201/07Target detection

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Evolutionary Computation (AREA)
  • Software Systems (AREA)
  • Health & Medical Sciences (AREA)
  • Artificial Intelligence (AREA)
  • Computing Systems (AREA)
  • General Health & Medical Sciences (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Multimedia (AREA)
  • Medical Informatics (AREA)
  • Databases & Information Systems (AREA)
  • General Engineering & Computer Science (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Biomedical Technology (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Data Mining & Analysis (AREA)
  • Molecular Biology (AREA)
  • Mathematical Physics (AREA)
  • Astronomy & Astrophysics (AREA)
  • Remote Sensing (AREA)
  • Image Analysis (AREA)

Abstract

本发明提供一种基于轻量化元学习的SAR图像小样本目标检测方法,包括:构建轻量化元特征提取器模块,根据轻量化元特征提取器模块,从输入的待查询SAR图像中提取出三个不同尺度的查询元特征;将带有标签的新类目标样本的支持图像输入到重加权模块中,输出三组与查询图像对应的重加权向量;构建基于transformer编码器的元特征聚合模块;将查询元特征和重加权向量通过元特征聚合模块重新校准;通过三个预测层分别对校准后的查询元特征和重加权向量进行预测,获得新类目标预测结果。本方法能在SAR图像目标新类仅有少量标注数据的情况下,达到更优的目标检测效能。

Description

一种基于轻量化元学习的SAR图像小样本目标检测方法
技术领域
本发明涉及目标探测技术领域,具体涉及一种基于轻量化元学习的SAR 图像小样本目标检测方法。
背景技术
合成孔径雷达(SAR)是遥感领域不可或缺的重要监测工具,它是一种主动微波成像传感器,也是获取SAR图像的主要途径,具有全天候、全天时成像和侦察能力。它可以提供不受天气和光照条件影响的高分辨率图像,已广泛应用于多个领域。近年来,随着机载SAR和星载SAR的发展,在SAR 目标检测方面进行了大量的研究。复杂场景下的多尺度SAR目标检测是其主要任务之一,仍然是一个巨大的挑战。
在人工智能发展的这几十年里,目标检测一直是研究人员重点研究的任务,并且已经取得了一系列的研究成果。在SAR图像目标检测领域中,已经有大量的模型和方法已经被研究人员开发出来用来检测SAR图像中的目标(例如坦克、舰船、飞机、桥梁等)。传统的SAR目标检测方法主要包括基于对比度信息的、基于几何和纹理特征的以及基于统计分析的。在现有SAR 目标的检测算法中,恒虚警率(CFAR)方法被称为最经典的检测算法,并且经常为研究人员所用。CFAR方法根据给定的虚警率和背景杂波的统计分布,计算出自适应阈值,然后通过像素强度与计算出的阈值进行比较,将目标像素与背景进行区分。该方法的性能在很大程度上取决于海杂波的统计建模和所选模型的参数估计。主要围绕这两个方面,提出了许多改进的方法。考虑到杂波的非均匀性,各种杂波模型,如对称稳定分布和广义伽马分布,都是基于非均匀分布来拟合变化的海况。然而,随着模型复杂性的增加,参数估计变得困难和耗时。Gao等人考虑了实际应用,试图在估计精度和速度之间取得良好的平衡。Xia等人通过将CNN和Transformer结合,使其能在SAR 图像上提取更加丰富的全局信息,最后在SSDD数据集上的准确率达到了比较高的水平,并且我们课题组一起参与构建了新的SAR多类目标检测数据集SMCDD,最后Xia等人在其验证了其方法的有效性。
此外,机器学习和GPU计算能力高速发展,使得卷积神经网络(cnn)在目标检测方面取得了显著突破。基于机器学习的方法具有强大的鲁棒特征提取和对象分类能力,与传统的使用人工设计特征的方法相比,深度神经网络可以从给定的数据中自动学习特征表示。不仅在光学图像还是在遥感SAR 图像领域中,都有大量基于CNN的目标检测方法被研究出来以解决各自领域的问题。现代基于CNN的检测器基本可分为两大类:基于锚的检测器和无锚的检测器。YOLO,SSD,Faster R-CNN等都是其中非常经典的目标检测器,已经经过很多工程和实验的检验,已经被广泛应用于各项目中。在SAR图像目标检测领域中,现在的研究大部分是基于计算机视觉的主流框架的。
尽管各种各样的深度学习的模型和方法在目标检测领域已经被提出来,但是这些方法首先都需要大规模、多样化的数据集来训练深度神经网络模型,尤其在军事领域中,现实情况可能不会允许我们能收集到那么多新的带有大量手工注释的SAR数据,比如敌方的飞机坦克等;其次这些方法都需要大量时间在收集的新的数据集上重新训练它们的参数。如果为了符合现实情况从样本量丰富的数据集中去抽取少量标注样本来训练网络模型的话,很容易出现过拟合现象,模型的泛化能力大大降低。所以,为了符合军事领域中只能从敌方获得的少量敌军SAR目标图像,急需一种特殊的学习机制来从少量的新类样本中学习到一定的特征知识。
发明内容
为解决上述问题,本发明提供了如下的技术方案。
一种基于轻量化元学习的SAR图像小样本目标检测方法,包括以下步骤:
构建SAR图像的小样本目标检测模型,其包括轻量化元特征提取器模块、基于transformer的元特征聚合模块、特征重加权模块和三个预测层;
将DarkNet53中每个block里面的3×3卷积替换为MobileNetV3的深度可分离卷积,近似残差的结构改成和MobileNetV2一样的具有线性瓶颈的逆残差结构,并引入SE模块,构建轻量化元特征提取器模块;
根据轻量化元特征提取器模块,从输入的待预测SAR图像中提取出三个不同尺度的查询元特征;
将带有标签的新类目标样本的支持图像输入到特征重加权模块中,输出三组与查询图像元特征对应的重加权向量;
构建基于transformer编码器的元特征聚合模块;将查询元特征和重加权向量通过元特征聚合模块重新校准;
通过三个预测层分别对校准后的查询元特征和重加权向量进行预测,获得SAR图像中的新类目标。
优选地,还包括:
构建基类训练集和新类训练集;基类训练集和新类训练集包括多个子集,每个所述子集包括一组来自同一类集的查询图像和一组带有标签并同一类集每类一个的支持图像;
通过基类训练集对SAR图像的小样本目标检测模型进行训练,输出SAR 图像的小样本目标检测的基本模型;
通过新类训练集对SAR图像的小样本目标检测的基本模型进行训练,输出最终的SAR图像的小样本目标检测模型。
优选地,还包括:
将Focal loss函数作为SAR图像的小样本目标检测模型训练时的分类损失函数。
优选地,所述轻量化元特征提取器模块的构建,包括以下步骤:
所述轻量化元特征提取器模块基于DarkNet53,将DarkNet53中每个 block里面的3×3卷积全部替换为MobileNetV3里使用的深度可分离卷积;将深度可分离卷积结构中使用H-swish激活函数代替深度卷积里的swish函数;
引入MobileNetV3里使用的SE模块;将SE模块中的expansion layer 的channel变为原来的1/4;将SE模块中的H-sigmoid代替sigmoid;所述 SE模块加在每个block里的深度卷积之后,点卷积之前;
将DarkNet53的block里近似残差的结构改成和MobileNetV2一样的具有线性瓶颈的逆残差结构,即先利用1×1卷积进行升维,再进行深度可分离卷积降维,并具有残差边。
优选地,所述重加权模块为轻量级CNN的重加权模块。
优选地,所述将查询元特征和重加权向量通过元特征聚合模块重新校准,包括以下步骤:
所述元特征聚合模块通过Transformer Encoder和信道乘积构成;
将N个目标类别的支持图像样和其标签将输入到重加权模块中,每次输入的一组支持图像都是从支持集中随机从这N类中抽取一张支持图像Ij和标签Mj所构成;
在经过重加权模块后,其被映射成一组特征向量,每类一个,表示为Vij= M(Ij,Mj);
将这组特征向量经过Transformer Encoder模块进行重新编码,得到V'ij,表示为V'ij=E(Vij);
由轻量化元特征提取器模块提取出的三组查询图像元特征Fi同样通过Transformer Encoder模块进行编码之后得到F'i,表示为F'i=E(Fi),将其通过信道乘积最后得到3组要输出到预测层进行预测的特征映射:
Figure RE-GDA0003820798450000051
优选地,还包括:
通过DIoU-NMS作为抑制准则对目标预测结果进行后处理。
本发明的有益效果:
本发明引入了最新光学遥感领域的FSODM方法作为基准框架;其次,为迎合SAR图像的独有特性和提升检测时效性,设计了一种轻量化的骨干,称为轻量化元特征提取器模块DarknetS;进一步地,构建了一种新的支持特征和查询特征的聚合模块,称为AggregationS,该模块通过Transformer将其编码到同一特征空间中,再通过信道乘法将特征聚合,这种设计可以更好提取出支持集里不同类之间的关联性与显著性,以及提高查询集在支持集里做查询时的准确性,减少误分类,让在大量基类数据集里学习到的知识更好的泛化,增强对新类的检测识别率。
本发明构建了新的SAR图像的小样本目标检测模型,解决了SAR图像军事领域中的小样本目标检测实际问题,为军事领域中识别检测敌军难以获取的军事目标提供了有效的方法。
附图说明
图1为FSODS模型总体结构示意图;
图2为SAR图像小样本目标检测基本思路示意图;
图3为深度可分离卷积示意图;
图4为SE模块流程图;
图5为Transformer Encoder结构图;
图6为四类目标示意图,图(a)、(e)为ship,(b)、(f)为airplane,(c)、(g) 为oil-tank,(d)、(h)为bridge。
具体实施方式
为了使本发明的目的、技术方案及优点更加清楚明白,以下结合附图及实施例,对本发明进行进一步详细说明。应当理解,此处所描述的具体实施例仅用以解释本发明,并不用于限定本发明。
实施例1
本发明的一种基于轻量化元学习的SAR图像小样本目标检测方法。图2 给出了SAR图像小样本目标检测方法的基本思路,包含两个阶段:基类训练阶段和新类微调阶段。在训练阶段,模型在易获取的带有大量标注的基类 SAR数据集上进行训练,然后在新类SAR数据微调阶段,从基类学习到的元知识能应用在只带有少量标注的新类SAR数据上,只需少量微调训练几次,就能其快速收敛,并识别新类,达到一定性能。如图1为模型的流程示意图,目标检测过程具体包括以下步骤:
S1:构建SAR图像的小样本目标检测模型模型FSODS,包括轻量化元特征提取器模块DarknetS、基于transformer的元特征聚合模块AggregationS、重加权模块和三个预测层。
S2:将DarkNet53中每个block里面的3×3卷积替换为MobileNetV3的深度可分离卷积,近似残差的结构改成和MobileNetV2的逆残差结构,并引入SE模块,构建轻量化元特征提取器模块;根据轻量化元特征提取器模块,从输入的待查询SAR图像中提取出三个不同尺度的查询元特征。
S3:将带有标签的新类目标样本的支持图像输入到重加权模块中,输出三组与查询图像对应的重加权向量。
S4:构建基于transformer的元特征聚合模块;将查询元特征和重加权向量通过元特征聚合模块重新校准,将来自查询集里样本的特征和支持集里样本的特征重新编码到相同的特征空间中,并突出每类的重要元特征以及每类之前的特征差异,这些元特征在对查询图像进行检测目标时信息是更加有效的。
S5:通过三个预测层分别对校准后的查询元特征和重加权向量进行预测,获得新类目标预测结果。
具体的:
1、轻量化的元特征提取器DarknetS
本发明重新设计了一种轻量化元特征提取器,称为DarknetS,它可以从输入的查询图像中提取出三个尺度的查询元特征。与FSODM使用DarkNet53 作为元特征提取器不同的是,本发明从实际工程应用和轻量化的角度出发,由于DarkNet提取的参数量和计算的flops过大,不利于真正实际应用中移植到嵌入式设备上,并且网络参数量的过大很容易造成训练样本过拟合,尤其在小样本这种样本数量过于悬殊的设定下,因此本发明在DarkNet53的基础上,借鉴MobileNetV3并针对SAR图像的特性,重新对其进行轻量化设计,本发明称其为DarknetS。
DarknetS首先将DarkNet53中每个block里面的3×3卷积全部替换为MobileNetV3里使用的深度可分离卷积。通道分离卷积是MobileNet系列的主要特点,也是其发挥轻量级作用的主要因素。如图3,通道可分离卷积分为两个过程:1.channel方向通道可分离卷积;2.正常的1×1卷积输出指定的channel数。
其次,DarknetS也引入了MobileNetV3里使用的SE通道注意力模块,该模块的核心思想是通过显示地建模网络卷积特征通道之间地相互依赖关系,以提高网络所产生地质量。具体来说,就是通过学习来自动获取到每个特征通道地重要程度,然后依照这一结果去提升有用的特征并抑制对当前任务用处不大的特征。所以,通过该模块,网络可以学习全局信息来选择性地强调SAR图像中的信息性特征,并抑制不太有用地SAR图像噪声特征。值得注意的是,这里的SE模块相比于传统的SE模块利用1×1卷积实现的FC 操作,本质上和FC是一样的,并且利用计算量更小的H-sigmoid代替sigmoid 操作:
Figure RE-GDA0003820798450000081
因为SE结构会消耗一定的时间,所以本发明在含有SE的结构中,将 expansionlayer的channel变为原来的1/4,这样即提高了精度,同时还没有增加时间消耗。本发明提出的FSODMS模型同样将改进后地SE模块加在每个block里的深度卷积之后点卷积之前。SE模块具体操作流程如图4所示。
并且本发明同样在FSODM的深度可分离卷积的结构中使用H-swish激活函数代替深度卷积里的swish函数,减少运算量,提高性能。H-swish并表示如下:
H-swish=x·H-sigmoid (2)
最后,本发明将DarkNet53的block里近似残差的结构改成和 MobileNetV2一样的具有线性瓶颈的逆残差结构(the inverted residual with linear bottleneck),即先利用1×1卷积进行升维,再进行深度可分离卷积降维,并具有残差边。MobileNetV2是将bottleneck结构变成了纺锤型,即resnet 是先缩小为原来的1/4,再放大,他是放大到原来的6倍,再缩小。而由于计算资源有限,本发明的FSODS是先放大原来的4倍,再缩小。总体的DarknetS如表1所示。
表1 DarknetS网络结构细节
Figure RE-GDA0003820798450000091
Figure RE-GDA0003820798450000101
2、特征聚合模块:AggregationS
基于transformer的元特征聚合模块是用来聚合两组元特征。一组是通过轻量级CNN的重加权模块从带有标签的支持图像中提取到支持元特征,它将每个支持图像映射到一组重加权向量,每类对应一个重加权向量。另一组是通过本发明的DarknetS从查询图像中提取出来的查询元特征,重加权向量将用于调整从查询图像中提取的查询元特征,突出每个查询特征的有用信息,从而更有利于在查询图像进行目标查询。
假设本发明有N个目标类别的支持图像样本,这N个支持图像和其标签将输入到特征重加权模块里。每次输入的一组支持图像都是从支持集中随机从这N类中抽取一张支持图像Ij和标签Mj所构成。在经过特征重加权模块后,其被映射成一组特征向量,每类一个,表示为Vij=M(Ij,Mj),接着这组特征向量经过本发明设计的图5所示的Transformer Encoder模块进行重新编码,得到V'ij,表示为V'ij=E(Vij)。由DarknetS提取出的三组查询图像元特征 Fi同样通过Transformer Encoder模块进行编码之后得到F'i,表示为F'i=E(Fi)。因为每个重加权向量与相应的元特征具有相同的维度,所以接着本发明再将其通过信道乘积最后得到3组要输出到预测层进行预测的特征映射:
Figure RE-GDA0003820798450000111
如图1所示Transformer Encoder和信道乘积构成了本发明FSODMS的AggregationS模块。Transformer encoder block可以捕获全局信息和丰富的上下文信息。每个Transformer encoder block包含2个子层。第1子层为 multi-head attentionlayer,第2子层(MLP)为全连接层。每个子层之间使用残差连接。Transformer encoderblock增加了捕获不同局部信息的能力。它还可以利用自注意力机制来挖掘特征表征潜能。因此本发明设计的AggregationS 模块可以突出SAR图像里目标的特征信息,减弱图像背景里的噪声信息,并且捕获不同类之间的类间关联,大大减少了相似类的误分类,增强了对新类的知识泛化。
3、Focal loss
小样本目标检测最大的问题通常就是样本不均衡,基类和新类的数量往往差的很多,所以通常经典检测算法训练出来的模型往往容易过拟合并且泛化能力很弱,因此本发明为了缓解这种样本不均衡,泛化能力弱的问题,本发明将FSODM里原本用交叉损失熵作为分类损失的函数更换为Focal loss 函数。Focal loss就是专门为一阶段检测算法设计的,将易区分负例的loss 权重降低,使得网络不会被大量的负样本带偏。Focal loss函数的公式如下所示:
FL(pt)=-(1-pt)γlog(pt) (4)
其中:
Figure RE-GDA0003820798450000121
γ为常数,当其为0时,FL就和普通的交叉熵损失函数相同。
4、DIou-NMS
本发明的FSODM为了在后处理端预测更加准确,将传统的NMS替换成DIoU-NMS。因为在传统NMS中,IoU指标常用于抑制冗余检测盒,其中重叠区域是唯一因素,对于遮挡情况经常产生错误抑制。IoU公式表示如下:
Figure RE-GDA0003820798450000122
其中:
Bgt=(xgt,ygt,wgt,hgt) (7)
代表的是真实框。
而DIoU-NMS将DIoU作为NMS的准则,因为在抑制准则中不仅应考虑重叠区域,而且还应考虑两个box之间的中心点距离,而DIoU就是同时考虑了重叠区域和两个box的中心距离。对于score最高的预测box M,可以将DIoU-NMS的si公式定义为:
Figure RE-GDA0003820798450000131
5、训练和推理
为了满足在小样本目标检测设定下训练,在本发明的FSODS训练时,训练集被本发明重新划分,分成了支持集(S)和查询集(Q)两个子集。其中支持集是按训练集里的目标类别进行划分。如果训练集里有N类目标,则本发明将其分为N组,i=1,2,3,…,N。每个支持图像都会带着自己的真实标签输入。支持集如下公式表示:
Figure RE-GDA0003820798450000132
查询集为包含一组查询图像及其标注(A):
Q={(I,A)} (10)
训练集划分成很多这样的集,每一个可以表示为:
Ej=Qj∪Sj (11)
查询图像和支持图像分别输入到图1所示的特征提取器和重加权模块里。
在小样本目标检测的设定下,本发明需要从数据集中选定某些类作为小样本类也就是新类,所以数据集将被本发明人为假设的划分为基类和新类。基类的样本不做削减,保持原来有着大量标签训练的状态。基类数据在本发明的模型上训练出一个性能比较好的基本模型,然后开启新类,在新类上开启新的训练任务,新类是从原始的数据中随机抽取的少数几个带有标签的样本,用来作为现实设定的小样本类,只能获取这么多的样本。
因此,本发明的训练过程可以分为两步:第一步是在有着大量数据的基类上进行训练,时间将会较长,在后续的训练中本发明不需要再重新在基类上训练一遍;第二步是在第一步的基础上,用训练出的模型开启新类,只需要训练很少的训练次数就能达到较好的性能。
6、数据集
本发明在一个公共基准SAR目标检测数据集上和一些非公开的SAR目标检测数据集上评估了本发明的模型性能,并且本发明将本发明提出的 FSODS和FSODM以及经典模型YOLOV3进行了比较,都显示出了FSODS 性能的优越性。
本次评估的公开SAR目标检测数据集是即将发布的SAR数据集 SMCDD,其数据都是我国首颗商用合成孔径雷达卫星采集而来。中国电子科技集团公司第38研究所、长沙天仪空间科学技术研究院有限公司等单位一起参与研制了HISEA-1卫星。HISEA-1卫星就一直能够提供稳定的数据服务,HISEA-1卫星不仅能执行多项成像任务而且能够为本发明提供稳定的数据服务,在此前本发明已经获得了两千多张条纹图像、七百多张聚焦图像和以及300张左右的扫描图像。其中,本发明采用了HISEA-1捕获的SAR 在复杂场景下的大图数据来构建SMCDD数据集的切片数据。本发明最后选取了包括:ship、airplane、bridge and oil-tank四种目标类型数据。为了方便训练和测试评估,本发明将大图分割成256、512、1024和2048个尺寸的小图。数据集最终通过本发明课题组的共同筛选和清洗以后,最后包含了1851座桥梁,39858艘船舶,12319个油罐,6368架飞机。SMCDD数据集四类样例如图6所示。除了公开的数据集,本发明还在未公开的SAR目标检测数据集上验证了本发明的模型在近似类型目标上检测的有效性,本发明选取了公开的海思数据集里的民船和未公开的数据里的某种军舰组成了一个新的数据集,称为SFSD。此外,本发明还选取了公开的海思数据集里的民船和未公开数据集的坦克和装甲车构成另一个新的数据集,本发明称为TFSD。
7、实验设置
为了评估本发明的FSODS模型在SAR图像上的小样本情况下的目标检测性能,本发明从即将公开的数据集SMCDD四类数据里总共随机抽取了 825张图片构成小样本数据集SMCDD-FS,并结合每类的照片的数量大致按照7:3的比例将其随机划分成训练集和验证集。训练集有572张SAR图像,验证集里有253张SAR图像。图像大小基本上为256×256,也有少数桥梁为1024×1024和2048×2048。并且本发明结合数量以及之前实验室在此数据集上的做的实验考虑,飞机目标的图片数量最少,并且基本上在图上大小仅为几个像素点大小,非常密集,一张256×256大小的图上可能甚至有几百架飞机,所以在基准上的性能并不是特别好,所以本次在SMCDD-FS数据集里选取飞机作为小样本类,船、油箱、桥梁作为基类训练。而由公开数据集和非公开数据集组成的数据集SFSD和TFSD里本发明均选取了民船作为基类,军舰作为SFSD里的新类,坦克和装甲车作为TFSD里的新类。
本发明在小样本条件下的识别检测能力明显优于其它经典检测模型,提升了小样本目标检测的精度。本方法的轻量化设计为本模型部署到机载或星载提供了可能性。
以上仅为本发明的较佳实施例而已,并不用以限制本发明,凡在本发明的精神和原则之内所作的任何修改、等同替换和改进等,均应包含在本发明的保护范围之内。

Claims (7)

1.一种基于轻量化元学习的SAR图像小样本目标检测方法,其特征在于,包括以下步骤:
构建SAR图像的小样本目标检测模型,其包括轻量化元特征提取器模块、基于transformer的元特征聚合模块、特征重加权模块和三个预测层;
将DarkNet53中每个block里面的3×3卷积替换为MobileNetV3的深度可分离卷积,近似残差的结构改成和MobileNetV2一样的具有线性瓶颈的逆残差结构,并引入SE模块,构建轻量化元特征提取器模块;
根据轻量化元特征提取器模块,从输入的待预测SAR图像中提取出三个不同尺度的查询元特征;
将带有标签的新类目标样本的支持图像输入到特征重加权模块中,输出三组与查询图像元特征对应的重加权向量;
构建基于transformer编码器的元特征聚合模块;将查询元特征和重加权向量通过元特征聚合模块重新校准;
通过三个预测层分别对校准后的查询元特征和重加权向量进行预测,获得SAR图像中的新类目标。
2.根据权利要求1所述的基于轻量化元学习的SAR图像小样本目标检测方法,其特征在于,还包括:
构建基类训练集和新类训练集;基类训练集和新类训练集包括多个子集,每个所述子集包括一组来自同一类集的查询图像和一组带有标签并同一类集每类一个的支持图像;
通过对SAR图像的小样本目标检测模型在基类训练集进行训练,输出SAR图像的小样本目标检测的基类检测基本模型;
通过对SAR图像的小样本目标检测的基本模型在新类训练集上进行微调训练,输出最终的SAR图像的小样本目标检测模型。
3.根据权利要求2所述的基于轻量化元学习的SAR图像小样本目标检测方法,其特征在于,还包括:
将Focal loss函数作为SAR图像的小样本目标检测模型训练时的分类损失函数。
4.根据权利要求1所述的基于轻量化元学习的SAR图像小样本目标检测方法,其特征在于,所述轻量化元特征提取器模块的构建,包括以下步骤:
所述轻量化元特征提取器模块基于DarkNet53,将DarkNet53中每个block里面的3×3卷积全部替换为MobileNetV3里使用的深度可分离卷积;将深度可分离卷积结构中使用H-swish激活函数代替深度卷积里的swish函数;
引入MobileNetV3里使用的SE模块;将SE模块中的expansion layer的channel变为原来的1/4;将SE模块中的H-sigmoid代替sigmoid;所述SE模块加在每个block里的深度卷积之后,点卷积之前;
将DarkNet53的block里近似残差的结构改成和MobileNetV2一样的具有线性瓶颈的逆残差结构,即先利用1×1卷积进行升维,再进行深度可分离卷积降维,并具有残差边。
5.根据权利要求1所述的基于轻量化元学习的SAR图像小样本目标检测方法,其特征在于,所述特征重加权模块为轻量级CNN的重加权模块。
6.根据权利要求5所述的基于轻量化元学习的SAR图像小样本目标检测方法,其特征在于,所述将查询元特征和重加权向量通过元特征聚合模块重新校准,包括以下步骤:
所述将查询元特征和重加权向量通过元特征聚合模块重新校准,包括以下步骤:
所述元特征聚合模块通过Transformer Encoder和信道乘积构成;
将N个目标类别的支持图像样和其标签将输入到重加权模块中,每次输入的一组支持图像都是从支持集中随机从这N类中抽取一张支持图像Ij和标签Mj所构成;
在经过重加权模块后,其被映射成一组特征向量,每类一个,表示为Vij=M(Ij,Mj);
将这组特征向量经过Transformer Encoder模块进行重新编码,得到V′ij,表示为V′ij=E(Vij);
由轻量化元特征提取器模块提取出的三组查询图像元特征Fi同样通过TransformerEncoder模块进行编码之后得到F′i,表示为F′i=E(Fi),将其通过信道乘积最后得到3组要输出到预测层进行预测的特征映射:
Figure FDA0003712531620000031
7.根据权利要求1所述的基于轻量化元学习的SAR图像小样本目标检测方法,其特征在于,还包括:
通过DIoU-NMS作为抑制准则对目标预测结果进行后处理。
CN202210723547.9A 2022-06-24 2022-06-24 一种基于轻量化元学习的sar图像小样本目标检测方法 Active CN115240078B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202210723547.9A CN115240078B (zh) 2022-06-24 2022-06-24 一种基于轻量化元学习的sar图像小样本目标检测方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202210723547.9A CN115240078B (zh) 2022-06-24 2022-06-24 一种基于轻量化元学习的sar图像小样本目标检测方法

Publications (2)

Publication Number Publication Date
CN115240078A true CN115240078A (zh) 2022-10-25
CN115240078B CN115240078B (zh) 2024-05-07

Family

ID=83670273

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202210723547.9A Active CN115240078B (zh) 2022-06-24 2022-06-24 一种基于轻量化元学习的sar图像小样本目标检测方法

Country Status (1)

Country Link
CN (1) CN115240078B (zh)

Cited By (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116994116A (zh) * 2023-08-04 2023-11-03 北京泰策科技有限公司 基于自注意力模型与yolov5的目标检测方法及系统

Citations (12)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2020073951A1 (zh) * 2018-10-10 2020-04-16 腾讯科技(深圳)有限公司 用于图像识别的模型的训练方法、装置、网络设备和存储介质
US20200265254A1 (en) * 2019-02-19 2020-08-20 Fujitsu Limited Object recognition method, apparatus and network
CN112818903A (zh) * 2020-12-10 2021-05-18 北京航空航天大学 一种基于元学习和协同注意力的小样本遥感图像目标检测方法
CN113240039A (zh) * 2021-05-31 2021-08-10 西安电子科技大学 基于空间位置特征重加权的小样本目标检测方法及系统
CN113673420A (zh) * 2021-08-19 2021-11-19 清华大学 一种基于全局特征感知的目标检测方法及系统
CN113936300A (zh) * 2021-10-18 2022-01-14 微特技术有限公司 建筑工地人员识别方法、可读存储介质和电子设备
CN114067217A (zh) * 2021-09-17 2022-02-18 北京理工大学 基于非下采样分解转换器的sar图像目标识别方法
CN114186622A (zh) * 2021-11-30 2022-03-15 北京达佳互联信息技术有限公司 图像特征提取模型训练方法、图像特征提取方法和装置
CN114359283A (zh) * 2022-03-18 2022-04-15 华东交通大学 基于Transformer的缺陷检测方法和电子设备
CN114511703A (zh) * 2022-01-21 2022-05-17 苏州医智影科技有限公司 面向分割任务的融合Swin Transformer与UNet的迁移学习方法和系统
CN114529821A (zh) * 2022-02-25 2022-05-24 盐城工学院 一种基于机器视觉的海上风电安全监测与预警方法
CN114579794A (zh) * 2022-03-31 2022-06-03 西安建筑科技大学 特征一致性建议的多尺度融合地标图像检索方法及系统

Patent Citations (12)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2020073951A1 (zh) * 2018-10-10 2020-04-16 腾讯科技(深圳)有限公司 用于图像识别的模型的训练方法、装置、网络设备和存储介质
US20200265254A1 (en) * 2019-02-19 2020-08-20 Fujitsu Limited Object recognition method, apparatus and network
CN112818903A (zh) * 2020-12-10 2021-05-18 北京航空航天大学 一种基于元学习和协同注意力的小样本遥感图像目标检测方法
CN113240039A (zh) * 2021-05-31 2021-08-10 西安电子科技大学 基于空间位置特征重加权的小样本目标检测方法及系统
CN113673420A (zh) * 2021-08-19 2021-11-19 清华大学 一种基于全局特征感知的目标检测方法及系统
CN114067217A (zh) * 2021-09-17 2022-02-18 北京理工大学 基于非下采样分解转换器的sar图像目标识别方法
CN113936300A (zh) * 2021-10-18 2022-01-14 微特技术有限公司 建筑工地人员识别方法、可读存储介质和电子设备
CN114186622A (zh) * 2021-11-30 2022-03-15 北京达佳互联信息技术有限公司 图像特征提取模型训练方法、图像特征提取方法和装置
CN114511703A (zh) * 2022-01-21 2022-05-17 苏州医智影科技有限公司 面向分割任务的融合Swin Transformer与UNet的迁移学习方法和系统
CN114529821A (zh) * 2022-02-25 2022-05-24 盐城工学院 一种基于机器视觉的海上风电安全监测与预警方法
CN114359283A (zh) * 2022-03-18 2022-04-15 华东交通大学 基于Transformer的缺陷检测方法和电子设备
CN114579794A (zh) * 2022-03-31 2022-06-03 西安建筑科技大学 特征一致性建议的多尺度融合地标图像检索方法及系统

Non-Patent Citations (9)

* Cited by examiner, † Cited by third party
Title
ANDREW HOWARD, AT EL.: ""Searching for MobileNetV3"", 《ARXIV:1905.02244V5 [CS.CV] 20 NOV 2019》, 20 November 2019 (2019-11-20), pages 1 - 11 *
ANDREW HOWARD; AT EL.: ""Searching for MobileNetV3"", 《ARXIV:1905.02244V5 [CS.CV]》, 20 November 2019 (2019-11-20), pages 1 - 11 *
BINGYI KANG, AT EL.: ""Few-shot Object Detection via Feature Reweighting"", 《2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION》, 31 December 2019 (2019-12-31), pages 8419 - 8428 *
MINGXING TAN, AT EL.: "MnasNet: Platform-Aware Neural Architecture Search for Mobile", 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, 31 December 2019 (2019-12-31) *
ZHENG ZHOU, AT EL.: ""FSODS: A Lightweight Metalearning Method for Few-Shot Object Detection on SAR Images"", 《IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING》, 21 July 2022 (2022-07-21), pages 1 - 17, XP011916037, DOI: 10.1109/TGRS.2022.3192996 *
刘宇等: "《智能搜索和推荐系统:原理、运算与应用》", vol. 1, 31 January 2021, 机械工业出版社, pages: 120 - 121 *
许志鹏,桑庆兵: ""基于轻量级 YOLOv3 的拉链缺陷检测系统设计与实现"", 《图形图像》, 15 September 2020 (2020-09-15), pages 33 - 39 *
邱锡鹏: "《神经网络与深度学习》", vol. 1, 31 July 2020, 中国电影出版社印刷厂, pages: 91 *
陈晨;王亚立;乔宇;: "任务相关的图像小样本深度学习分类方法研究", 集成技术, no. 03, 15 May 2020 (2020-05-15) *

Cited By (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116994116A (zh) * 2023-08-04 2023-11-03 北京泰策科技有限公司 基于自注意力模型与yolov5的目标检测方法及系统
CN116994116B (zh) * 2023-08-04 2024-04-16 北京泰策科技有限公司 基于自注意力模型与yolov5的目标检测方法及系统

Also Published As

Publication number Publication date
CN115240078B (zh) 2024-05-07

Similar Documents

Publication Publication Date Title
CN112308019B (zh) 基于网络剪枝和知识蒸馏的sar舰船目标检测方法
Chen et al. Target classification using the deep convolutional networks for SAR images
Wang et al. Study on the combined application of CFAR and deep learning in ship detection
CN109558803B (zh) 基于卷积神经网络与np准则的sar目标鉴别方法
CN114926693A (zh) 基于加权距离的sar图像小样本识别方法及装置
CN116342894A (zh) 基于改进YOLOv5的GIS红外特征识别系统及方法
CN112084897A (zh) 一种gs-ssd的交通大场景车辆目标快速检测方法
CN114331950A (zh) 基于稠密连接稀疏激活网络的sar图像舰船检测方法
Chen et al. Subcategory-aware feature selection and SVM optimization for automatic aerial image-based oil spill inspection
CN115240078B (zh) 一种基于轻量化元学习的sar图像小样本目标检测方法
CN118097304A (zh) 一种基于神经架构搜索的声纳图像分类方法
Chang et al. MLSDNet: Multi-class Lightweight SAR Detection Network Based on Adaptive scale distribution attention
Huang et al. EST-YOLOv5s: SAR image aircraft target detection model based on improved YOLOv5s
Huang et al. Ship detection based on YOLO algorithm for visible images
CN117710841A (zh) 一种无人机航拍图像的小目标检测方法、装置
Chen et al. Ship tracking for maritime traffic management via a data quality control supported framework
CN116030300A (zh) 一种用于零样本sar目标识别的渐进式域自适应识别方法
Da et al. Remote sensing image ship detection based on improved YOLOv3
CN114913504A (zh) 一种融合自注意力机制的遥感图像的车辆目标识别方法
Bao et al. Attention-guided multi-modal and multi-scale fusion for multispectral pedestrian detection
Wang et al. A Prediction and Prior Information Guided SAR Ship Detection Method
Wu et al. G-NAS: Generalizable Neural Architecture Search for Single Domain Generalization Object Detection
Hu et al. Ship Identification Based on Improved SSD
Gou et al. A Semantic Consistency Feature Alignment Object Detection Model Based on Mixed-Class Distribution Metrics
Lu et al. Few-Shot Incremental Object Detection in Aerial Imagery via Dual-Frequency Prompt

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