极坐标编解码的轻量化SAR图像舰船斜框检测算法
作者机构:辽宁工程技术大学软件学院 辽宁工程技术大学基础教学部
出 版 物:《计算机科学与探索》 (Journal of Frontiers of Computer Science and Technology)
年 卷 期:2024年
核心收录:
学科分类:11[军事学] 080904[工学-电磁场与微波技术] 0810[工学-信息与通信工程] 0809[工学-电子科学与技术(可授工学、理学学位)] 08[工学] 081105[工学-导航、制导与控制] 081001[工学-通信与信息系统] 081002[工学-信号与信息处理] 0825[工学-航空宇航科学与技术] 1109[军事学-军事装备学] 0811[工学-控制科学与工程]
基 金:国家自然科学基金面上项目(51874166,52274206) 国家自然基金青年基金(51904144)
主 题:合成孔径雷达 舰船目标斜框检测 深度学习 多尺度 特征增强
摘 要:针对目前合成孔径雷达(SAR)图像舰船目标斜框检测算法中存在的参数量过大难以满足实时化检测需求以及边界不连续的问题,提出了一种基于极坐标编解码的轻量化SAR舰船目标斜框检测算法。首先,基于ShuffleNetV2的shuffle单元并利用轻量高效的PC卷积和Ghost卷积,提出了卷积协同单元,实现优化卷积操作,减少算法的复杂度。然后,引入极坐标编解码方法,并提出余弦调和IOU加权损失函数,动态调节极坐标编码损失,解决斜框检测存在的边界不连续问题,同时使用双峰最大池化和椭圆二维高斯分布对极坐标编解码方法进行改进,以提高对近岸密集排布舰船的检测精度。最后,提出多尺度条形卷积注意力模块和空间选择特征增强模块,通过获取不同尺度特征信息,以提高网络特征提取能力。在SSDD+和DSSDD数据集上的实验结果表明,该算法分别实现了85.7%和90.6%的检测精度,并在HRSID数据集上进行泛化测试,检测精度可达82.3%。相较于同类算法在检测精度相近的情况下,参数量和浮点运算量低于同类算法的十分之一,仅为0.87MB和1.21GB,检测速度提升14%,可达135FPS,满足实时性检测需求。