水下无人系统学报 (Oct 2024)

Domain-Adaptive Underwater Target Detection Method Based on GPA + CBAM

  • Qidong LIU,
  • Xin SHEN,
  • Hailu LIU,
  • Lu CONG,
  • Xianping FU

DOI
https://doi.org/10.11993/j.issn.2096-3920.2023-0149
Journal volume & issue
Vol. 32, no. 5
pp. 846 – 854

Abstract

Read online

Underwater target detection is often more susceptible to domain shift and reduced detection accuracy. In response to this phenomenon, this article proposed a domain-adaptive underwater target detection method based on graph-induced prototype alignment(GPA). GPA obtained instance-level features in the image through graph-based information propagation between region proposals and then derived prototype representations for category-level domain alignment. The above operations could effectively aggregate different modal information of underwater targets, thereby achieving alignment between the source and target domains and reducing the impact of domain shift. In addition, in order to make the neural network focus on instance-level features under different water domain distributions, a convolutional block attention module(CBAM) was added. The experimental results have shown that the proposed method can effectively improve detection accuracy during domain shift.

Keywords