CN112766087A - 一种基于知识蒸馏的光学遥感图像舰船检测方法 - Google Patents
一种基于知识蒸馏的光学遥感图像舰船检测方法 Download PDFInfo
- Publication number
- CN112766087A CN112766087A CN202110002376.6A CN202110002376A CN112766087A CN 112766087 A CN112766087 A CN 112766087A CN 202110002376 A CN202110002376 A CN 202110002376A CN 112766087 A CN112766087 A CN 112766087A
- Authority
- CN
- China
- Prior art keywords
- network
- student
- teacher
- detection
- student network
- 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.)
- Pending
Links
- 238000001514 detection method Methods 0.000 title claims abstract description 95
- 230000003287 optical effect Effects 0.000 title claims abstract description 30
- 238000013140 knowledge distillation Methods 0.000 title claims abstract description 18
- 238000012549 training Methods 0.000 claims abstract description 23
- 238000012360 testing method Methods 0.000 claims abstract description 16
- 238000000034 method Methods 0.000 claims abstract description 9
- 238000004364 calculation method Methods 0.000 claims abstract description 5
- 230000008569 process Effects 0.000 claims abstract description 4
- 238000010586 diagram Methods 0.000 claims description 23
- 238000000605 extraction Methods 0.000 claims description 9
- 230000006870 function Effects 0.000 claims description 9
- 238000013527 convolutional neural network Methods 0.000 claims description 6
- 230000007246 mechanism Effects 0.000 claims description 6
- 238000005457 optimization Methods 0.000 claims description 6
- 238000005070 sampling Methods 0.000 claims description 6
- 208000037170 Delayed Emergence from Anesthesia Diseases 0.000 claims description 3
- 230000003044 adaptive effect Effects 0.000 claims description 3
- 238000011176 pooling Methods 0.000 claims description 3
- 230000002401 inhibitory effect Effects 0.000 claims description 2
- 230000004927 fusion Effects 0.000 description 3
- 230000011218 segmentation Effects 0.000 description 2
- WDLRUFUQRNWCPK-UHFFFAOYSA-N Tetraxetan Chemical compound OC(=O)CN1CCN(CC(O)=O)CCN(CC(O)=O)CCN(CC(O)=O)CC1 WDLRUFUQRNWCPK-UHFFFAOYSA-N 0.000 description 1
- 238000007792 addition Methods 0.000 description 1
- 238000013528 artificial neural network Methods 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000003032 molecular docking Methods 0.000 description 1
- 238000012544 monitoring process Methods 0.000 description 1
- 238000007781 pre-processing Methods 0.000 description 1
- 230000009467 reduction Effects 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000000926 separation method Methods 0.000 description 1
- 230000003595 spectral effect Effects 0.000 description 1
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
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
- G06F18/2415—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches based on parametric or probabilistic models, e.g. based on likelihood ratio or false acceptance rate versus a false rejection rate
-
- 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
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/40—Scenes; Scene-specific elements in video content
- G06V20/41—Higher-level, semantic clustering, classification or understanding of video scenes, e.g. detection, labelling or Markovian modelling of sport events or news items
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Software Systems (AREA)
- Computational Linguistics (AREA)
- General Health & Medical Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Mathematical Physics (AREA)
- Computing Systems (AREA)
- Health & Medical Sciences (AREA)
- Multimedia (AREA)
- Bioinformatics & Computational Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Molecular Biology (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Evolutionary Biology (AREA)
- Remote Sensing (AREA)
- Astronomy & Astrophysics (AREA)
- Probability & Statistics with Applications (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种基于知识蒸馏的光学遥感图像舰船检测方法,属于遥感图像目标检测领域,解决了现有技术中模型参数量大、计算复杂度高导致的舰船检测速度低的问题。本发明的技术方案包括:收集或下载公开的光学遥感图像舰船检测数据集;搭建基于知识蒸馏的舰船检测网络框架,采用Hint损失使得训练过程中学生网络能充分模仿教师网络的行为;联合训练教师网络和学生网络,保存训练后的学生网络的模型参数;利用学生网络进行测试,得到最后的舰船检测结果。本发明利用检测准确率较高的教师网络指导学生网络的训练,在保持学生网络原有检测速度的同时提高学生网络的检测准确率,从而实现快速精确的光学遥感图像舰船检测。
Description
技术领域
本发明属于遥感图像目标检测技术领域,具体涉及一种基于知识蒸馏的光学遥感图像舰船检测方法。
背景技术
舰船目标作为海上监测和战时打击的重点目标,对其进行精确检测可有效判断敌方的军事动向,为打赢战争、捍卫领海、保证国家和人民的生命安全提供重要的信息支持。随着光学遥感技术的迅猛发展,光学遥感卫星图像的空间分辨率和光谱分辨率都有了极大的提高,更加丰富的图像细节和目标信息使得利用光学遥感图像进行舰船检测得到越来越多的重视。现有技术中,基于深度卷积神经网络的舰船检测技术已经成为遥感图像目标检测领域的重要研究方向。
申请号(CN201811571859.2)的专利文献提出了一种基于深度残差密集网络的光学遥感图像舰船检测方法,在海陆分离和提取舰船候选区之后,利用由局部残差密集块组成的深度残差密集网络对切片进行分类,得到最后的检测结果。该方法可以克服天气因素和复杂环境对于舰船检测的干扰,极大地提高了舰船检测的精度。
申请号(CN201910674642.2)的专利文献提出了一种基于海陆分割和特征金字塔网络的港口停靠舰船检测方法,利用基于候选区提取网络的目标检测框架和特征金字塔网络,构建舰船检测模型,对初始分割的切片数据进行检测,得到待处理遥感影像的舰船检测结果。该方法能够应对多尺度舰船目标检测问题,有效降低了舰船检测的虚警率。
上述方法虽然提高了光学遥感图像的舰船检测精度,但是采用的卷积神经网络层数较多,导致了模型参数量和计算复杂度的急剧增加,忽略了对舰船检测速度的影响,在实际应用中难以实现实时的舰船检测。
发明内容
针对现有技术的不足,本发明提出一种基于知识蒸馏的光学遥感图像舰船检测方法,将知识蒸馏技术应用于光学遥感图像舰船检测领域,结合深度卷积神经网络,旨在提高舰船检测准确率的同时降低模型的参数量和复杂度,提升模型的运行速度,从而实现快速精确的光学遥感图像舰船检测。
为实现上述目的,本发明的技术方案包含以下步骤:
步骤1,下载公开的舰船数据集,并划分训练集和测试集;
步骤2,构建基于知识蒸馏的舰船检测网络,包含教师网络和学生网络,通过设计损失函数,让层数多检测准确率高的教师网络指导学生网络的学习,在保持学生网络原本检测速度的同时提高学生网络检测的准确率;
所述教师网络和学生网络均由骨干网络、特征加权模块、分类和边框回归子网络三部分组成,将输入图片同时输入到教师网络和学生网络中,所述骨干网络用于特征提取,特征加权模块用于进一步提取更具判别性的特征,并且抑制光学遥感图像中的背景干扰,最后由分类和边框回归子网络输出每个目标类别的概率以及回归得到目标的位置坐标,根据教师网络和学生网络的检测结果分别计算交叉熵损失和回归损失;
步骤3,利用训练集对上述教师网络和学生网络进行联合训练,保存训练后的学生网络模型参数;
在训练过程中,根据真实的舰船检测标签与学生网络检测出的结果计算Softmax和SmoothL1损失,通过反向传播,优化学生网络的模型参数;
步骤4,测试阶段利用学生网络对测试集的所有图片进行测试,计算舰船检测的准确率,得到检测结果。
进一步的,所述教师网络和学生网络选用不同的骨干网络用于特征提取,其中教师网络选用卷积层层数较多且检测精度较高的骨干网络,学生网络选用卷积层层数较少且检测速度较快的骨干网络。
进一步的,教师网络中选用ResNet101作为骨干网络,学生网络中选用Shufflenet作为骨干网络。
进一步的,所述教师网络和学生网络的具体结构如下;
根据卷积神经网络中特征图的分辨率大小,将同一尺寸的特征图所在的卷积层称为一个阶段,以残差网络作为特征提取的基础网络进行说明,C1~C5分别表示Conv1~Conv5阶段中最后一层残差块输出的特征图,C6则表示在C5后面再加一层3×3卷积得到的特征图,将C2进行1×1卷积和下采样得到与C3相同大小的特征图,将C4进行1×1卷积和上采样得到与C3相同大小的特征图,并且将这两部分特征图与经过1×1卷积后的C3进行相加,得到融合后的特征图P3;同理,将C4进行1×1卷积和下采样得到与C5相同大小的特征图,将C6进行1×1卷积和上采样得到与C5相同大小的特征图,并且将这两部分特征图与经过1×1卷积后的C5进行相加,得到融合后的高层特征图P5;
为了进一步提取更具判别性的特征,并且抑制光学遥感图像中的背景干扰,设计了一种融合了通道注意力和空间注意力的特征加权模块,其结构为:设融合后的特征图Pi大小为W×H×D,i=3或5,W、H和D分别表示特征图的宽、高和通道数,通道注意力模块由1个3×3卷积层、1个全局平均池化层和2个全连接层组成,,将输入特征图Pi与经过通道注意力模块之后得到的通道权重进行点乘,得到经过通道注意力机制优化后的特征图Pc,大小为W×H×D;空间注意力模块由1×1卷积层+BN层+Sigmoid函数组成,将优化的特征图Pc与经过空间注意力模块Sigmoid层输出的空间权重进行点乘,得到经过空间注意力机制优化后的特征图Ps,大小为W×H×D;
分类和边框回归子网络由2个分支组成,输入为经过优化后的特征图Ps,大小为W×H×D,K表示目标的类别数,分类子网络由5个卷积层组成,前4个是通道数为D的3×3卷积层,最后一层是通道数为KA的3×3卷积层,用于预测每个锚点框的类别概率;边框回归子网络也由5个卷积层组成,前4个是通道数为D的3×3卷积层,最后一层是通道数为4A的3×3卷积层,用于预测每个锚点框的坐标,A表示网络预设的锚点框个数。
进一步的,所述教师网络和学生网络的骨干网络中采用Hint损失函数,在训练的时候使得学生网络尽可能学习教师网络的特征表达能力,其计算公式如下:
其中LHint表示Hint损失,Ft表示从教师网络中选取的中间层特征图,Fs表示从学生网络中选取的中间层特征图,表示两个特征图之间的L2距离;为了保证中间层特征图Ft和Fs具有相同的维度,在学生网络的中间层特征图之后加入了自适应层,将Fs调整到与教师网络的中间层特征图Ft维度一致,然后计算Hint损失。
与现有的舰船检测技术相比,本发明具有以下优点和有益效果:
(1)本发明设计了一种基于知识蒸馏的光学遥感图像舰船检测方法,利用检测准确率较高的教师网络指导学生网络的训练,在保持学生网络原有检测速度的同时提高学生网络的检测准确率;
(2)本发明为了使得学生网络充分学习并模仿教师网络的行为,采用Hint损失让学生网络学习教师网络中间层特征图的表达能力,从而进一步提高学生网络的检测精度。
(3)本发明为了进一步提高对于舰船小目标检测的准确率,设计了一种基于特征融合和特征加权的舰船检测网络,聚合不同层的多尺度局部信息和全局信息,以生成区分性更强的特征,并设计了特征加权模块,进一步提取更具判别性的特征,从而提高检测性能。
附图说明
图1为本发明实施例的流程图。
图2为本发明实施例在教师网络与学生网络联合训练时所用的舰船检测框架。
图3为本发明设计的基于特征融合和特征加权的舰船检测网络结构图。
图4为特征加权模块的结构图。
图5为本发明实施例在测试阶段学生网络所用的舰船检测框架。
具体实施方式
为了更清楚的阐述本发明的目的、技术方案和有益效果,下面结合附图和实施例进一步说明。应当理解,本发明不应限于实施例公开的内容,本发明要求保护的范围以权利要求书界定的范围为准。
如图1所示,本发明公开了一种基于知识蒸馏的光学遥感图像舰船检测方法,具体实施时步骤如下:
步骤1,下载公开的舰船数据集,现有的数据集可划分为两类:一类是多类遥感目标检测数据集,里面包含舰船这个类别,常用的如DOTA数据集、DIOR数据集、NWPU VHR-10数据集等;另一类是只包含舰船这个目标类别的数据集,如HRSC2016数据集以及自行构建的光学遥感图像目标检测数据集MSD。按照数据集官方论文或网址给出的标准划分训练集和测试集;
步骤2,构建基于知识蒸馏的舰船检测网络:
步骤2.1,构建教师网络与学生网络联合训练时所用的舰船检测框架,其结构如图2所示,教师网络和学生网络均由骨干网络、特征加权模块、分类和边框回归子网络三部分组成。教师网络选用卷积层层数较多且检测精度较高的骨干网络,具体实施时可以选用ResNet101或其他较深的网络,学生网络选用卷积层层数较少且检测速度较快的骨干网络,具体实施时可以选用Shufflenet或其他轻量化的网络。将输入图片同时输入到教师网络和学生网络中,由分类和边框回归子网络部分输出每个目标类别的概率以及回归得到目标的位置坐标,根据教师网络和学生网络的检测结果分别计算交叉熵损失和回归损失。在训练过程中,同样根据真实的舰船检测标签与学生网络检测出的结果计算Softmax和SmoothL1损失,通过反向传播,优化学生网络的模型参数;
步骤2.2,具体地,针对舰船特有的目标特性,如形状狭长、在光学遥感图像中舰船属于小目标检测等问题,本发明为教师网络和学生网络设计了一种基于特征融合和特征加权的舰船检测网络,其结构如图3所示。根据卷积神经网络中特征图的分辨率大小,将同一尺寸的特征图所在的卷积层称为一个阶段,以残差网络作为特征提取的基础网络进行说明。图3中,C1~C5分别表示Conv1~Conv5阶段中最后一层残差块输出的特征图,C6则表示在C5后面再加一层3×3卷积得到的特征图。在CNN中,低层特征语义信息相对较少,但是分辨率高,目标定位比较准确;高层特征语义丰富,但是分辨率低,目标定位比较粗糙。对于检测舰船这样的小目标,需要聚合不同层的多尺度局部信息和全局信息,以生成区分性更强的特征,从而提高舰船检测的准确率。在本发明的具体实施例中,将C2进行1×1卷积和下采样得到与C3相同大小的特征图,将C4进行1×1卷积和上采样得到与C3相同大小的特征图,并且将这两部分特征图与经过1×1卷积后的C3进行相加,得到融合后的特征图P3;同理,将C4进行1×1卷积和下采样得到与C5相同大小的特征图,将C6进行1×1卷积和上采样得到与C5相同大小的特征图,并且将这两部分特征图与经过1×1卷积后的C5进行相加,得到融合后的高层特征图P5。上述横向连接用到的1×1卷积均设置相同的通道数D。
步骤2.3,为了进一步提取更具判别性的特征,并且抑制光学遥感图像中的背景干扰,设计的特征加权模块结构如图4所示,该模块包含了了通道注意力和空间注意力。设融合后的特征图Pi(i=3或5)大小为W×H×D,W、H和D分别表示特征图的宽、高和通道数。通道注意力模块由1个3×3卷积层、1个全局平均池化层和2个全连接层组成,图4中r为缩减因子(具体实施时可取r=16)。将输入特征图Pi与经过通道注意力模块之后得到的通道权重进行点乘,得到经过通道注意力机制优化后的特征图Pc,大小为W×H×D。空间注意力模块由1×1卷积层+BN层+Sigmoid函数组成,Sigmoid函数用于约束结果得到[0,1]内的值。将优化的特征图Pc与经过空间注意力模块Sigmoid层输出的空间权重进行点乘,得到经过空间注意力机制优化后的特征图Ps,大小为W×H×D。
步骤2.4,图3中,分类和边框回归子网络由2个分支组成,输入为经过优化后的特征图Ps,大小为W×H×D,图中K表示目标的类别数,A表示网络预设的锚点框个数。分类子网络由5个卷积层组成,前4个是通道数为D的3×3卷积层,最后一层是通道数为KA的3×3卷积层,用于预测每个锚点框的类别概率;边框回归子网络也由5个卷积层组成,前4个是通道数为D的3×3卷积层,最后一层是通道数为4A的3×3卷积层,用于预测每个锚点框的坐标。
步骤2.5,在教师网络和学生网络的骨干网络中,采用Hint损失函数,在训练的时候使得学生网络尽可能学习教师网络的特征表达能力,其计算公式如下:
其中LHint表示Hint损失,Ft表示从教师网络中选取的中间层特征图(一般选取网络比较深层次的特征图),Fs表示从学生网络中选取的中间层特征图(一般选取网络比较深层次的特征图),表示两个特征图之间的L2距离。为了保证中间层特征图Ft和Fs具有相同的维度,加入了自适应层,如图2所示,将学生网络的中间层特征图Fs调整到与教师网络的中间层特征图Ft维度一致,然后计算Hint损失。
步骤3,联合训练教师网络和学生网络:
步骤3.1,选用光学遥感图像目标检测数据集MSD对构建的目标检测网络进行训练,设置训练集与测试集的比例为8:2;
步骤3.2,对训练集中的图片进行预处理,然后训练步骤2搭建的基于知识蒸馏的舰船检测网络,保存训练后的学生网络的模型参数;
步骤4,利用学生网络进行测试:
步骤4.1,在测试阶段,不需要教师网络,只保留学生网络,如图5所示,将测试集中所有图片输入到学生网络中进行舰船检测,得到最后的检测结果;
步骤4.2,根据测试集所有图片的检测结果计算舰船检测的精度指标mAP和速度指标FPS,大量实验表明本发明能准确地检测出光学遥感图像中多尺度、形态各异的舰船目标,有效克服了云层、海浪、云雾等虚景的干扰,在保证舰船检测速度的前提下极大地提高了舰船检测的精度。
本文中所描述的具体实施例仅仅是对本发明精神作举例说明。本发明所属技术领域的技术人员可以对所描述的具体实施例做各种各样的修改或补充或采用类似的方式替代,但并不会偏离本发明的精神或者超越所附权利要求书所定义的范围。
Claims (5)
1.一种基于知识蒸馏的光学遥感图像舰船检测方法,其特征在于,包括如下步骤:
步骤1,下载公开的舰船数据集,并划分训练集和测试集;
步骤2,构建基于知识蒸馏的舰船检测网络,包含教师网络和学生网络,通过设计损失函数,让层数多检测准确率高的教师网络指导学生网络的学习,在保持学生网络原本检测速度的同时提高学生网络检测的准确率;
所述教师网络和学生网络均由骨干网络、特征加权模块、分类和边框回归子网络三部分组成,将输入图片同时输入到教师网络和学生网络中,所述骨干网络用于特征提取,特征加权模块用于进一步提取更具判别性的特征,并且抑制光学遥感图像中的背景干扰,最后由分类和边框回归子网络输出每个目标类别的概率以及回归得到目标的位置坐标,根据教师网络和学生网络的检测结果分别计算交叉熵损失和回归损失;
步骤3,利用训练集对上述教师网络和学生网络进行联合训练,保存训练后的学生网络模型参数;
在训练过程中,根据真实的舰船检测标签与学生网络检测出的结果计算Softmax和SmoothL1损失,通过反向传播,优化学生网络的模型参数;
步骤4,测试阶段利用学生网络对测试集的所有图片进行测试,计算舰船检测的准确率,得到检测结果。
2.如权利要求1所述的一种基于知识蒸馏的光学遥感图像舰船检测方法,其特征在于:所述教师网络和学生网络选用不同的骨干网络用于特征提取,其中教师网络选用卷积层层数较多且检测精度较高的骨干网络,学生网络选用卷积层层数较少且检测速度较快的骨干网络。
3.如权利要求1所述的一种基于知识蒸馏的光学遥感图像舰船检测方法,其特征在于:教师网络中选用ResNet101作为骨干网络,学生网络中选用Shufflenet作为骨干网络。
4.如权利要求1所述的一种基于知识蒸馏的光学遥感图像舰船检测方法,其特征在于:所述教师网络和学生网络的具体结构如下;
构建骨干网络:根据卷积神经网络中特征图的分辨率大小,将同一尺寸的特征图所在的卷积层称为一个阶段,以残差网络作为特征提取的基础网络进行说明,C1~C5分别表示Conv1~Conv5阶段中最后一层残差块输出的特征图,C6则表示在C5后面再加一层3×3卷积得到的特征图,将C2进行1×1卷积和下采样得到与C3相同大小的特征图,将C4进行1×1卷积和上采样得到与C3相同大小的特征图,并且将这两部分特征图与经过1×1卷积后的C3进行相加,得到融合后的特征图P3;同理,将C4进行1×1卷积和下采样得到与C5相同大小的特征图,将C6进行1×1卷积和上采样得到与C5相同大小的特征图,并且将这两部分特征图与经过1×1卷积后的C5进行相加,得到融合后的高层特征图P5;
为了进一步提取更具判别性的特征,并且抑制光学遥感图像中的背景干扰,设计了一种融合了通道注意力和空间注意力的特征加权模块,其结构为:设融合后的特征图Pi大小为W×H×D,i=3或5,W、H和D分别表示特征图的宽、高和通道数,通道注意力模块由1个3×3卷积层、1个全局平均池化层和2个全连接层组成,,将输入特征图Pi与经过通道注意力模块之后得到的通道权重进行点乘,得到经过通道注意力机制优化后的特征图Pc,大小为W×H×D;空间注意力模块由1×1卷积层+BN层+Sigmoid函数组成,将优化的特征图Pc与经过空间注意力模块Sigmoid层输出的空间权重进行点乘,得到经过空间注意力机制优化后的特征图Ps,大小为W×H×D;
分类和边框回归子网络由2个分支组成,输入为经过优化后的特征图Ps,大小为W×H×D,K表示目标的类别数,分类子网络由5个卷积层组成,前4个是通道数为D的3×3卷积层,最后一层是通道数为KA的3×3卷积层,用于预测每个锚点框的类别概率;边框回归子网络也由5个卷积层组成,前4个是通道数为D的3×3卷积层,最后一层是通道数为4A的3×3卷积层,用于预测每个锚点框的坐标,A表示网络预设的锚点框个数。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110002376.6A CN112766087A (zh) | 2021-01-04 | 2021-01-04 | 一种基于知识蒸馏的光学遥感图像舰船检测方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110002376.6A CN112766087A (zh) | 2021-01-04 | 2021-01-04 | 一种基于知识蒸馏的光学遥感图像舰船检测方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN112766087A true CN112766087A (zh) | 2021-05-07 |
Family
ID=75699030
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110002376.6A Pending CN112766087A (zh) | 2021-01-04 | 2021-01-04 | 一种基于知识蒸馏的光学遥感图像舰船检测方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112766087A (zh) |
Cited By (27)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113139500A (zh) * | 2021-05-10 | 2021-07-20 | 重庆中科云从科技有限公司 | 烟雾检测方法、系统、介质及设备 |
CN113449680A (zh) * | 2021-07-15 | 2021-09-28 | 北京理工大学 | 一种基于知识蒸馏的多模小目标检测方法 |
CN113486898A (zh) * | 2021-07-08 | 2021-10-08 | 西安电子科技大学 | 一种基于改进ShuffleNet的雷达信号RD图像干扰辨识方法及系统 |
CN113592742A (zh) * | 2021-08-09 | 2021-11-02 | 天津大学 | 一种去除图像摩尔纹的方法 |
CN113610126A (zh) * | 2021-07-23 | 2021-11-05 | 武汉工程大学 | 基于多目标检测模型无标签的知识蒸馏方法及存储介质 |
CN113642606A (zh) * | 2021-07-13 | 2021-11-12 | 浙江工商大学 | 一种基于注意力机制的海上舰船检测方法 |
CN113705532A (zh) * | 2021-09-10 | 2021-11-26 | 中国人民解放军国防科技大学 | 基于中低分辨率遥感图像的目标检测方法、装置及设备 |
CN113743514A (zh) * | 2021-09-08 | 2021-12-03 | 庆阳瑞华能源有限公司 | 一种基于知识蒸馏的目标检测方法及目标检测终端 |
CN113947590A (zh) * | 2021-10-26 | 2022-01-18 | 四川大学 | 一种基于多尺度注意力引导和知识蒸馏的表面缺陷检测方法 |
CN114022727A (zh) * | 2021-10-20 | 2022-02-08 | 之江实验室 | 一种基于图像知识回顾的深度卷积神经网络自蒸馏方法 |
CN114155436A (zh) * | 2021-12-06 | 2022-03-08 | 大连理工大学 | 长尾分布的遥感图像目标识别逐步蒸馏学习方法 |
CN114220053A (zh) * | 2021-12-15 | 2022-03-22 | 北京建筑大学 | 一种基于车辆特征匹配的无人机视频车辆检索方法 |
CN114241285A (zh) * | 2021-11-25 | 2022-03-25 | 华南理工大学 | 一种基于知识蒸馏和半监督学习的船舶快速检测方法 |
CN114283402A (zh) * | 2021-11-24 | 2022-04-05 | 西北工业大学 | 基于知识蒸馏训练与时空联合注意力的车牌检测方法 |
CN114373133A (zh) * | 2022-01-10 | 2022-04-19 | 中国人民解放军国防科技大学 | 一种基于稠密特征分组蒸馏的缺失模态地物分类方法 |
CN114596497A (zh) * | 2022-05-09 | 2022-06-07 | 北京世纪好未来教育科技有限公司 | 目标检测模型的训练方法、目标检测方法、装置及设备 |
CN114898086A (zh) * | 2022-07-13 | 2022-08-12 | 山东圣点世纪科技有限公司 | 一种基于级联温控蒸馏的目标关键点检测方法 |
CN114998570A (zh) * | 2022-07-19 | 2022-09-02 | 上海闪马智能科技有限公司 | 一种对象检测框的确定方法、装置、存储介质及电子装置 |
CN115115907A (zh) * | 2022-06-29 | 2022-09-27 | 桂林电子科技大学 | 一种基于cqd蒸馏的低照度目标检测方法 |
CN115330759A (zh) * | 2022-10-12 | 2022-11-11 | 浙江霖研精密科技有限公司 | 一种基于豪斯多夫距离计算距离损失的方法及装置 |
CN115965609A (zh) * | 2023-01-03 | 2023-04-14 | 江南大学 | 利用知识蒸馏的陶瓷基板瑕疵智能检测方法 |
CN116168256A (zh) * | 2023-04-19 | 2023-05-26 | 浙江华是科技股份有限公司 | 一种船舶检测方法、系统及计算机存储介质 |
CN116416212A (zh) * | 2023-02-03 | 2023-07-11 | 中国公路工程咨询集团有限公司 | 路面破损检测神经网络训练方法及路面破损检测神经网络 |
CN116503737A (zh) * | 2023-05-10 | 2023-07-28 | 中国人民解放军61646部队 | 基于空间光学图像的船舶检测方法和装置 |
CN116612379A (zh) * | 2023-05-30 | 2023-08-18 | 中国海洋大学 | 一种基于多知识蒸馏的水下目标检测方法及系统 |
CN117058556A (zh) * | 2023-07-04 | 2023-11-14 | 南京航空航天大学 | 基于自监督蒸馏的边缘引导sar图像舰船检测方法 |
CN114155436B (zh) * | 2021-12-06 | 2024-05-24 | 大连理工大学 | 长尾分布的遥感图像目标识别逐步蒸馏学习方法 |
Citations (8)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109034245A (zh) * | 2018-07-27 | 2018-12-18 | 燕山大学 | 一种利用特征图融合的目标检测方法 |
CN109816671A (zh) * | 2019-01-31 | 2019-05-28 | 深兰科技(上海)有限公司 | 一种目标检测方法、装置及存储介质 |
CN110533084A (zh) * | 2019-08-12 | 2019-12-03 | 长安大学 | 一种基于自注意力机制的多尺度目标检测方法 |
CN110969627A (zh) * | 2019-11-29 | 2020-04-07 | 北京达佳互联信息技术有限公司 | 图像处理方法及装置 |
CN111429433A (zh) * | 2020-03-25 | 2020-07-17 | 北京工业大学 | 一种基于注意力生成对抗网络的多曝光图像融合方法 |
CN111626330A (zh) * | 2020-04-23 | 2020-09-04 | 南京邮电大学 | 基于多尺度特征图重构和知识蒸馏的目标检测方法与系统 |
CN111767882A (zh) * | 2020-07-06 | 2020-10-13 | 江南大学 | 一种基于改进yolo模型的多模态行人检测方法 |
CN111784623A (zh) * | 2020-09-07 | 2020-10-16 | 腾讯科技(深圳)有限公司 | 图像处理方法、装置、计算机设备和存储介质 |
-
2021
- 2021-01-04 CN CN202110002376.6A patent/CN112766087A/zh active Pending
Patent Citations (8)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109034245A (zh) * | 2018-07-27 | 2018-12-18 | 燕山大学 | 一种利用特征图融合的目标检测方法 |
CN109816671A (zh) * | 2019-01-31 | 2019-05-28 | 深兰科技(上海)有限公司 | 一种目标检测方法、装置及存储介质 |
CN110533084A (zh) * | 2019-08-12 | 2019-12-03 | 长安大学 | 一种基于自注意力机制的多尺度目标检测方法 |
CN110969627A (zh) * | 2019-11-29 | 2020-04-07 | 北京达佳互联信息技术有限公司 | 图像处理方法及装置 |
CN111429433A (zh) * | 2020-03-25 | 2020-07-17 | 北京工业大学 | 一种基于注意力生成对抗网络的多曝光图像融合方法 |
CN111626330A (zh) * | 2020-04-23 | 2020-09-04 | 南京邮电大学 | 基于多尺度特征图重构和知识蒸馏的目标检测方法与系统 |
CN111767882A (zh) * | 2020-07-06 | 2020-10-13 | 江南大学 | 一种基于改进yolo模型的多模态行人检测方法 |
CN111784623A (zh) * | 2020-09-07 | 2020-10-16 | 腾讯科技(深圳)有限公司 | 图像处理方法、装置、计算机设备和存储介质 |
Non-Patent Citations (2)
Title |
---|
SHIQI CHEN 等: "《Learning Slimming SAR Ship Object Detector Through Network Pruning and Knowledge Distillation 》", 《IEEE》 * |
YUXING MAO 等: "《Network Slimming Method for SAR Ship Detection Based on Knowlegde Distillation 》", 《2020 INTERNATIONAL CONFERENCE ON AVIATION SAFETY AND INFORMATION TECHNOLOGY》 * |
Cited By (45)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113139500B (zh) * | 2021-05-10 | 2023-10-20 | 重庆中科云从科技有限公司 | 烟雾检测方法、系统、介质及设备 |
CN113139500A (zh) * | 2021-05-10 | 2021-07-20 | 重庆中科云从科技有限公司 | 烟雾检测方法、系统、介质及设备 |
CN113486898A (zh) * | 2021-07-08 | 2021-10-08 | 西安电子科技大学 | 一种基于改进ShuffleNet的雷达信号RD图像干扰辨识方法及系统 |
CN113486898B (zh) * | 2021-07-08 | 2024-05-31 | 西安电子科技大学 | 一种基于改进ShuffleNet的雷达信号RD图像干扰辨识方法及系统 |
CN113642606B (zh) * | 2021-07-13 | 2024-01-09 | 浙江工商大学 | 一种基于注意力机制的海上舰船检测方法 |
CN113642606A (zh) * | 2021-07-13 | 2021-11-12 | 浙江工商大学 | 一种基于注意力机制的海上舰船检测方法 |
CN113449680A (zh) * | 2021-07-15 | 2021-09-28 | 北京理工大学 | 一种基于知识蒸馏的多模小目标检测方法 |
CN113449680B (zh) * | 2021-07-15 | 2022-08-30 | 北京理工大学 | 一种基于知识蒸馏的多模小目标检测方法 |
CN113610126B (zh) * | 2021-07-23 | 2023-12-05 | 武汉工程大学 | 基于多目标检测模型无标签的知识蒸馏方法及存储介质 |
CN113610126A (zh) * | 2021-07-23 | 2021-11-05 | 武汉工程大学 | 基于多目标检测模型无标签的知识蒸馏方法及存储介质 |
CN113592742A (zh) * | 2021-08-09 | 2021-11-02 | 天津大学 | 一种去除图像摩尔纹的方法 |
CN113743514A (zh) * | 2021-09-08 | 2021-12-03 | 庆阳瑞华能源有限公司 | 一种基于知识蒸馏的目标检测方法及目标检测终端 |
CN113705532A (zh) * | 2021-09-10 | 2021-11-26 | 中国人民解放军国防科技大学 | 基于中低分辨率遥感图像的目标检测方法、装置及设备 |
CN113705532B (zh) * | 2021-09-10 | 2023-05-23 | 中国人民解放军国防科技大学 | 基于中低分辨率遥感图像的目标检测方法、装置及设备 |
CN114022727A (zh) * | 2021-10-20 | 2022-02-08 | 之江实验室 | 一种基于图像知识回顾的深度卷积神经网络自蒸馏方法 |
CN114022727B (zh) * | 2021-10-20 | 2024-04-26 | 之江实验室 | 一种基于图像知识回顾的深度卷积神经网络自蒸馏方法 |
CN113947590A (zh) * | 2021-10-26 | 2022-01-18 | 四川大学 | 一种基于多尺度注意力引导和知识蒸馏的表面缺陷检测方法 |
CN114283402B (zh) * | 2021-11-24 | 2024-03-05 | 西北工业大学 | 基于知识蒸馏训练与时空联合注意力的车牌检测方法 |
CN114283402A (zh) * | 2021-11-24 | 2022-04-05 | 西北工业大学 | 基于知识蒸馏训练与时空联合注意力的车牌检测方法 |
CN114241285B (zh) * | 2021-11-25 | 2024-05-28 | 华南理工大学 | 一种基于知识蒸馏和半监督学习的船舶快速检测方法 |
CN114241285A (zh) * | 2021-11-25 | 2022-03-25 | 华南理工大学 | 一种基于知识蒸馏和半监督学习的船舶快速检测方法 |
CN114155436B (zh) * | 2021-12-06 | 2024-05-24 | 大连理工大学 | 长尾分布的遥感图像目标识别逐步蒸馏学习方法 |
CN114155436A (zh) * | 2021-12-06 | 2022-03-08 | 大连理工大学 | 长尾分布的遥感图像目标识别逐步蒸馏学习方法 |
CN114220053A (zh) * | 2021-12-15 | 2022-03-22 | 北京建筑大学 | 一种基于车辆特征匹配的无人机视频车辆检索方法 |
CN114373133A (zh) * | 2022-01-10 | 2022-04-19 | 中国人民解放军国防科技大学 | 一种基于稠密特征分组蒸馏的缺失模态地物分类方法 |
CN114596497A (zh) * | 2022-05-09 | 2022-06-07 | 北京世纪好未来教育科技有限公司 | 目标检测模型的训练方法、目标检测方法、装置及设备 |
CN114596497B (zh) * | 2022-05-09 | 2022-08-19 | 北京世纪好未来教育科技有限公司 | 目标检测模型的训练方法、目标检测方法、装置及设备 |
CN115115907B (zh) * | 2022-06-29 | 2024-03-29 | 桂林电子科技大学 | 一种基于cqd蒸馏的低照度目标检测方法 |
CN115115907A (zh) * | 2022-06-29 | 2022-09-27 | 桂林电子科技大学 | 一种基于cqd蒸馏的低照度目标检测方法 |
CN114898086A (zh) * | 2022-07-13 | 2022-08-12 | 山东圣点世纪科技有限公司 | 一种基于级联温控蒸馏的目标关键点检测方法 |
CN114898086B (zh) * | 2022-07-13 | 2022-09-20 | 山东圣点世纪科技有限公司 | 一种基于级联温控蒸馏的目标关键点检测方法 |
CN114998570A (zh) * | 2022-07-19 | 2022-09-02 | 上海闪马智能科技有限公司 | 一种对象检测框的确定方法、装置、存储介质及电子装置 |
CN115330759B (zh) * | 2022-10-12 | 2023-03-10 | 浙江霖研精密科技有限公司 | 一种基于豪斯多夫距离计算距离损失的方法及装置 |
CN115330759A (zh) * | 2022-10-12 | 2022-11-11 | 浙江霖研精密科技有限公司 | 一种基于豪斯多夫距离计算距离损失的方法及装置 |
CN115965609B (zh) * | 2023-01-03 | 2023-08-04 | 江南大学 | 利用知识蒸馏的陶瓷基板瑕疵智能检测方法 |
CN115965609A (zh) * | 2023-01-03 | 2023-04-14 | 江南大学 | 利用知识蒸馏的陶瓷基板瑕疵智能检测方法 |
CN116416212B (zh) * | 2023-02-03 | 2023-12-08 | 中国公路工程咨询集团有限公司 | 路面破损检测神经网络训练方法及路面破损检测神经网络 |
CN116416212A (zh) * | 2023-02-03 | 2023-07-11 | 中国公路工程咨询集团有限公司 | 路面破损检测神经网络训练方法及路面破损检测神经网络 |
CN116168256A (zh) * | 2023-04-19 | 2023-05-26 | 浙江华是科技股份有限公司 | 一种船舶检测方法、系统及计算机存储介质 |
CN116503737B (zh) * | 2023-05-10 | 2024-01-09 | 中国人民解放军61646部队 | 基于空间光学图像的船舶检测方法和装置 |
CN116503737A (zh) * | 2023-05-10 | 2023-07-28 | 中国人民解放军61646部队 | 基于空间光学图像的船舶检测方法和装置 |
CN116612379B (zh) * | 2023-05-30 | 2024-02-02 | 中国海洋大学 | 一种基于多知识蒸馏的水下目标检测方法及系统 |
CN116612379A (zh) * | 2023-05-30 | 2023-08-18 | 中国海洋大学 | 一种基于多知识蒸馏的水下目标检测方法及系统 |
CN117058556B (zh) * | 2023-07-04 | 2024-03-22 | 南京航空航天大学 | 基于自监督蒸馏的边缘引导sar图像舰船检测方法 |
CN117058556A (zh) * | 2023-07-04 | 2023-11-14 | 南京航空航天大学 | 基于自监督蒸馏的边缘引导sar图像舰船检测方法 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112766087A (zh) | 一种基于知识蒸馏的光学遥感图像舰船检测方法 | |
CN112308019B (zh) | 基于网络剪枝和知识蒸馏的sar舰船目标检测方法 | |
CN110188705B (zh) | 一种适用于车载系统的远距离交通标志检测识别方法 | |
CN110163187B (zh) | 基于f-rcnn的远距离交通标志检测识别方法 | |
CN114202696A (zh) | 基于上下文视觉的sar目标检测方法、装置和存储介质 | |
CN111242037A (zh) | 基于结构信息的车道线检测方法 | |
CN110163108B (zh) | 基于双路径特征融合网络的鲁棒声呐目标检测方法 | |
CN112434672B (zh) | 一种基于改进YOLOv3的海上人体目标检测方法 | |
CN111079739B (zh) | 一种多尺度注意力特征检测方法 | |
CN109271856A (zh) | 基于扩张残差卷积的光学遥感图像目标检测方法 | |
CN107609525A (zh) | 基于剪枝策略构建卷积神经网络的遥感图像目标检测方法 | |
CN113486764B (zh) | 一种基于改进的YOLOv3的坑洼检测方法 | |
CN111680706A (zh) | 一种基于编码和解码结构的双通道输出轮廓检测方法 | |
CN114565860B (zh) | 一种多维度增强学习合成孔径雷达图像目标检测方法 | |
CN111368671A (zh) | 基于深度学习的sar图像舰船目标检测识别一体化方法 | |
CN110647802A (zh) | 基于深度学习的遥感影像舰船目标检测方法 | |
CN110991257B (zh) | 基于特征融合与svm的极化sar溢油检测方法 | |
CN114241003B (zh) | 一种全天候轻量化高实时性海面船只检测与跟踪方法 | |
CN113487600B (zh) | 一种特征增强尺度自适应感知船舶检测方法 | |
CN113807188A (zh) | 基于锚框匹配和Siamese网络的无人机目标跟踪方法 | |
CN115223017B (zh) | 一种基于深度可分离卷积的多尺度特征融合桥梁检测方法 | |
CN116469020A (zh) | 一种基于多尺度和高斯Wasserstein距离的无人机图像目标检测方法 | |
CN116580322A (zh) | 一种地面背景下无人机红外小目标检测方法 | |
CN115527098A (zh) | 基于全局均值对比度空间注意力的红外小目标检测方法 | |
Fan et al. | A novel sonar target detection and classification algorithm |
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 | ||
RJ01 | Rejection of invention patent application after publication |
Application publication date: 20210507 |
|
RJ01 | Rejection of invention patent application after publication |