To solve the problem of poor recognition accuracy caused by various colors, uneven distribution, and occlusion by branches and leaves of Camellia oleifera fruits under natural growth conditions, this study proposes an improved deep learning network, YOLOv5s-CEDB, for Camellia oleifera fruit detection based on YOLOv5s. Coordinate Attention Mechanism (CoordAtt), Deformable Convolution (DConv), and Explicit Visual Center (EVC) are introduced to enhance the network’s local and global feature extraction performance. To improve the detection performance for small and dense targets, the feature fusion module of the network was replaced with the designed light-bidirectional feature pyramid (light-BiFPN) structure. GhostConv was used to reduce the parameters and inference speed of the structure. A dataset with different light conditions, colors, and density levels of Camellia oleifera fruits was established, and performance evaluation experiments were conducted. Experimental results showed that the mean Average Precision (mAP) and F1-score of the designed YOLOv5s-CEDB network reached 91.4 % and 89.6 %, respectively, which were 2.6 % and 3.6 % higher than those of the original YOLOv5s model, respectively, and the influencing frame rate arrived at 37.6 FPS. Under different colors, distribution densities, occlusion scenarios, and light intensities, the detection accuracy of the YOLOv5s-CEDB network model was significantly better than those of the YOLOv5s, YOLOv8s and Faster-RCNN networks. It was verified that the proposed YOLOv5s-CEDB network could significantly improve the accuracy and stability of Camellia oleifera fruit detection, satisfying the requirements of yield estimation and intelligent harvesting.
Read full abstract