Research on object detection algorithm in complex scene based on improved YOLOv5

Wanzhen Zhou; Yi Chen; Sicheng Dun; Huicong Wu

doi:10.1117/12.3011336

8 November 2023 Research on object detection algorithm in complex scene based on improved YOLOv5

Wanzhen Zhou, Yi Chen, Sicheng Dun, Huicong Wu

Proceedings Volume 12923, Third International Conference on Artificial Intelligence, Virtual Reality, and Visualization (AIVRV 2023); 129231B (2023) https://doi.org/10.1117/12.3011336
Event: 3rd International Conference on Artificial Intelligence, Virtual Reality and Visualization (AIVRV 2023), 2023, Chongqing, China

Abstract

The main task of object detection is the recognition and localisation of objects in images. It is the cornerstone of image understanding and the preparation for vision tasks such as image segmentation, target tracking and pose recognition. YOLOv5 is a object detection technique that is widely used in military, medical and traffic applications. However, in complex and changing real-world scenes, there are often multiple interfering factors, so the detection effect of YOLOv5 is greatly affected. In this paper, we propose an improved YOLOv5 algorithm for complex scenes from the perspectives of background information interference and inaccurate bounding box localisation. Firstly, the SimAM attention module is added to YOLOv5 to enable the model to focus on the differences in information in the feature maps and to enhance the anti-interference ability of the network; then the EIoU loss is used in the loss function and the Focal EIoU loss function is proposed by combining the optimization idea of Focal loss to deal with the category imbalance and improve the convergence speed of the algorithm; finally ,a comparison experiment is conducted on the PASCAL VOC dataset to demonstrate the reliability of the improved algorithm.

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

Citation Download Citation

Wanzhen Zhou, Yi Chen, Sicheng Dun, and Huicong Wu "Research on object detection algorithm in complex scene based on improved YOLOv5", Proc. SPIE 12923, Third International Conference on Artificial Intelligence, Virtual Reality, and Visualization (AIVRV 2023), 129231B (8 November 2023); https://doi.org/10.1117/12.3011336

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