Lightweight pear detection algorithm based on improved YOLOv5

Xiaomei Hu; Yunyou Zhang; Yi Chen; Jianfei Chai; Jun Wu

doi:10.1117/12.2671817

28 February 2023 Lightweight pear detection algorithm based on improved YOLOv5

Xiaomei Hu, Yunyou Zhang, Yi Chen, Jianfei Chai, Jun Wu

Proceedings Volume 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022); 125961X (2023) https://doi.org/10.1117/12.2671817
Event: International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 2022, Changsha, China

Abstract

Pear recognition is one of the key technologies of pear picking robot, and the pear recognition algorithm based on convolutional neural network has high computing cost and large parameters, which is difficult to be deployed on pear picking robot with low computer resources. This paper presents a lightweight pear real-time detection method based on YOLOv5. This method designs a lightweight feature extraction network based on Ghost bottom-leneck, and embeds the SE module into the designed network, which improves the ability of feature extraction while reducing the amount of network parameters. The experimental results show that compared with YOLOv5l, the parameters of the improved lightweight model are reduced by 48.17 %, mAP is increased by 0.9 %, and the recognition speed is increased by 36 %. The improved model is more suitable to be deployed on the picking robot with limited computing power and provides a solution for the vision system of pear picking robot.

Citation Download Citation

Xiaomei Hu, Yunyou Zhang, Yi Chen, Jianfei Chai, and Jun Wu "Lightweight pear detection algorithm based on improved YOLOv5", Proc. SPIE 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 125961X (28 February 2023); https://doi.org/10.1117/12.2671817

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

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
5 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object detection

Detection and tracking algorithms

Convolution

Feature extraction

Target detection

Education and training

Neck

Show All Keywords

Keywords/Phrases

Search In:

Publication Years