当前位置: 首页 > 文章 > 融合超分辨率重建的YOLOv5松枯死木识别模型 农业工程学报 2023 (5) 137-145
Position: Home > Articles > 融合超分辨率重建的YOLOv5松枯死木识别模型 Transactions of the Chinese Society of Agricultural Engineering 2023 (5) 137-145

融合超分辨率重建的YOLOv5松枯死木识别模型

作  者:
王文瑾;游子绎;邵历江;李小林;吴松青;张珠河;黄世国;张飞萍
单  位:
福建农林大学林学院;福建农林大学计算机与信息学院;福州市森林病虫害防治检疫站
关键词:
无人机;图像识别;松枯死木;小目标检测;超分辨率重建;特征融合
摘  要:
为解决山地地形起伏大、无人机飞行高度高导致图像中尺度小且纹理模糊的松枯死木识别困难问题,该研究提出了一种在特征层级进行超分辨率重建的YOLOv5松枯死木识别算法。在YOLOv5网络中添加选择性核特征纹理迁移模块生成有细节纹理的高清检测特征图,自适应改变感受野的机制分配权重,将更多注意力集中在纹理细节,提升了小目标和模糊目标的识别精度。同时,使用前景背景平衡损失函数抑制背景噪声干扰,增加正样本的梯度贡献,改善正负样本分布不平衡问题。试验结果表明,改进后算法在交并比(intersection over union,IoU)阈值取0.5时的平均精度均值(mean average precision,mAP50)为92.7%,mAP50~95(以步长0.05从0.5到0.95间取IoU阈值下的平均mAP)为62.1%,APsmall(小目标平均精度值)为53.2%,相比于原算法mAP50提高了3.2个百分点,mAP50~95提升了8.3个百分点,APsmall提升了15.8个百分点。不同算法对比分析表明,该方法优于Faster R-CNN、YOLOv4、YOLOX、MT-YOLOv6,QueryDet、DDYOLOv5等深度学习算法,mAP50分别提高了16.7、15.3、2.5、2.8、12.3和1.2个百分点。改进后松枯死木识别算法具有较高精度,有效缓解了小目标与纹理模糊目标识别困难问题,为后续疫木清零提供技术支持。

相似文章

计量
文章访问数: 17
HTML全文浏览量: 0
PDF下载量: 0

所属期刊

推荐期刊