一种轻量级YOLOv5S农作物虫害目标检测模型

郭小燕, 于帅卿

PDF(4816 KB)
PDF(4816 KB)
南京农业大学学报 ›› 2024, Vol. 47 ›› Issue (05) : 1009-1018.

一种轻量级YOLOv5S农作物虫害目标检测模型

  • 郭小燕, 于帅卿
作者信息 +
History +

摘要

[目的]本文提出一种轻量级YOLOv5S农作物虫害目标检测模型以解决在样本数量不足的情况下农作物虫害识别的问题。[方法]利用Ghost技术将2个Ghost Bottle Block线性特征提取模块封装为1个GB模块,代替YOLOv5S中前7个CBL、CSP、SPP非线性特征提取模块,从而约简了YOLOv5S的网络参数,减轻了网络体量。[结果]在保证虫害检测效果的前提下降低网络对计算硬件与训练样本的依赖。为了验证模型的有效性,对水稻、玉米、棉花、马铃薯、苜蓿5种农作物的12类虫害进行识别与定位,多类别平均精度(mAP)为91.31%,比YOLOv5S模型高出2.56百分点。[结论]通过与SSD、Faster-RCNN、YOLOv5S模型对比发现,本文提出的模型在mAP、F1-score、精确率(Precision)、召回率(Recall)4个评价指标均占优势,尤其在小目标、密集目标、与背景相似目标的检测方面表现突出。

关键词

农作物 / 虫害 / YOLOv5S / 轻量级 / 目标检测

中图分类号

TP183 / TP391.41 / S43

引用本文

导出引用
郭小燕, 于帅卿. 一种轻量级YOLOv5S农作物虫害目标检测模型. 南京农业大学学报. 2024, 47(05): 1009-1018

基金

国家自然科学基金项目(32360437); 甘肃农业大学青年导师基金(QAU-QDFC-2021-18);甘肃农业大学科技创新基金(盛彤笙创新基金)(GSAU-STS-2021-16)

评论

PDF(4816 KB)

Accesses

Citation

Detail

段落导航
相关文章

/