1College of Agricultural Engineering, Shanxi Agricultural University, Taigu, 030800, China
2Dryland Farm Machinery Key Technology and Equipment Key Laboratory of Shanxi Province, Taigu, 030800, China
*Corresponding Author's E-mail Address: zsujuan1@163.com
Online published on 6 June, 2025.
This study introduces an improved YOLOv8 (You Only Look Once) model YOLOFSL for rapid detection of plum maturity under natural conditions. Firstly, the Faster Net Block module was introduced into the CSP (Cross Stage Partial) Bottleneck with 2 convolutions (C2f) module of the YOLOv8s backbone network, forming the C2f_Faster module. Secondly, squeeze and excitation (SE) attention mechanism after each C2f_Faster module was added in the backbone network. Finally, the improved YOLOv8s model was pruned using channel pruning algorithm to simplify the detection model and ensure detection efficiency. The experimental results show that the mean average precision (mAP) and model size of the improved YOLO-FSL model is 91.2% and 4.1 MB (Mega Byte), respectively, which reduces the number of parameters by 82.57% and improves the mAP by 1.6% compared with the original network YOLOv8s. Compared with the mainstream target detection algorithms of YOLOv3-tiny, YOLOv5s, YOLOv6s, and YOLOv7-tiny, the mAP of the YOLO-FSL model is improved by 4.4%, 2.0%, 1.0%, and 1.6%, respectively. The YOLO-FSL model has a higher detection accuracy, and at the same time reduces the amount of computation and the number of parameters, and lowers the hardware dependence, which provides a technical reference for the automated picking of plums.
Deep learning, Lightweight model, Mean average precision, Object detection, Pruning techniques, Squeeze and excitation attention mechanism