Abstract:The precision detection of dense small targets in orchards is critical for the visual perception of agricultural picking robots. At present, the visual detection algorithms for plums still have a poor recognition effect due to the characteristics of small plum shapes and dense growth. Thus, this paper proposed a lightweight model based on the improved You Only Look Once version 4 (YOLOv4) to detect dense plums in orchards. First, we employed a data augmentation method based on category balance to alleviate the imbalance in the number of plums of different maturity levels and insufficient data quantity. Second, we abandoned Center and Scale Prediction Darknet53 (CSPDarknet53) and chose a lighter MobilenetV3 on selecting backbone feature extraction networks. In the feature fusion stage, we used depthwise separable convolution (DSC) instead of standard convolution to achieve the purpose of reducing model parameters. To solve the insufficient feature extraction problem of dense targets, this model achieved fine-grained detection by introducing a 152 × 152 feature layer. The Focal loss and complete intersection over union (CIOU) loss were joined to balance the contribution of hard-to-classify and easy-to-classify samples to the total loss. Then, the improved model was trained through transfer learning at different stages. Finally, several groups of detection experiments were designed to evaluate the performance of the improved model. The results showed that the improved YOLOv4 model had the best mean average precision (mAP) performance than YOLOv4, YOLOv4-tiny, and MobileNet-Single Shot Multibox Detector (MobileNet-SSD). Compared with some results from the YOLOv4 model, the model size of the improved model is compressed by 77.85%, the parameters are only 17.92% of the original model parameters, and the detection speed is accelerated by 112%. In addition, the influence of the automatic data balance algorithm on the accuracy of the model and the detection effect of the improved model under different illumination angles, different intensity levels, and different types of occlusions were discussed in this paper. It is indicated that the improved detection model has strong robustness and high accuracy under the real natural environment, which can provide data reference for the subsequent orchard yield estimation and engineering applications of robot picking work.

A Lightweight Improved YOLOv5s Model and Its Deployment for Detecting Pitaya Fruits in Daytime and Nighttime Light-Supplement Environments

A Lightweight and High-Precision Passion Fruit YOLO Detection Model for Deployment in Embedded Devices

Improved YOLOv4 recognition algorithm for pitaya based on coordinate attention and combinational convolution

Lightweight Apple Detection in Complex Orchards Using YOLOV5-PRE

YOLO-P: An efficient method for pear fast detection in complex orchard picking environment

YOLOC-tiny: a generalized lightweight real-time detection model for multiripeness fruits of large non-green-ripe citrus in unstructured environments

GSE-YOLO: A Lightweight and High-Precision Model for Identifying the Ripeness of Pitaya (Dragon Fruit) Based on the YOLOv8n Improvement

Fast and precise detection of litchi fruits for yield estimation based on the improved YOLOv5 model

Real-time and Accurate Detection of Citrus in Complex Scenes Based on HPL-YOLOv4.

Lightweight Pear Detection Algorithm Based on Improved YOLOv5

Dragon fruit detection in natural orchard environment by integrating lightweight network and attention mechanism

Precision Detection of Dense Plums in Orchards Using the Improved YOLOv4 Model

PeachYOLO: A Lightweight Algorithm for Peach Detection in Complex Orchard Environments

Precision Detection of Dense Litchi Fruit in UAV Images Based on Improved YOLOv5 Model

Lightweight Algorithm for Apple Detection Based on an Improved YOLOv5 Model

Using an improved lightweight YOLOv8 model for real-time detection of multi-stage apple fruit in complex orchard environments

Apple target recognition method in complex environment based on improved YOLOv4

Study on the Detection Method for Daylily Based on YOLOv5 under Complex Field Environments

YOLOv8n-CSD: A Lightweight Detection Method for Nectarines in Complex Environments

Xiaomila Green Pepper Target Detection Method under Complex Environment Based on Improved YOLOv5s

Rapid detection of Yunnan Xiaomila based on lightweight YOLOv7 algorithm