Addressing the current reliance on manual sorting and grading of spray rose cut flowers, this paper proposed an improved YOLOv5s model for intelligent recognition and grading detection of rose color series and flowering index of spray rose cut flowers. By incorporating small-scale anchor boxes and small object feature output, the model enhanced the annotation accuracy and the detection precision for occluded rose flowers. Additionally, a convolutional block attention module attention mechanism was integrated into the original network structure to improve the model’s feature extraction capability. The WIoU loss function was employed in place of the original CIoU loss function to increase the precision of the model’s post-detection processing. Test results indicated that for two types of spray rose cut flowers, Orange Bubbles and Yellow Bubbles, the improved YOLOv5s model achieved an accuracy and recall improvement of 10.2% and 20.0%, respectively. For randomly collected images of spray rose bouquets, the model maintained a detection accuracy of 95% at a confidence threshold of 0.8.
Read full abstract