基于注意力机制的SOLOA船舶实例分割算法

摘要: 目前，可见光船舶图像的实例分割仍然是较有挑战性的工作。由于船舶图像场景复杂多变，多数实例分割算法无法对复杂场景下的船舶图像进行有效分割。本文提出了基于注意力机制的依靠位置分割目标（attention based segmenting objects by locations, SOLOA）船舶实例分割算法，针对网络特征图里实例信息不完善、背景干扰较多的问题，使用空间注意力机制来充分利用分类特征中的实例信息，建模图像实例间的相互关系并与分割特征相融合。实验结果表明,在新构建的船舶图像数据集上进行训练和测试，改进的网络模型能有效地增强网络特征中的实例信息、减弱背景的干扰。SOLOA算法的船舶实例分割准确率高于其他算法，可以很好地适应复杂场景下的船舶分割需求。

• 关键词: 
• 船舶目标  /  
• 实例分割  /  
• 复杂海上场景  /  
• 深度学习  /  
• 卷积神经网络  /  
• 注意力机制  /  
• 单阶段实例分割  /  
• 可见光图像  

近年来，芯片的生产能力以及计算机技术不断提升，包括计算机视觉在内的多个人工智能领域蓬勃发展，人类已步入数字化、智能化时代。随着船舶数量、载运量不断增加，船舶交通日益稠密，人们对高性能的船舶智