Object recognition based on improved YOLOv5

Hangong Chen; Weimin Qi

doi:10.1117/12.2671298

16 March 2023 Object recognition based on improved YOLOv5

Hangong Chen, Weimin Qi

Proceedings Volume 12593, Second Guangdong-Hong Kong-Macao Greater Bay Area Artificial Intelligence and Big Data Forum (AIBDF 2022); 125930O (2023) https://doi.org/10.1117/12.2671298
Event: 2nd Guangdong-Hong Kong-Macao Greater Bay Area Artificial Intelligence and Big Data Forum (AIBDF 2022), 2022, Guangzhou, China

Abstract

At presen, object recognition task is troubled by its huge kinds of objects. In this paper, the SIoU loss function and YOLOv5 deep learning convolutional neural network are innovatively used to improve the training efficiency and recognition accuracy. Unlike the traditional bounding box regression loss function (e.g. Giou, Diou[1] , CIoU) , which only focuses on the distance between the prediction box and the ground true box, the size of the overlap area, and one or more of the aspect ratios, and sets the impact factor on this basis, the SIoU loss function also introduces Angle cost to fit the best regression direction, which makes the direction of bounding box regression more reasonable and improves the regression efficiency[1].In this paper, the defects of traditional loss function and the calculation method of SIoU loss function are introduced, and the performance between SIoU and CIoU is compared.

Citation Download Citation

Hangong Chen and Weimin Qi "Object recognition based on improved YOLOv5", Proc. SPIE 12593, Second Guangdong-Hong Kong-Macao Greater Bay Area Artificial Intelligence and Big Data Forum (AIBDF 2022), 125930O (16 March 2023); https://doi.org/10.1117/12.2671298

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