一种基于改进YOLOv5算法的航拍鼠洞检测方法及装置

本发明公开了一种基于改进YOLOv5算法的航拍鼠洞检测方法及装置，该方法包括：获取航拍草原图像，进行图像筛选和预处理，建立航拍鼠洞图像数据集；对图像数据集中的图像标注鼠洞位置，构建对应的标签集；将鼠洞图像数据集按预设比例，划分为训练集和验证集；对基于YOLOv5的航拍鼠洞识别模型进行改进；包括：在主干网络中添加ECA注意力模块、推理时利用结构重参数化对C3模块进行压缩；使用标注好的鼠洞图像数据集对改进后的YOLOv5鼠洞检测模型进行训练和优化，得出最优检测模型；基于最优检测模型进行鼠洞检测。本发明建立草原鼠洞数据集，改进YOLOv5通过增加注意力机制和简化特征提取网络构建轻量化模型结构，实现对鼠洞图像的快速、准确检测。