Enhanced YOLOv5 with ECA Module for Vision-Based Apple Harvesting Using a 6-DOF Robotic Arm in Occluded Environments

Abstract

Accurate target recognition and localization remain significant challenges for robotic fruit harvesting in unstructured orchard environments characterized by branch occlusion and leaf clutter. To address the difficulty in identifying and locating apples under such visually complex conditions, this paper proposes an improved YOLOv5-based visual recognition algorithm incorporating an efficient channel attention (ECA) module. The ECA module is strategically integrated into specific C3 layers (C3-3, C3-6, C3-9) of the YOLOv5 network architecture to enhance feature representation for occluded targets. During operation, the system simultaneously acquires apple pose information and achieves precise spatial localization through coordinate transformation matrices. Comprehensive experimental evaluations demonstrate the effectiveness of the proposed system. The custom-designed six-degree-of-freedom (6-DOF) robotic arm exhibits a wide operational range with a maximum working angle of 120°. The ECA-enhanced YOLOv5 model achieves a confidence level of 90% and an impressive in-range apple recognition rate of 98%, representing a 2.5% improvement in the mean Average Precision (mAP) compared to the baseline YOLOv5s algorithm. The end-effector positioning error is consistently controlled within 1.5 mm. The motion planning success rate reaches 92%, with the picking completed within 23 s per apple. This work provides a novel and effective vision recognition solution for future development of harvesting robots.