Lightweight vegetable detection algorithm based on improved YOLOv5

Song Guo; QingYao Zhang

doi:10.1117/12.2660006

3 February 2023 Lightweight vegetable detection algorithm based on improved YOLOv5

Song Guo, QingYao Zhang

Proceedings Volume 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022); 125112G (2023) https://doi.org/10.1117/12.2660006
Event: Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 2022, Hulun Buir, China

Abstract

With the rapid development of deep learning, vegetable detection based on computer vision in smart supermarkets can save a lot of manpower. To solve the problem that the reported target detection algorithm fails to balance detection accuracy and model size well, a lightweight vegetable detection method that is based on yolov5 model was proposed in this work. The method firstly improved the detection accuracy by adding a fusion attention mechanism (Convolutional Block Attention Module) in the backbone part, and then reduces the model parameters by replacing the normal convolution in the network with Ghost convolution to ensure the accuracy at the same time. The accuracy of localization was improved by using Alpha-IoU as the loss function of the bounding box regression. The experimental results show that the improved method achieves an average accuracy of 92.5%, which is 3.2% higher than yolov5, and the total model parameters are 6.1 MB, which is 1.0 MB lower than yolov5, meeting the requirement of balancing detection accuracy and model size.

Citation Download Citation

Song Guo and QingYao Zhang "Lightweight vegetable detection algorithm based on improved YOLOv5", Proc. SPIE 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 125112G (3 February 2023); https://doi.org/10.1117/12.2660006

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