基于GAN目标数据增强的海面小目标检测

    Sea-surface Small Target Detection Based on Target Data Augmentation Using GAN

    • 摘要: 由于海面存在大量背景杂波,以及目标的多样性,海面小目标检测一直是一个具有挑战性的问题。近年来,基于深度学习的海面小目标检测逐渐流行,具有较好的性能。然而,这类方法严重依赖于训练样本的数量和质量,通常情况下,目标样本数量远少于海杂波样本数量,这会给训练带来困难同时降低检测性能。基于此,文中提出了一种基于目标数据增强的海面小目标检测方法。通过对目标的时频特征进行分析,基于生成对抗网络(GAN)生成相似的特征分布样本,从而扩展目标数据集,缩小目标样本和海杂波样本之间的数量差距,使更多的目标信息被用于检测器的训练过程中,以提高检测器性能。在IPIX数据集上的实验结果表明该方法可以获得更好的检测性能,及所提出的检测器可以实现虚警率的控制。

       

      Abstract: Due to the presence of a large amount of background clutter and the diversity of targets on the sea surface, sea-surface small target detection has always been a challenging problem. In recent years, sea surface small target detection based on deep learning is gradually becoming popular and has good detection performance. However, such methods heavily rely on the quantity and quality of training samples. Typically, the number of target samples is much smaller than the number of sea clutter samples, which can bring difficulties to training and reduce detection performance. Based on this, this paper proposes a sea surface small target detection method based on target data augmentation. By analyzing the time-frequency characteristics of the target, similar feature distribution samples are generated based on the Generative adversarial network (GAN), so as to expand the target data set, narrow the number gap between target samples and sea clutter samples, and make more target information be used in the training process of the detector to improve the performance of the detector. The experimental results on the IPIX dataset indicate that this method can achieve better detection performance. The experimental results also indicate that the proposed detector can achieve control of false alarm rate.

       

    /

    返回文章
    返回