江西农业大学学报2024,Vol.46Issue(4):1025-1036,12.DOI:10.3724/aauj.2024091
基于改进YOLOv5s的自然环境下番茄成熟度检测方法
Tomato ripening detection in natural environment based on improved YOLOv5s
摘要
Abstract
[Objective]In the tomato recognition task,the existing target recognition algorithms are slow and have low accuracy in recognizing occluded tomatoes and small tomatoes,which affects their deployment and application on embedded devices.In order to realize the fast and accurate recognition of tomato fruits by agricultural robots in complex environments,this study proposes a tomato ripeness recognition method based on the improved YOLOv5s model.[Method]Combined with the distribution characteristics of tomato growing environment,MobileNetv3 network and ECANet(Efficient Channel Attention Network)mechanism were introduced to improve the YOLOv5s target detection algorithm.[Result]Compared with YOLOv5,the results of the improved YOLOv5s(Im-YOLOv5s)showed that the accuracy,recall,and average accuracy were improved by 3.4%,2.4%,and 2.3%,the weight size was reduced by 48.6%,and the detection speed was increased by 52.9%,which improved the detection performance and shortened the model inference time.[Conclusion]Compared with a variety of mainstream target detection models,the improved YOLOv5s algorithm greatly reduces the omission and false detection of tomato ripeness.With better recognition effect,good robustness and real-time performance,it meets the demand for accurate real-time identification of tomatoes with different ripeness,and it is suitable for the deployment and application on embedded devices,thus providing technical support for automated tomato picking.关键词
YOLOv5s/番茄/成熟度检测/深度学习/MobileNetV3/机器视觉Key words
YOLOv5s/tomatoes/ripening detection/deep learning/MobileNetV3/machine vision分类
农业科技引用本文复制引用
常文龙,谭钰,周立峰,杨启良..基于改进YOLOv5s的自然环境下番茄成熟度检测方法[J].江西农业大学学报,2024,46(4):1025-1036,12.基金项目
国家自然科学基金项目(52379041)Project supported by the National Natural Science Foundation of China(52379041) (52379041)