Indoor object recognition based on YOLOv5 with EIOU loss function

Weimin Qi; Hangong Chen; Yuting Ye; Guoshuai Yu

doi:10.1117/12.3005836

10 October 2023 Indoor object recognition based on YOLOv5 with EIOU loss function

Weimin Qi, Hangong Chen, Yuting Ye, Guoshuai Yu

Proceedings Volume 12799, Third International Conference on Advanced Algorithms and Signal Image Processing (AASIP 2023); 127993M (2023) https://doi.org/10.1117/12.3005836
Event: 3rd International Conference on Advanced Algorithms and Signal Image Processing (AASIP 2023), 2023, Kuala Lumpur, Malaysia

Abstract

At present, large-scale object classification and recognition tasks inevitably encounter problems of training efficiency and model accuracy. The solution of this problem highly depends on the definition of loss function. A better loss function can get a more accurate model with fewer epoches, which undoubtedly helps to save computility and time cost and improve the security of the model. Nowadays, the task of indoor object recognition plays an important role in the fields of computer vision aided blind people, home robots and so on. However, the task of indoor object recognition is plagued by the above problems due to its huge category of recognition objects. Starting with the task of indoor object recognition, this paper innovatively uses EIOU loss function and YOLOv5 deep learning convolutional neural network in this field, which improves training efficiency and recognition accuracy. Some original bounding box regression loss functions of YOLO series (such as GIOU, DIOU, CIOU) have defects in that the prediction box coincides with the center point of the truth value box, and the horizontal vertical ratio is the same. The paper uses the EIOU loss function to solve this problem. This paper focuses on the network structure of YOLOv5, the defects of YOLOv5's native loss function, the calculation method and advantages of EIOU loss function, and compares the performance gap between EIOU and CIOU. Finally, the EIOU loss function is used to complete the indoor article identification task.

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

Citation Download Citation

Weimin Qi, Hangong Chen, Yuting Ye, and Guoshuai Yu "Indoor object recognition based on YOLOv5 with EIOU loss function", Proc. SPIE 12799, Third International Conference on Advanced Algorithms and Signal Image Processing (AASIP 2023), 127993M (10 October 2023); https://doi.org/10.1117/12.3005836

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
6 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object recognition

Computer vision technology

Object detection

Detection and tracking algorithms

Education and training

Distributed interactive simulations

Pattern recognition

Show All Keywords

Keywords/Phrases

Search In:

Publication Years