The real-time detection transformer has been widely used in infrared ship detection, but there are shortcomings in terms of computational complexity and feature fusion. Therefore, an infrared vessel detection model based on bidirectional selective path aggregation network was proposed in this paper. In this model, the StarNet was introduced to replace the traditional ResNet as the backbone network, and the lightweight of the model was achieved while feature extraction capabilities were maintained. It adopted an efficient additive attention mechanism to replace the original multi-head self-attention, reducing computational complexity and enhancing the perception accuracy of key ship areas. At the same time, to solve the problem of information loss in feature fusion at different scales, a bidirectional routing aggregation network optimization feature fusion strategy was proposed, which can better deal with ship targets of different sizes through semantic enhancement feature selection module and detailed perception feature fusion module. The experimental results show that the model exhibits excellent detection performance on the infrared ship dataset. It ensures detection accuracy and achieves higher real-time processing capabilities, and can effectively adapt to complex and changing sea surface monitoring environments.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

An Infrared Vessel Detection Model Based on Bidirectional Selective Path Aggregation Network

  • Zhicong Lin,
  • Jichang Chen,
  • Taoshen Li

摘要

The real-time detection transformer has been widely used in infrared ship detection, but there are shortcomings in terms of computational complexity and feature fusion. Therefore, an infrared vessel detection model based on bidirectional selective path aggregation network was proposed in this paper. In this model, the StarNet was introduced to replace the traditional ResNet as the backbone network, and the lightweight of the model was achieved while feature extraction capabilities were maintained. It adopted an efficient additive attention mechanism to replace the original multi-head self-attention, reducing computational complexity and enhancing the perception accuracy of key ship areas. At the same time, to solve the problem of information loss in feature fusion at different scales, a bidirectional routing aggregation network optimization feature fusion strategy was proposed, which can better deal with ship targets of different sizes through semantic enhancement feature selection module and detailed perception feature fusion module. The experimental results show that the model exhibits excellent detection performance on the infrared ship dataset. It ensures detection accuracy and achieves higher real-time processing capabilities, and can effectively adapt to complex and changing sea surface monitoring environments.