The intelligent detection of young apple fruits based on deep learning faced various challenges such as varying scale sizes and colors similar to the background, which increased the risk of misdetection or missed detection. To effectively address these issues, a method for young apple fruit detection based on improved YOLOv5 was proposed in this paper. Firstly, a young apple fruits dataset was established. Subsequently, a prediction layer was added to the detection head of the model, and four layers of CA attention mechanism were integrated into the detection neck (Neck). Additionally, the GIOU function was introduced as the model's loss function to enhance its overall detection performance. The accuracy on the validation dataset reached 94.6%, with an average precision of 82.2%. Compared with YOLOv3, YOLOv4, and the original YOLOv5 detection methods, the accuracy increased by 0.4%, 1.3%, and 4.6% respectively, while the average precision increased by 0.9%, 1.6%, and 1.2% respectively. The experiments demonstrated that the algorithm effectively recognized young apple fruits in complex scenes while meeting real-time detection requirements, providing support for intelligent apple orchard management.
Real Time Impact Factor:
Pending
Author Name: Yonghui DU, Ang GAO, Yuepeng SONG, Jing GUO, Wei MA, Longlong REN
URL: View PDF
Keywords: YOLOv5; Young apple fruit; Prediction layer; CA attention mechanism; GIOU loss function
ISSN: 2068 - 4215
EISSN: 2068 - 2239
EOI/DOI: 10.35633/inmateh-73-07
Add Citation
Views: 1