摘要
针对侧扫声纳水下目标图像稀缺,获取难度大、成本高,导致基于深度学习的目标检测模型性能差的问题,结合光学域类目标数据集丰富的现状,提出一种基于通道和空间注意力(channel and spatial attention,CSA)模块、最小二乘生成对抗生成网络(least squares generative adversarial networks,LSGAN)及循环对抗生成网络(cycle generative adversarial networks,CycleGAN)的侧扫声纳水下目标图像样本扩增方法。首先,受CycleGAN的启发,设计基于循环一致性的单循环网络结构,保证模型的训练效率。然后,在生成器中融合CSA模块,减少信息弥散的同时增强跨纬度交互。最后,设计了基于LSGAN的损失函数,提高生成图像质量的同时提高训练稳定性。在船舶光学域数据集与侧扫声纳沉船数据集上进行实验,所提方法实现了光学-侧扫声纳样本间信息的高效、稳健转换以及大量侧扫声纳目标样本的扩增。同时,基于本文生成样本训练后的检测模型进行了水下目标检测,结果表明,使用本文样本扩增数据训练后的模型在少样本沉船目标检测的平均准确率达到了84.71%,证明了所提方法实现了零样本和小样本水下强代表性目标样本的高质量扩增,并为高性能水下目标检测模型构建提供了一种新的途径。
In view of the scarcity,difficulty and high cost of side-scan sonar underwater target images,and the poor performance of deep-learning based target detection model,combined with the abundant target data set in optical domain,a sample augmentation method for underwater target side-scan sonar images based on channel and spatial attention(CSA)module and least squares generative adversarial networks(LSGAN)and cycle generative adversarial networks(CycleGAN)is propesed.Firstly,inspired by CycleGAN,a single cycle network structure based on cycle consistency is designed to ensure the training efficiency of the model.Then,the CSA module is integrated into the generator to reduce information dispersion while enhancing cross-latitude interaction.Finally,a loss function based on LSGAN is designed to improve the quality of the generated image while improving the training stability.Experiments are carried out on ship optical domain data set and side-scan sonar shipwreck data set.The results show that the proposed method achieves efficient and robust conversion of information between optical and side-scan sonar samples and augmentation of a large number of side-scan sonar target samples.At the same time,the underwater target detection is carried out based on the detection model generated after sample training in this paper.The results show that the average precision value of the model after training with sample augmentation data in this paper reachs 84.71%in detecting shipwreck targets with few samples,which proves that the method in this paper achieves high-quality amplification of highly representative underwater target samples with zero samples and small samples.It also provides a new way to construct high-performance underwater target detection model.
作者
汤寓麟
王黎明
余德荧
李厚朴
刘敏
张卫东
TANG Yulin;WANG Liming;YU Deying;LI Houpu;LIU Min;ZHANG Weidong(College of Electrical Engineering,Naval University of Engineering,Wuhan 430033,China;Unit 91001 of the PLA,Beijing 100841,China;Unit 31016 of the PLA,Beijing 100088,China)
出处
《系统工程与电子技术》
EI
CSCD
北大核心
2024年第5期1514-1524,共11页
Systems Engineering and Electronics
基金
国家优秀青年科学基金(42122025)
国家自然科学基金(41974005,41971416,42074074)
湖北省杰出青年科学基金(2019CFA086)资助课题。
关键词
样本扩增
侧扫声纳
循环生成对抗网络
通道和空间注意力模块
最小二乘生成对抗网络
sample augmentation
side-scan sonar
cycle generative adversarial networks(CycleGAN)
channel and spatial attention(CSA)module
least squares generative adversarial networks(LSGAN)
