Improved YOLOv5s recognition of cotton top buds with fusion of attention and feature weighting

Lei Yin; Jian Wu; Qikong Liu; Wenxiong Wu

doi:10.1117/12.3014605

9 January 2024 Improved YOLOv5s recognition of cotton top buds with fusion of attention and feature weighting

Lei Yin, Jian Wu, Qikong Liu, Wenxiong Wu

Proceedings Volume 12969, International Conference on Algorithm, Imaging Processing, and Machine Vision (AIPMV 2023); 1296928 (2024) https://doi.org/10.1117/12.3014605
Event: International Conference on Algorithm, Imaging Processing and Machine Vision (AIPMV 2023), 2023, Qingdao, China

Abstract

In order to improve the accuracy and real-time performance of cotton top bud recognition, an improved YOLOv5s target real-time detection model is proposed. First, the SE module and the CBAM module in the attention mechanism are added to optimize the weight ratio of channel attention and spatial attention to improve accuracy; then the BiFPN structure of bidirectional weighted features is introduced to strengthen the fusion between high-level features and low-level features; finally, a new bounding box regression loss function EIoU is used for ablation experiments, and more position information of cotton buds can be obtained by reducing the bounding box loss. The experimental results show that, by applying the improved algorithm in the identification of cotton top buds, compared with the original YOLOv5s model, the accuracy of the C3SE-4l+BiFPN+EIoU model has increased by 7.9%, the recall rate has increased by 2.8%, and the average precision an increase of 5.7%. These improvements and optimizations provide a new idea and method, which can provide a more efficient solution for the identification of cotton top buds.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Lei Yin, Jian Wu, Qikong Liu, and Wenxiong Wu "Improved YOLOv5s recognition of cotton top buds with fusion of attention and feature weighting", Proc. SPIE 12969, International Conference on Algorithm, Imaging Processing, and Machine Vision (AIPMV 2023), 1296928 (9 January 2024); https://doi.org/10.1117/12.3014605

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