Underwater object detection faces significant challenges, including ambiguity and occlusion, which greatly undermine the accuracy of traditional algorithms. To address these issues, we propose Bi2F-YOLO, an algorithm, specifically designed for underwater environments. Bi2F-YOLO integrates the BiFormer module into the YOLOv7 backbone, utilizing Bi-Level routing attention (BRA) to focus on key features such as object edges and textures. This effectively addresses the problem of object ambiguity. In the detection head, we replace the conventional ELAN component with the FasterNet module. This update enhances detection efficiency and accuracy through the use of partial convolution (PConv), which redistributes the convolution kernel weights based on the sparsity of the input feature map. By doing so, it prevents the dilution of critical underwater object features caused by interference from irrelevant data. This effectively resolves the occlusion problem in underwater target detection while simultaneously reducing model parameters and computational costs. The experimental results show that Bi2F-YOLO achieves 87.3% <InlineEquation ID="IEq1"> <InlineMediaObject> <ImageObject Color="BlackWhite" FileRef="44295_2025_60_Article_IEq1.gif" Format="GIF" Height="16" Rendition="HTML" Resolution="72" Type="Linedraw" Width="54" /> </InlineMediaObject> <EquationSource Format="TEX">${{mAP}}_{50}$</EquationSource> <EquationSource Format="MATHML"><math> <msub> <mrow> <mi mathvariant="italic">mAP</mi> </mrow> <mn>50</mn> </msub> </math></EquationSource> </InlineEquation> and 86.8% <InlineEquation ID="IEq2"> <InlineMediaObject> <ImageObject Color="BlackWhite" FileRef="44295_2025_60_Article_IEq2.gif" Format="GIF" Height="16" Rendition="HTML" Resolution="72" Type="Linedraw" Width="39" /> </InlineMediaObject> <EquationSource Format="TEX">${{AP}}_{50}$</EquationSource> <EquationSource Format="MATHML"><math> <msub> <mrow> <mi mathvariant="italic">AP</mi> </mrow> <mn>50</mn> </msub> </math></EquationSource> </InlineEquation> on the RUOD dataset. Compared to existing methods, it reduces model parameters by 14.3% fewer parameters (Params) and lowers computational costs by 5.5%, highlighting its efficiency and accuracy in underwater object detection.

Bi2F-YOLO: a novel framework for underwater object detection based on YOLOv7

期刊论文
出版模式：

开放获取
发表日期：

2025年3月26日

查看全文

Xiaopeng Liu,
Keke Zhao,
Cong Liu,
Long Chen

摘要

Underwater object detection faces significant challenges, including ambiguity and occlusion, which greatly undermine the accuracy of traditional algorithms. To address these issues, we propose Bi2F-YOLO, an algorithm, specifically designed for underwater environments. Bi2F-YOLO integrates the BiFormer module into the YOLOv7 backbone, utilizing Bi-Level routing attention (BRA) to focus on key features such as object edges and textures. This effectively addresses the problem of object ambiguity. In the detection head, we replace the conventional ELAN component with the FasterNet module. This update enhances detection efficiency and accuracy through the use of partial convolution (PConv), which redistributes the convolution kernel weights based on the sparsity of the input feature map. By doing so, it prevents the dilution of critical underwater object features caused by interference from irrelevant data. This effectively resolves the occlusion problem in underwater target detection while simultaneously reducing model parameters and computational costs. The experimental results show that Bi2F-YOLO achieves 87.3% ${{mAP}}_{50}$ ${mAP}_{50}$ and 86.8% ${{AP}}_{50}$ ${AP}_{50}$ on the RUOD dataset. Compared to existing methods, it reduces model parameters by 14.3% fewer parameters (Params) and lowers computational costs by 5.5%, highlighting its efficiency and accuracy in underwater object detection.