A Real-Time Apple Targets Detection Method for Picking Robot Based on Improved YOLOv5

The apple target recognition algorithm is one of the core technologies of the apple picking robot. However, most of the existing apple detection algorithms cannot distinguish between the apples that are occluded by tree branches and occluded by other apples. The apples, grasping end-effector and mechanical picking arm of the robot are very likely to be damaged if the algorithm is directly applied to the picking robot. Based on this practical problem, in order to automatically recognize the graspable and ungraspable apples in an apple tree image, a light-weight apple targets detection method was proposed for picking robot using improved YOLOv5s. Firstly, BottleneckCSP module was improved designed to BottleneckCSP-2 module which was used to replace the BottleneckCSP module in backbone architecture of original YOLOv5s network. Secondly, SE module, which belonged to the visual attention mechanism network, was inserted to the proposed improved backbone network. Thirdly, the bonding fusion mode of feature maps, which were inputs to the target detection layer of medium size in the original YOLOv5s network, were improved. Finally, the initial anchor box size of the original network was improved. The experimental results indicated that the graspable apples, which were unoccluded or only occluded by tree leaves, and the ungraspable apples, which were occluded by tree branches or occluded by other fruits, could be identified effectively using the proposed improved network model in this study. Specifically, the recognition recall, precision, mAP and F1 were 91.48%, 83.83%, 86.75% and 87.49%, respectively. The average recognition time was 0.015 s per image. Contrasted with original YOLOv5s, YOLOv3, YOLOv4 and EfficientDet-D0 model, the mAP of the proposed improved YOLOv5s model increased by 5.05%, 14.95%, 4.74% and 6.75% respectively, the size of the model compressed by 9.29%, 94.6%, 94.8% and 15.3% respectively. The average recognition speeds per image of the proposed improved YOLOv5s model were 2.53, 1.13 and 3.53 times of EfficientDet-D0, YOLOv4 and YOLOv3 and model, respectively. The proposed method can provide technical support for the real-time accurate detection of multiple fruit targets for the apple picking robot.

[1]  Maya Malinda,et al.  The forecasting of consumer exchange-traded funds (ETFs) via grey relational analysis (GRA) and artificial neural network (ANN) , 2021, Empirical Economics.

[2]  G. Vosniakos,et al.  An agent-based Flexible Manufacturing System controller with Petri-net enabled algebraic deadlock avoidance , 2020, Reports in Mechanical Engineering.

[3]  Jorge Antonio Silva Centeno,et al.  ATSS Deep Learning-Based Approach to Detect Apple Fruits , 2020, Remote. Sens..

[4]  Xin Zhang,et al.  Faster R–CNN–based apple detection in dense-foliage fruiting-wall trees using RGB and depth features for robotic harvesting , 2020 .

[5]  Xiaoling Zhang,et al.  High-Speed Ship Detection in SAR Images Based on a Grid Convolutional Neural Network , 2019, Remote. Sens..

[6]  Yiannis Ampatzidis,et al.  Laboratory and UAV-Based Identification and Classification of Tomato Yellow Leaf Curl, Bacterial Spot, and Target Spot Diseases in Tomato Utilizing Hyperspectral Imaging and Machine Learning , 2020, Remote. Sens..

[7]  Jing Zhang,et al.  Multi-class object detection using faster R-CNN and estimation of shaking locations for automated shake-and-catch apple harvesting , 2020, Comput. Electron. Agric..

[8]  Verónica Vilaplana,et al.  Multi-modal deep learning for Fuji apple detection using RGB-D cameras and their radiometric capabilities , 2019, Comput. Electron. Agric..

[9]  Matthias Schubert,et al.  Automated Detection of Conifer Seedlings in Drone Imagery Using Convolutional Neural Networks , 2019, Remote. Sens..

[10]  Edgar Roman-Rangel,et al.  Fig Plant Segmentation from Aerial Images Using a Deep Convolutional Encoder-Decoder Network , 2019, Remote. Sens..

[11]  Rui Li,et al.  Application of consumer RGB-D cameras for fruit detection and localization in field: A critical review , 2020, Comput. Electron. Agric..

[12]  Xin Zhang,et al.  Multi-class fruit-on-plant detection for apple in SNAP system using Faster R-CNN , 2020, Comput. Electron. Agric..

[13]  Chao Chen,et al.  Fast implementation of real-time fruit detection in apple orchards using deep learning , 2020, Comput. Electron. Agric..

[14]  Herzegovina,et al.  ANFIS model for the prediction of generated electricity of photovoltaic modules , 2019 .

[15]  Radu-Emil Precup,et al.  Model-based fuzzy control results for networked control systems , 2020 .

[16]  Dongjian He,et al.  Combining an information-maximization-based attention mechanism and illumination invariance theory for the recognition of green apples in natural scenes , 2020, Multimedia Tools and Applications.

[17]  Josep Ramon Morros,et al.  Fruit detection and 3D location using instance segmentation neural networks and structure-from-motion photogrammetry , 2020, Comput. Electron. Agric..

[18]  Kushtrim Bresilla,et al.  Single-Shot Convolution Neural Networks for Real-Time Fruit Detection Within the Tree , 2019, Front. Plant Sci..

[19]  Zhenyu Liu,et al.  Petri-net-based dynamic scheduling of flexible manufacturing system via deep reinforcement learning with graph convolutional network , 2020 .

[20]  Adel Hafiane,et al.  VddNet: Vine Disease Detection Network Based on Multispectral Images and Depth Map , 2020, Remote. Sens..

[21]  Chao Chen,et al.  Fruit Detection and Segmentation for Apple Harvesting Using Visual Sensor in Orchards , 2019, Sensors.

[22]  Indranil Ghosh,et al.  FEB-Stacking and FEB-DNN Models for Stock Trend Prediction: A Performance Analysis for Pre and Post Covid-19 Periods , 2021, Decision Making: Applications in Management and Engineering.

[23]  Paulo Flores,et al.  Technology progress in mechanical harvest of fresh market apples , 2020, Comput. Electron. Agric..

[24]  Volkan Isler,et al.  A comparative study of fruit detection and counting methods for yield mapping in apple orchards , 2018, J. Field Robotics.

[25]  Arnold W. Schumann,et al.  Detection of a Potato Disease (Early Blight) Using Artificial Intelligence , 2021, Remote. Sens..

[26]  Chao Chen,et al.  Fruit Detection, Segmentation and 3D Visualisation of Environments in Apple Orchards , 2019, Comput. Electron. Agric..

[27]  Gerassimos Peteinatos,et al.  Weed Identification in Maize, Sunflower, and Potatoes with the Aid of Convolutional Neural Networks , 2020, Remote. Sens..

[28]  Thorsten Hoeser,et al.  Object Detection and Image Segmentation with Deep Learning on Earth Observation Data: A Review-Part I: Evolution and Recent Trends , 2020, Remote. Sens..