Suppr超能文献

YOLOv8-CML:一种用于智能农业中变色甜瓜成熟度的轻量级目标检测方法。

YOLOv8-CML: a lightweight target detection method for color-changing melon ripening in intelligent agriculture.

作者信息

Chen Guojun, Hou Yongjie, Cui Tao, Li Huihui, Shangguan Fengyang, Cao Lei

机构信息

Qingdao Institute of Software, College of Computer Science and Technology, China University of Petroleum (East China), Qingdao, 266580, China.

College of Information Science and Technology, Qingdao University of Science and Technology, Qingdao, 266100, China.

出版信息

Sci Rep. 2024 Jun 22;14(1):14400. doi: 10.1038/s41598-024-65293-w.

Abstract

Color-changing melon is an ornamental and edible fruit. Aiming at the problems of slow detection speed and high deployment cost for Color-changing melon in intelligent agriculture equipment, this study proposes a lightweight detection model YOLOv8-CML.Firstly, a lightweight Faster-Block is introduced to reduce the number of memory accesses while reducing redundant computation, and a lighter C2f structure is obtained. Then, the lightweight C2f module fusing EMA module is constructed in Backbone to collect multi-scale spatial information more efficiently and reduce the interference of complex background on the recognition effect. Next, the idea of shared parameters is utilized to redesign the detection head to simplify the model further. Finally, the α-IoU loss function is adopted better to measure the overlap between the predicted and real frames using the α hyperparameter, improving the recognition accuracy. The experimental results show that compared to the YOLOv8n model, the parametric and computational ratios of the improved YOLOv8-CML model decreased by 42.9% and 51.8%, respectively. In addition, the model size is only 3.7 MB, and the inference speed is improved by 6.9%, while mAP@0.5, accuracy, and FPS are also improved. Our proposed model provides a vital reference for deploying Color-changing melon picking robots.

摘要

变色瓜是一种兼具观赏性和食用性的水果。针对智能农业设备中变色瓜检测速度慢、部署成本高的问题,本研究提出了一种轻量级检测模型YOLOv8-CML。首先,引入轻量级Faster-Block以减少内存访问次数并减少冗余计算,从而得到更轻量级的C2f结构。然后,在主干网络中构建融合EMA模块的轻量级C2f模块,以更高效地收集多尺度空间信息,并减少复杂背景对识别效果的干扰。接下来,利用共享参数的思想重新设计检测头,进一步简化模型。最后,采用α-IoU损失函数,利用α超参数更好地衡量预测框与真实框之间的重叠度,提高识别准确率。实验结果表明,与YOLOv8n模型相比,改进后的YOLOv8-CML模型的参数比和计算比分别下降了42.9%和51.8%。此外,模型大小仅为3.7MB,推理速度提高了6.9%,同时mAP@0.5、准确率和FPS也得到了提高。我们提出的模型为部署变色瓜采摘机器人提供了重要参考。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/4581/11193782/dbf4c42f369c/41598_2024_65293_Fig1_HTML.jpg

文献检索

告别复杂PubMed语法,用中文像聊天一样搜索,搜遍4000万医学文献。AI智能推荐,让科研检索更轻松。

立即免费搜索

文件翻译

保留排版,准确专业,支持PDF/Word/PPT等文件格式,支持 12+语言互译。

免费翻译文档

深度研究

AI帮你快速写综述,25分钟生成高质量综述,智能提取关键信息,辅助科研写作。

立即免费体验