一种基于遥感时敏目标智能检测的数据增强方法技术

本发明专利技术提出了一种基于遥感时敏目标智能检测的数据增强方法，该方法为：构建多尺度目标生成模型；获得图像样本数据集，然后利用对抗损失和重建损失对构建的图像增强多尺度目标生成模型进行训练，训练结束后得到图像增强多尺度目标生成模型；对含有目标的图像数据集，通过所述图像增强多尺度目标生成模型对三组含有目标的图像数据集进行数据集增强，每个数据集中的每张图片均生产多张图片。本发明专利技术效解决时敏目标数据数量少，从而导致深度学习检测模型训练不足，易产生过拟合等问题，能够通过少量的时敏目标数据生成大量包含时敏目标的图像，从而实现数据集增强，为目标检测任务提供良好的数据支撑。具有一定的实用价值。具有一定的实用价值。具有一定的实用价值。

【技术实现步骤摘要】
一种基于遥感时敏目标智能检测的数据增强方法


[0001]本专利技术属于目标数据集增强方法应用
，具体涉及一种基于遥感时敏目标智能检测的数据增强方法。

技术介绍

[0002]近几年，得益于深度学习以及计算机显卡等软硬件技术的研究发展，通过机器代替人眼进行判断和决策的计算机视觉算法作为人工智能的一项重要分支发展迅速，并在当今社会起到了极为广泛的应用。
[0003]基于深度学习的目标检测技术是计算机视觉领域最基本的问题之一，通过目标检测算法可以实现目标的定位和分类，并能够判读物体的大小和形状。如今目标检测技术在人脸识别、工业检测、无人驾驶等多重领域展开重大应用。但是时敏目标即时间敏感目标，该类目标的位置会随着时间的变化而变化，具有有限的时间窗口期，由于时敏目标数据获取难度高，因此为了提升时敏目标检测精度，研究数据集增强算法意义重大。

技术实现思路

[0004]本专利技术所要解决的技术问题在于针对上述现有技术的不足，提供了一种基于遥感时敏目标智能检测的数据增强方法，该方法能够有效解决时敏目标数据数量少，从而导致深度学习检测模型训练不足，易产生过拟合等问题，能够通过少量的时敏目标数据生成大量包含时敏目标的图像，从而实现数据集增强，为目标检测任务提供良好的数据支撑。
[0005]为解决上述技术问题，本专利技术采用的技术方案是：一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，包括：
[0006]S1、构建图像增强多尺度目标生成模型；所述图像增强多尺度目标生成模型包括倒金字塔结构多尺度生成器和倒金字塔结构的多尺度判别器；
[0007]S2、获得图像样本数据集，然后利用对抗损失和重建损失对构建的图像增强多尺度目标生成模型进行训练，训练结束后得到图像增强多尺度目标生成模型；
[0008]S3、对含有目标的图像数据集，通过所述图像增强多尺度目标生成模型对三组含有目标的图像数据集进行数据集增强，每个数据集中的每张图片均生产多张图片。
[0009]优选地，利用倒金字塔结构构建多尺度生成器{G0,...,G
N
}，然后对金字塔中的图像x:{x0,...,x
N
}进行训练，金字塔中图像为图像样本数据集中的图片，其中生成的低维图像通过一个因子r上采样至高维图像；与多尺度生成器对应的，利用倒金字塔结构构建构建多尺度判别器{D0,...,D
N
}，生成器G
n
通过学习去欺骗其相对应的判别器D
n
，判别器的作用对生成的样本图像块与原始图像块进行辨别。
[0010]优选地，在金字塔的最底层，使用的是一个纯生成器，如式(1)：
[0011][0012]其中，G
N
将高斯白噪声z
N
映射到一个图像样本；在该层中的有效感受野通常为图像高度的一半，因此G
N
生成图像的总体布局和目标的全局结构，每个具有更细尺度(n＜
N)的生成器G
n
都添加了以前尺度没有生成的细节；因此，除了噪声空间z
N
，每个生成器G
n
增加一个来自上层尺度图像的上采样版本，如式(2)：
[0013][0014]使用的所有生成器具有相似的架构，在每个尺度架构n中，来自上一层尺度的图像通过上采样变成，然后被加入到输入噪声z
n
当中，然后将相加后的结果输入到五个卷积层当中，卷积层的作用是生成中缺失的详细信息；生成器G
n
生成的图像如式(3)：
[0015][0016]其中，为一个拥有五个卷积块的全卷积网，每个卷积块包括3
×
3卷积层、批归一化层以及LeakyReLU激活函数。
[0017]优选地，第n个网络的训练的多尺度判别器损失由一个对抗性损失和一个重建损失组成：
[0018][0019]对抗损失L
adv
用来惩罚x
n
中图像块分布与生成样本中图像块分布之间的距离；重建损失L
rec
确保了一组可以产生x
n
的特定噪声映射的存在；
[0020]每个生成器G
n
都与一个马尔可夫判别器D
n
相耦合，该判别器将其输入的每个重叠块进行真或假分类，D
n
的架构与G
n
中的网络相同，他们中的图像块尺寸都为11
×
11；
[0021]为了确保存在一组特定的输入噪声图可以生成原始图像x；分别选择其中z
*
为训练前先去的固定的噪声图，训练开始后将不再改变，重建损失如下：
[0022][0023]重建的图像在训练中有另一个作用，即确定每个尺度中噪声z
n
的标准差σ
n
。
[0024]优选地，选取三组含有目标的图像数据集，分别为光学遥感图像数据集A、可见光图像数据集B、SAR图像数据集C，其中A＝{a0,a1,...,a
n
}，B＝{b0,b1,...,b
n
}，C＝{c0,c1,
…
,c
n
}；对光学遥感图像数据集A、可见光图像数据集B、SAR图像数据集C中的每张图片进行数据集增强，每张图片均生产多张图片。
[0025]优选地，每张图片均生产50张图片。
[0026]本专利技术与现有技术相比具有以下优点：
[0027]1、本专利技术针对时敏目标数据数量少，从而导致深度学习检测模型训练不足，易产生过拟合等问题，设计一种基于遥感时敏目标智能检测的数据增强方法，首先构建基于倒
金字塔结构的多尺度目标生成模型，然后构建图像生成模型并利用对抗损失和重建损失进行模型训练，最后实现基于可将光遥感图像、红外图像、SAR图像的时敏目标数据集增强。该方法能够通过少量的时敏目标数据生成大量包含时敏目标的图像，从而实现数据集增强，为目标检测任务提供良好的数据支撑，具有一定的实用价值。
[0028]下面通过附图和实施例对本专利技术的技术方案作进一步的详细说明。
附图说明
[0029]图1为本专利技术实施例1公开的一种基于遥感时敏目标智能检测的数据增强方法的流程示意图。
[0030]图2为本专利技术实施例1构建的图像增强多尺度目标生成模型的模型示意图。
[0031]图3为本专利技术实施例1倒金字塔结构多尺度生成器的模型示意图。
[0032]图4为利用本实施例1增强后的数据生成红外图像示意图。
[0033]图5为利用本实施例1增强后的数据生成可见光遥感图像示意图。
[0034]图6为利用本实施例1增强后的数据生成SAR图像示意图。
[0035]图7为SAR检测示意图。
[0036]图8为利用本实施例1增强后的数据训练混淆矩阵。
具体实施方式
[0037]实施例1
[0038]如图1所示，本本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，包括：S1、构建图像增强多尺度目标生成模型；所述图像增强多尺度目标生成模型包括倒金字塔结构多尺度生成器和倒金字塔结构的多尺度判别器；S2、获得图像样本数据集，然后利用对抗损失和重建损失对构建的图像增强多尺度目标生成模型进行训练，训练结束后得到图像增强多尺度目标生成模型；S3、对含有目标的图像数据集，通过所述图像增强多尺度目标生成模型对三组含有目标的图像数据集进行数据集增强，每个数据集中的每张图片均生产多张图片。2.根据权利要求1所述的一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，利用倒金字塔结构构建多尺度生成器{G0,
…
,G
N
}，然后对金字塔中的图像x:{x0,...,x
N
}进行训练，金字塔中图像为图像样本数据集中的图片，其中生成的低维图像通过一个因子r上采样至高维图像；与多尺度生成器对应的，利用倒金字塔结构构建构建多尺度判别器{D0,...,D
N
}，生成器G
n
通过学习去欺骗其相对应的判别器D
n
，判别器的作用对生成的样本图像块与原始图像块进行辨别。3.根据权利要求2所述的一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，在金字塔的最底层，使用的是一个纯生成器，如式(1)：其中，G
N
将高斯白噪声z
N
映射到一个图像样本在该层中的有效感受野通常为图像高度的一半，因此G
N
生成图像的总体布局和目标的全局结构，每个具有更细尺度(n＜N)的生成器G
n
都添加了以前尺度没有生成的细节；因此，除了噪声空间z
N
，每个生成器G
n
增加一个来自上层尺度图像的上采样版本，如式(2)：使用的所有生成器具有相似的架构，在每个尺度架构n中，来自上一层尺度的图像通过上采样变成然后被加入到输入噪声z
n
当中，然后将相加后的结果输...

【专利技术属性】
技术研发人员：卢瑞涛，王思宇，杨小冈，李清格，李云松，范继伟，夏克寒，张涛，
申请(专利权)人：中国人民解放军火箭军工程大学，
类型：发明
国别省市：