基于改进YOLOv5s的轻量化牛油果成熟度检测方法

扫码查看

原文链接

万方数据
维普

中文摘要：为实现牛油果成熟度的快速精确检测,提升牛油果物流链的自动化水平,提出了一种轻量化YOLOv5s模型检测方法.首先,将RepGhost作为轻量化主干网络引入为模型的主干网络,大幅度加快检测速度;其次,添加激活函数Mish模块,使得模型更好地实现相关特征传递.在此基础上引入数据增强,提升模型泛化能力和模型检测的性能.结果表明,改进后的YOLOv5s模型对未成熟、半成熟、成熟牛油果的检测精度分别提升了 5.1、4.1、9.6 个百分点,平均精度提升了 6.3 个百分点,@map0.5 提升了 7.1 个百分点,检测速度达到 117 帧/s.对比当前主流检测模型YOLOv5s,改进后的模型性能更加优异,能够为牛油果物流链条提供一种轻量化、高精度的自动化成熟度检测方案.

作者：

孙毅、张双德

展开 >

作者单位：

武汉轻工大学电气与电子工程学院湖北武汉 430023

关键词：

牛油果成熟度检测 RepGhost Mish YOLOv5 轻量化

出版年：

2024

DOI：

10.3969/j.issn.1672-9528.2024.10.024

信息技术与信息化

山东电子学会

信息技术与信息化

影响因子：0.29

ISSN：1672-9528

年,卷(期)：2024.(10)