Back to Search Start Over

An improved YOLOv5 for object detection in visible and thermal infrared images based on contrastive learning

Authors :
Xiaoguang Tu
Zihao Yuan
Bokai Liu
Jianhua Liu
Yan Hu
Houqiang Hua
Lin Wei
Source :
Frontiers in Physics, Vol 11 (2023)
Publication Year :
2023
Publisher :
Frontiers Media S.A., 2023.

Abstract

An improved algorithm has been proposed to address the challenges encountered in object detection using visible and thermal infrared images. These challenges include the diversity of object detection perspectives, deformation of the object, occlusion, illumination, and detection of small objects. The proposed algorithm introduces the concept of contrastive learning into the YOLOv5 object detection network. To extract image features for contrastive loss calculation, object and background image regions are randomly cropped from image samples. The contrastive loss is then integrated into the YOLOv5 network, and the combined loss function of both object detection and contrastive learning is used to optimize the network parameters. By utilizing the strategy of contrastive learning, the distinction between the background and the object in the feature space is improved, leading to enhanced object detection performance of the YOLOv5 network. The proposed algorithm has shown pleasing detection results in both visible and thermal infrared images.

Details

Language :
English
ISSN :
2296424X
Volume :
11
Database :
Directory of Open Access Journals
Journal :
Frontiers in Physics
Publication Type :
Academic Journal
Accession number :
edsdoj.0d07171bd274463dadce069ecab7faf1
Document Type :
article
Full Text :
https://doi.org/10.3389/fphy.2023.1193245