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

本发明属于目标数据集增强方法应用，具体涉及一种基于遥感时敏目标智能检测的数据增强方法。

背景技术：

1、近几年，得益于深度学习以及计算机显卡等软硬件技术的研究发展，通过机器代替人眼进行判断和决策的计算机视觉算法作为人工智能的一项重要分支发展迅速，并在当今社会起到了极为广泛的应用。

2、基于深度学习的目标检测技术是计算机视觉领域最基本的问题之一，通过目标检测算法可以实现目标的定位和分类，并能够判读物体的大小和形状。如今目标检测技术在人脸识别、工业检测、无人驾驶等多重领域展开重大应用。但是时敏目标即时间敏感目标，该类目标的位置会随着时间的变化而变化，具有有限的时间窗口期，由于时敏目标数据获取难度高，因此为了提升时敏目标检测精度，研究数据集增强算法意义重大。

技术实现思路

1、本发明所要解决的技术问题在于针对上述现有技术的不足，提供了一种基于遥感时敏目标智能检测的数据增强方法，该方法能够有效解决时敏目标数据数量少，从而导致深度学习检测模型训练不足，易产生过拟合等问题，能够通过少量的时敏目标数据生成大量包含时敏目标的图像，从而实现数据集增强，为目标检测任务提供良好的数据支撑。

2、为解决上述技术问题，本发明采用的技术方案是：一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，包括：

3、s1、构建图像增强多尺度目标生成模型；所述图像增强多尺度目标生成模型包括倒金字塔结构多尺度生成器和倒金字塔结构的多尺度判别器；

4、s2、获得图像样本数据集，然后利用对抗损失和重建损失对构建的图像增强多尺度目标生成模型进行训练，训练结束后得到图像增强多尺度目标生成模型；

5、s3、对含有目标的图像数据集，通过所述图像增强多尺度目标生成模型对三组含有目标的图像数据集进行数据集增强，每个数据集中的每张图片均生产多张图片。

6、优选地，利用倒金字塔结构构建多尺度生成器{g0,...,gn}，然后对金字塔中的图像x:{x0,...,xn}进行训练，金字塔中图像为图像样本数据集中的图片，其中生成的低维图像通过一个因子r上采样至高维图像；与多尺度生成器对应的，利用倒金字塔结构构建构建多尺度判别器{d0,...,dn}，生成器gn通过学习去欺骗其相对应的判别器dn，判别器的作用对生成的样本图像块与原始图像块进行辨别。

7、优选地，在金字塔的最底层，使用的是一个纯生成器，如式(1)：

8、

9、其中，gn将高斯白噪声zn映射到一个图像样本；在该层中的有效感受野通常为图像高度的一半，因此gn生成图像的总体布局和目标的全局结构，每个具有更细尺度(n＜n)的生成器gn都添加了以前尺度没有生成的细节；因此，除了噪声空间zn，每个生成器gn增加一个来自上层尺度图像的上采样版本，如式(2)：

10、

11、使用的所有生成器具有相似的架构，在每个尺度架构n中，来自上一层尺度的图像通过上采样变成，然后被加入到输入噪声zn当中，然后将相加后的结果输入到五个卷积层当中，卷积层的作用是生成中缺失的详细信息；生成器gn生成的图像如式(3)：

12、

13、其中，为一个拥有五个卷积块的全卷积网，每个卷积块包括3×3卷积层、批归一化层以及leakyrelu激活函数。

14、优选地，第n个网络的训练的多尺度判别器损失由一个对抗性损失和一个重建损失组成：

15、

16、对抗损失ladv用来惩罚xn中图像块分布与生成样本中图像块分布之间的距离；重建损失lrec确保了一组可以产生xn的特定噪声映射的存在；

17、每个生成器gn都与一个马尔可夫判别器dn相耦合，该判别器将其输入的每个重叠块进行真或假分类，dn的架构与gn中的网络相同，他们中的图像块尺寸都为11×11；

18、为了确保存在一组特定的输入噪声图可以生成原始图像x；分别选择其中z*为训练前先去的固定的噪声图，训练开始后将不再改变，重建损失如下：

19、

20、重建的图像在训练中有另一个作用，即确定每个尺度中噪声zn的标准差σn。

21、优选地，选取三组含有目标的图像数据集，分别为光学遥感图像数据集a、可见光图像数据集b、sar图像数据集c，其中a＝{a0,a1,...,an}，b＝{b0,b1,...,bn}，c＝{c0,c1,…,cn}；对光学遥感图像数据集a、可见光图像数据集b、sar图像数据集c中的每张图片进行数据集增强，每张图片均生产多张图片。

22、优选地，每张图片均生产50张图片。

23、本发明与现有技术相比具有以下优点：

24、1、本发明针对时敏目标数据数量少，从而导致深度学习检测模型训练不足，易产生过拟合等问题，设计一种基于遥感时敏目标智能检测的数据增强方法，首先构建基于倒金字塔结构的多尺度目标生成模型，然后构建图像生成模型并利用对抗损失和重建损失进行模型训练，最后实现基于可将光遥感图像、红外图像、sar图像的时敏目标数据集增强。该方法能够通过少量的时敏目标数据生成大量包含时敏目标的图像，从而实现数据集增强，为目标检测任务提供良好的数据支撑，具有一定的实用价值。

25、下面通过附图和实施例对本发明的技术方案作进一步的详细说明。

技术特征：

1.一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，包括：

2.根据权利要求1所述的一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，利用倒金字塔结构构建多尺度生成器{g0,…,gn}，然后对金字塔中的图像x:{x0,...,xn}进行训练，金字塔中图像为图像样本数据集中的图片，其中生成的低维图像通过一个因子r上采样至高维图像；与多尺度生成器对应的，利用倒金字塔结构构建构建多尺度判别器{d0,...,dn}，生成器gn通过学习去欺骗其相对应的判别器dn，判别器的作用对生成的样本图像块与原始图像块进行辨别。

3.根据权利要求2所述的一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，在金字塔的最底层，使用的是一个纯生成器，如式(1)：

4.根据权利要求2所述的一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，第n个网络的训练的多尺度判别器损失由一个对抗性损失和一个重建损失组成：

5.根据权利要求1或2所述的一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，选取三组含有目标的图像数据集，分别为光学遥感图像数据集a、可见光图像数据集b、sar图像数据集c，其中a＝{a0,a1,...,an}，b＝{b0,b1,...,bn}，c＝{c0,c1,...,cn}；对光学遥感图像数据集a、可见光图像数据集b、sar图像数据集c中的每张图片进行数据集增强，每张图片均生产多张图片。

6.根据权利要求5所述的一种基于遥感时敏目标智能检测的数据增强方法，其特征在于，每张图片均生产50张图片。

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

技术研发人员：卢瑞涛,王思宇,杨小冈,李清格,李云松,范继伟,夏克寒,张涛
受保护的技术使用者：中国人民解放军火箭军工程大学
技术研发日：
技术公布日：2024/1/15