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

SOLOA ship instance segmentation algorithm based on attention

作　　者：孙雨鑫 苏丽[1,2] 陈禹升 苑守正 孟浩[1,2] SUN Yuxin;SU Li;CHEN Yusheng;YUAN Shouzheng;MENG Hao(College of Intelligent Science and Engineering,Harbin Engineering University,Harbin 150001,China;Key Laboratory of Ministry of Education on Intelligent Technology and Application of Marine Equipment,Harbin Engineering University,Harbin 150001,China)

机构地区：[1]哈尔滨工程大学智能科学与工程学院,黑龙江哈尔滨150001 [2]哈尔滨工程大学船舶装备智能化技术与应用教育部重点实验室,黑龙江哈尔滨150001

出　　处：《智能系统学报》2023年第6期1197-1204,共8页CAAI Transactions on Intelligent Systems

基　　金：国家重点研发计划项目(2019YFE0105400);船舶态势智能感知系统研制项目(MC-201920-X01);中央高校基本科研业务费专项资金-博士研究生创新基金项目(3072022GIP0403)。

摘　　要：目前,可见光船舶图像的实例分割仍然是较有挑战性的工作。由于船舶图像场景复杂多变,多数实例分割算法无法对复杂场景下的船舶图像进行有效分割。本文提出了基于注意力机制的依靠位置分割目标(attention based segmenting objects by locations,SOLOA)船舶实例分割算法,针对网络特征图里实例信息不完善、背景干扰较多的问题,使用空间注意力机制来充分利用分类特征中的实例信息,建模图像实例间的相互关系并与分割特征相融合。实验结果表明,在新构建的船舶图像数据集上进行训练和测试,改进的网络模型能有效地增强网络特征中的实例信息、减弱背景的干扰。SOLOA算法的船舶实例分割准确率高于其他算法,可以很好地适应复杂场景下的船舶分割需求。At present,the instance segmentation of visible ship images remains a highly challenging task.Most instance segmentation algorithms cannot effectively segment ship images in complex scenes due to the intricate and variable nature of ship images.A segmenting objects by locations based on attention(SOLOA)algorithm for ship instance segmentation,which utilizes the spatial attention mechanism to maximize the instance information in the classification features,is proposed in this paper.Here,the interrelationships between the image instances are modeled and fused with segmentation features.Training and testing results of the newly constructed ship image dataset show that the improved network model can effectively enhance the instance information in the network features and reduce the background interferences.The accuracy of ship instance segmentation by the SOLOA algorithm is higher than that of other algorithms;hence,the proposed algorithm can be effectively adapted to meet the demands of ship segmentation in complex scenes.

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

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]