arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.14211cs.CVcs.LG

提升YOLOv8在相机陷阱目标检测中的泛化性能

Improving Generalization Performance of YOLOv8 for Camera Trap Object Detection

University of Cincinnati(辛辛那提大学);College of Engineering and Applied Science(工程与应用科学学院);Graduate College(研究生院)

Aroj Subedi

AI总结:

本文针对YOLOv8在相机陷阱目标检测中的泛化问题,提出引入GAM模块、改进多尺度特征融合及WIoUv3损失函数,显著提升了模型在未见数据集上的泛化与抗噪能力。

摘要:

相机陷阱已成为野生动物保护的重要工具,提供了非侵入性的手段来监测和研究自然栖息地中的野生动物。利用目标检测算法自动识别相机陷阱图像中的物种对研究和保护目的具有重大意义。然而,泛化问题普遍存在,即训练好的模型无法将其学到的知识应用于从未见过的数据集。本论文探讨了对YOLOv8目标检测算法的改进,以解决泛化问题。本研究深入分析了基线YOLOv8模型的局限性,强调了其在真实世界环境中泛化能力不足的困境。为克服这些局限性,提出了多项改进,包括引入全局注意力机制(GAM)模块、改进的多尺度特征融合,以及将Wise Intersection over Union(WIoUv3)作为边界框回归损失函数。全面的评估和消融实验表明,改进后的模型能够抑制背景噪声,聚焦于目标属性,并在新环境中展现出稳健的泛化能力。所提出的改进不仅解决了相机陷阱数据集固有的挑战,还为在真实世界保护场景中的更广泛应用铺平了道路,最终有助于有效管理野生动物种群及其栖息地。

英文摘要:

Camera traps have become integral tools in wildlife conservation, providing non-intrusive means to monitor and study wildlife in their natural habitats. The utilization of object detection algorithms to automate species identification from Camera Trap images is of huge importance for research and conservation purposes. However, the generalization issue, where the trained model is unable to apply its learnings to a never-before-seen dataset, is prevalent. This thesis explores the enhancements made to the YOLOv8 object detection algorithm to address the problem of generalization. The study delves into the limitations of the baseline YOLOv8 model, emphasizing its struggles with generalization in real-world environments. To overcome these limitations, enhancements are proposed, including the incorporation of a Global Attention Mechanism (GAM) module, modified multi-scale feature fusion, and Wise Intersection over Union (WIoUv3) as a bounding box regression loss function. A thorough evaluation and ablation experiments reveal the improved model's ability to suppress the background noise, focus on object properties, and exhibit robust generalization in novel environments. The proposed enhancements not only address the challenges inherent in camera trap datasets but also pave the way for broader applicability in real-world conservation scenarios, ultimately aiding in the effective management of wildlife populations and habitats.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑