首页 >

基于多模型协同的船舰目标检测(备注弱小目标检测专栏)

肖欣林,施伟超,郑向涛,高跃明,卢孝强   

  1. 福州大学
  • 收稿日期:2024-01-26 修回日期:2024-04-03 出版日期:2024-04-10 发布日期:2024-04-10
  • 通讯作者: 郑向涛
  • 基金资助:
    国家自然科学基金项目(62271484)

Multiple models collaboration for Ship target detection

  • Received:2024-01-26 Revised:2024-04-03 Online:2024-04-10 Published:2024-04-10

摘要: 在全球化背景下,船舰监测的重要性日益凸显,并且随着遥感成像技术的不断进步,船舰目标检测也成为确保海上运输安全和效率的关键手段,对海上交通、环境保护及国家安全至关重要。然而,由于船舰目标尺度差异大、背景复杂等问题,现有单一模型的方法过渡依赖训练数据,无法适应尺度多变的舰船目标。本文提出了一种多模型协同训练的框架,将多个已训练好的舰船检测模型视作辅助,通过知识迁移的方式优化主网络中的训练。首先,在辅助网络与主网络中引入三元关系约束传递知识,随后采用软标签引导策略进一步提高船舶检测的准确性。实验结果表明,相较于现有主流方法,本文所提出的方法在DOTA和xView数据集上展示了较好的性能,克服了单一模型的局限性,为遥感图像的目标检测提供了新的解决思路。

关键词: 舰船识别, 目标检测, 多尺度表达, 多模型协同, 知识融合

Abstract: In the context of globalization, the importance of ship monitoring is becoming more and more prominent, moreover, with the continuous progress of remote sensing imaging technology, ship target detection has also become a key means to ensure the safety and efficiency of maritime transportation, which is crucial for maritime transportation, environmental protection and national security. However, due to the large-scale difference and complex background of ship targets, the existing approach relies on transitional manner, it can’t adapt to ship targets with variable scales. In this paper, we propose a multi-model co-training framework, in which multiple trained ship detection models are regarded as auxiliary, optimizing the main network training by knowledge migration. First, ternary relationship constraints are introduced to transfer knowledge between the auxiliary network and the main network, followed by a soft-label guidance strategy to further improve the accuracy of ship detection. The experimental results show that compared with the existing main-stream methods, the proposed method demonstrates better performance on DOTA and xView datasets, overcomes the limitation of a single model, and provides a new solution idea for target detection in remote sensing images.

Key words: ship recognition, target detection, multi-scale representation, multi-model collaboration, knowledge fusion