Paper
9 August 2023 Infrared small target recognition in waterways based on YOLOv5 algorithm
Yikai Fan, Yingjun Zhang
Author Affiliations +
Proceedings Volume 12782, Third International Conference on Image Processing and Intelligent Control (IPIC 2023); 127820R (2023) https://doi.org/10.1117/12.3002081
Event: Third International Conference on Image Processing and Intelligent Control (IPIC 2023), 2023, Kuala Lumpur, Malaysia
Abstract
YOLOv5 is one of the target detection algorithms with fast detection speed and high accuracy, but it has the problems of insufficient sensory field and low accuracy of small target detection. In order to solve above problems, an improved YOLOv5 network model, i.e., an improved YOLOv5-TI model based on the attention mechanism, is proposed. The attention module is added to the backbone network when extracting features to improve the target detection accuracy, and the input features are shifted windowed for self-attention calculation to effectively utilize the features and improve the small target detection accuracy; the proposed model YOLOv5-TI is experimented on the self-built inland infrared dataset, and the mAP value reaches 95.5%, and the results show that YOLOv5-TI can effectively improve the target detection accuracy. The inland vessels equipped with visual intelligent perception system can effectively identify the targets on water, and they have wide applications in the fields of surface exploration and autonomous search and rescue.
(2023) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Yikai Fan and Yingjun Zhang "Infrared small target recognition in waterways based on YOLOv5 algorithm", Proc. SPIE 12782, Third International Conference on Image Processing and Intelligent Control (IPIC 2023), 127820R (9 August 2023); https://doi.org/10.1117/12.3002081
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Detection and tracking algorithms

Target detection

Convolution

Small targets

Infrared detectors

Visual process modeling

Back to Top