An improved YOLO algorithm for detecting flowers and fruits on strawberry seedlings

Abstract

Accurately identifying the flowers and fruits of strawberry seedlings in the greenhouse is the key to automated flower and fruit thinning, which can improve efficiency and reduce labour costs in the cultivation. To address the challenges resulting from the small size, similar colour, and overlapping occlusion of strawberry seedling flowers and fruits, this paper proposes a real-time recognition algorithm (Improved YOLO) for accurately identifying them. Firstly, a Swin Transformer prediction head on the high-resolution feature map of YOLO v7 was constructed to better utilise spatial location information to enhance the detection of small target flowers and fruits, and improve the model's spatial interaction and feature extraction ability in scenes with similar colours and overlapping occlusions. Secondly, the GS-ELAN Optimisation Module for neck of network by GSConv was constructed to suppress shallow noise interference from the high-resolution prediction head and mitigate the increase of parameters resulting from high-resolution prediction heads. The experimental results showed that the Precision(P), Recall(R), and mean Average Precision (mAP) of Improved YOLO are 92.6%, 89.6%, and 92.1%. In the meantime, the Improved YOLO algorithm achieves a frame rate of 45 f/s, satisfying the real-time detection requirements. It is 3.2%, 2.7%, and 4.6% higher than the precision, recall, and mAP of YOLOv7, respectively. The accuracy of this model for detecting flowers and fruits was 93.9% and 91.3%, the recall was 93% and 86.3%, and the average precision was 94.7% and 89.5%, respectively. The Improved YOLO algorithm has a high level of robustness and real-time detection performance, allowing it to quickly and accurately identify the flowers and fruits of strawberry seedlings and provides effective support for the automated management of flower and fruit thinning of strawberry seedlings in greenhouse environments.