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

被引:21
作者
Chen, Guojun [1 ]
Hou, Yongjie [1 ]
Cui, Tao [1 ]
Li, Huihui [1 ]
Shangguan, Fengyang [2 ]
Cao, Lei [3 ,4 ]
机构
[1] China Univ Petr East China, Qingdao Inst Software, Coll Comp Sci & Technol, Qingdao 266580, Peoples R China
[2] Qingdao Univ Sci & Technol, Coll Informat Sci & Technol, Qingdao 266061, Peoples R China
[3] Qilu Univ Technol, Fac Light Ind, Jinan 250300, Peoples R China
[4] Shandong Acad Sci, State Key Lab Biobased Mat & Green Papermaking, Jinan 250300, Peoples R China
来源
SCIENTIFIC REPORTS | 2024年 / 14卷 / 01期
关键词
Attention mechanisms; Color-changing melon dataset; Intelligent agriculture; Target detection; YOLOv8n; REAL-TIME DETECTION; FRUIT DETECTION; SYSTEM; IDENTIFICATION; RESIDUALS; PLANTS;
D O I
10.1038/s41598-024-65293-w
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
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 alpha-IoU loss function is adopted better to measure the overlap between the predicted and real frames using the alpha 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.
引用
收藏
页数:16
相关论文
共 56 条
[1]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, DOI 10.48550/ARXIV.2004.10934]
[2]   Ripening indices and harvesting times of different olive cultivars for continuous harvest [J].
Camposeo, Salvatore ;
Vivaldi, Gaetano Alessandro ;
Gattullo, Concetta Eliana .
SCIENTIA HORTICULTURAE, 2013, 151 :1-10
[3]   Real-Time Detection of Mango Based on Improved YOLOv4 [J].
Cao, Zhipeng ;
Yuan, Ruibo .
ELECTRONICS, 2022, 11 (23)
[4]   Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks [J].
Chen, Jierun ;
Kao, Shiu-Hong ;
He, Hao ;
Zhuo, Weipeng ;
Wen, Song ;
Lee, Chul-Ho ;
Chan, S. -H. Gary .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, :12021-12031
[5]   Detecting ripe fruits under natural occlusion and illumination conditions [J].
Chen, Jiqing ;
Wu, Jiahua ;
Wang, Zhikui ;
Qiang, Hu ;
Cai, Ganwei ;
Tan, Chengzhi ;
Zhao, Chaoyang .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2021, 190
[6]   CitrusYOLO: A Algorithm for Citrus Detection under Orchard Environment Based on YOLOv4 [J].
Chen, Wenkang ;
Lu, Shenglian ;
Liu, Binghao ;
Chen, Ming ;
Li, Guo ;
Qian, Tingting .
MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (22) :31363-31389
[7]   Identification of various food residuals on denim based on hyperspectral imaging system and combination optimal strategy [J].
Chen, Yuzhen ;
Xu, Ziyi ;
Tang, Wencheng ;
Hu, Menghan ;
Tang, Douning ;
Zhai, Guangtao ;
Li, Qingli .
ARTIFICIAL INTELLIGENCE IN AGRICULTURE, 2021, 5 :125-132
[8]   Plant Disease Recognition Model Based on Improved YOLOv5 [J].
Chen, Zhaoyi ;
Wu, Ruhui ;
Lin, Yiyan ;
Li, Chuyu ;
Chen, Siyu ;
Yuan, Zhineng ;
Chen, Shiwei ;
Zou, Xiangjun .
AGRONOMY-BASEL, 2022, 12 (02)
[9]   A proposed framework for accelerating technology trajectories in agriculture: a case study in China [J].
Clark, Beth ;
Jones, Glyn D. ;
Kendall, Helen ;
Taylor, James ;
Cao, Yiying ;
Li, Wenjing ;
Zhao, Chunjiang ;
Chen, Jing ;
Yang, Guijun ;
Chen, Liping ;
Li, Zhenhong ;
Gaulton, Rachel ;
Frewer, Lynn J. .
FRONTIERS OF AGRICULTURAL SCIENCE AND ENGINEERING, 2018, 5 (04) :485-498
[10]   DSConv: Efficient Convolution Operator [J].
do Nascimento, Marcelo Gennari ;
Fawcett, Roger ;
Prisacariu, Victor Adrian .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :5147-5156