An improved pear recognition and localization algorithm based on yolov5 model

Xiaomei Hu; Yi Chen; Jun Wu

doi:10.1117/12.2639309

15 July 2022 An improved pear recognition and localization algorithm based on yolov5 model

Xiaomei Hu, Yi Chen, Jun Wu

Proceedings Volume 12258, International Conference on Neural Networks, Information, and Communication Engineering (NNICE 2022); 1225819 (2022) https://doi.org/10.1117/12.2639309
Event: International Conference on Neural Networks, Information, and Communication Engineering (NNICE 2022), 2022, Qingdao, China

Abstract

Improving the visual recognition and positioning accuracy in the outdoor environment is an important way to improve the picking efficiency of fruit picking robots. With the rapid development of artificial intelligence, the convolutional neural network algorithm has gradually become an important research direction for machine recognition and localization. It can automatically extract target features, with high recognition accuracy, high speed and strong robustness. This paper takes pears as the research object, and proposes an improved pear recognition and localization algorithm based on the yolov5 model. The generalization ability of the model is improved by preprocessing and data enhancement of the data set, and an improved k-means clustering algorithm is proposed to realize the optimal calculation of the initial anchor frame. Compared with the original yolov5 model, the fitness and best recall rate of the improved algorithm in recognizing pears are increased by 6% and 9%, respectively.

Citation Download Citation

Xiaomei Hu, Yi Chen, and Jun Wu "An improved pear recognition and localization algorithm based on yolov5 model", Proc. SPIE 12258, International Conference on Neural Networks, Information, and Communication Engineering (NNICE 2022), 1225819 (15 July 2022); https://doi.org/10.1117/12.2639309

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available