《农业工程学报》2024年第40卷第14期刊载了山东理工大学等单位顾洪宇、李志合、李涛、李天豪、李宁与魏忠彩的论文——“基于YOLOv5的马铃薯种薯芽眼轻量化检测算法”。该研究由国家自然科学基金项目(项目号:52105266)等资助。
引文信息:顾洪宇,李志合,李涛,等. 基于YOLOv5的马铃薯种薯芽眼轻量化检测算法[J]. 农业工程学报,2024,40(14):126-136.
DOI: 10.11975/j.issn.1002-6819.202404033
为了实现种薯芽眼的精准检测,方便后续实现马铃薯种薯的智能化切块,该研究提出一种基于深度学习一阶段目标检测算法YOLO的种薯芽眼检测改进模型。
改进后的模型在YOLOv5检测模型基础上引入C3 Faster,降低参数量的同时加强了芽眼特征的提取能力;引入GOLD-YOLO中信息聚集-分发结构,提高模型检测芽眼的准确性;使用WIoU Loss代替CIoU Loss作为边界框损失函数,加快网络模型收敛的同时提高检测精度;使用遗传算法对超参数进行优化;最后使用剪枝与蒸馏技术,降低模型运行参数量与运行内存。
优化后的模型大小为8.7 MB,仅为原始模型的61.3%,模型参数量约为原始模型的57.1%,最终的检测平均精确度在自制的种薯数据集中的测试集与验证集上分别为90.5%以及90.1%,
该改进模型于自制种薯数据集的测试集上相较同类型的轻量级网络YOLOv7-tiny、YOLOv8n、YOLOv5n、YOLOv5s,平均精度均值分别高出0.5、1.3、2.8、1.1个百分点,在验证集上平均精度均值分别高出2.9、1.9、3.2、1.6个百分点,在本地计算机上检测速度达到了27.5帧/s,该研究结果可为后续种薯芽眼识别及实时切块技术提供参考。
本文由丨《农业工程学报》编辑部丨精编发布
欢迎留言、分享、点赞
转载、投稿、咨询
邮箱:tcsae@tcsae.org
发布征集
欢迎广大作者、读者投稿至我刊公众号,包括但不限于创新科研成果、科技写作技巧、书籍推介、优秀科技工作者介绍、科研团队招聘/招生、行业资讯以及相关活动等农业工程领域信息。
公众号文章投稿邮箱:abe-newmedia@tcsae.org
»» 点击 阅读原文 免费获取全文