Yan Hongwen, Cai Songrui, Li Qiangsheng, Tian Feng, Kan Sitong, Wang Meimeng
College of Information Science and Engineering, Shanxi Agricultural University, Jinzhong 030801, China.
Plants (Basel). 2023 Apr 26;12(9):1769. doi: 10.3390/plants12091769.
Intelligent detection is vital for achieving the intelligent picking operation of daylily, but complex field environments pose challenges due to branch occlusion, overlapping plants, and uneven lighting. To address these challenges, this study selected an intelligent detection model based on YOLOv5s for daylily, the depth and width parameters of the YOLOv5s network were optimized, with Ghost, Transformer, and MobileNetv3 lightweight networks used to optimize the CSPDarknet backbone network of YOLOv5s, continuously improving the model's performance. The experimental results show that the original YOLOv5s model increased mean average precision (mAP) by 49%, 44%, and 24.9% compared to YOLOv4, SSD, and Faster R-CNN models, optimizing the depth and width parameters of the network increased the mAP of the original YOLOv5s model by 7.7%, and the YOLOv5s model with Transformer as the backbone network increased the mAP by 0.2% and the inference speed by 69% compared to the model after network parameter optimization. The optimized YOLOv5s model provided precision, recall rate, mAP, and inference speed of 81.4%, 74.4%, 78.1%, and 93 frames per second (FPS), which can achieve accurate and fast detection of daylily in complex field environments. The research results can provide data and experimental references for developing intelligent picking equipment for daylily.
智能检测对于实现黄花菜的智能采摘作业至关重要,但复杂的田间环境因枝条遮挡、植株重叠和光照不均而带来挑战。为应对这些挑战,本研究选用基于YOLOv5s的黄花菜智能检测模型,对YOLOv5s网络的深度和宽度参数进行优化,采用Ghost、Transformer和MobileNetv3轻量级网络对YOLOv5s的CSPDarknet主干网络进行优化,不断提升模型性能。实验结果表明,原始的YOLOv5s模型相比于YOLOv4、SSD和Faster R-CNN模型,平均精度均值(mAP)分别提高了49%、44%和24.9%,优化网络的深度和宽度参数使原始YOLOv5s模型的mAP提高了7.7%,以Transformer作为主干网络的YOLOv5s模型相比于网络参数优化后的模型,mAP提高了0.2%,推理速度提高了69%。优化后的YOLOv5s模型的精确率、召回率、mAP和推理速度分别为81.4%、74.4%、78.1%和93帧每秒(FPS),能够在复杂田间环境中实现对黄花菜的准确快速检测。研究结果可为开发黄花菜智能采摘设备提供数据和实验参考。