改进YOLOv10实现大水面死鱼高效精准检测
A method for detecting dead fish on large water surfaces based on improved YOLOv10
- 用FasterNet替换主干网络,提升小目标检测能力
- 在大水面测试中达到92.3%精度、88.7%召回率
- 模型更轻量快速,适合实时部署于养殖系统
由于多种因素,死鱼常出现在水面上。若未及时发现并清除,将导致水质恶化、生态系统破坏及疾病传播。传统人工检测方法受限于人力与时间,难以应对复杂水域环境。本文提出一种基于改进YOLOv10的端到端检测模型,专为大水面死鱼快速精准识别而设计。主要改进包括:(1) 用FasterNet替代原骨干网络,降低模型复杂度同时保持高精度;(2) 通过增强特征融合连接方式,并以CSPStage模块替换原C2f模块优化颈部结构;(3) 增加轻量化检测头,强化小目标检测性能。实验表明,该模型在精度(P)、召回率(R)和平均精度(AP)上均优于基线YOLOv10n。相比其他YOLO系列模型,其参数量显著减少,推理速度更快,且在保持高AP表现的同时实现最优检测效果。消融实验系统评估了各组件对整体性能的贡献。
原文摘要 · Abstract (English)
Dead fish frequently appear on the water surface due to various factors. If not promptly detected and removed, these dead fish can cause significant issues such as water quality deterioration, ecosystem damage, and disease transmission. Consequently, it is imperative to develop rapid and effective detection methods to mitigate these challenges. Conventional methods for detecting dead fish are often constrained by manpower and time limitations, struggling to effectively manage the intricacies of aquatic environments. This paper proposes an end-to-end detection model built upon an enhanced YOLOv10 framework, designed specifically to swiftly and precisely detect deceased fish across extensive water surfaces.Key enhancements include: (1) Replacing YOLOv10's backbone network with FasterNet to reduce model complexity while maintaining high detection accuracy; (2) Improving feature fusion in the Neck section through enhanced connectivity methods and replacing the original C2f module with CSPStage modules; (3) Adding a compact target detection head to enhance the detection performance of smaller objects. Experimental results demonstrate significant improvements in P(precision), R(recall), and AP(average precision) compared to the baseline model YOLOv10n. Furthermore, our model outperforms other models in the YOLO series by significantly reducing model size and parameter count, while sustaining high inference speed and achieving optimal AP performance. The model facilitates rapid and accurate detection of dead fish in large-scale aquaculture systems. Finally, through ablation experiments, we systematically analyze and assess the contribution of each model component to the overall system performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。