用于声呐水下目标跟踪的增强型SiamMask网络
An enhanced SiamMask network for underwater target tracking in sonar
-
摘要: 针对声呐图像成像中存在的水下噪声干扰以及目标边界模糊问题, 提出一种基于SiamMask的新型水下目标跟踪方法。通过结合混合注意力与互相关机制增强网络对水下目标边界的感知能力, 缓解噪声带来的干扰问题; 此外, 采用排序损失优化策略对网络的原损失附加约束, 联合正样本的前景置信度分数与交并比值, 减小网络分类分支和回归分支之间的差距, 降低误匹配风险。评估结果表明, 所提方法在自制声呐数据集与公共声呐数据集中均取得了领先的性能, 模型能够以较快的速度实现具有竞争力的声呐图像水下轮廓跟踪任务。Abstract: In order to solve the problem of underwater noise interference and target boundary blurring in sonar image imaging, this paper proposes a new underwater target tracking method based on SiamMask. By combining hybrid attention and cross-correlation mechanisms to enhance the network’s ability to perceive underwater target boundaries, the method mitigates the interference caused by noise. Furthermore, a ranking loss optimization strategy is employed to impose additional constraints on the original loss of the network. In particular, the discrepancy between the classification and regression branches of the network is diminished by integrating the prospective confidence scores of the positive samples with the IoU (intersection over union) values, which effectively mitigates the risk of misalignment. The evaluation results indicate that the proposed method has achieved leading performance in both self-made sonar datasets and public sonar datasets, and the model can achieve competitive underwater contour tracking tasks for sonar images at a fast speed.
下载: