@inproceedings{36bbe0534627413a9b072b03af0ccf17,
title = "Object detection in infrared images using modified YOLOv4 models and an image enhancement module",
abstract = "Deep learning-based object detection approaches have shown excellent performance in RGB images. However, when used to detect objects from infrared images, the accuracy may reduce significantly due to low contrast, obscure textures and strong noise of infrared images. To alleviate the problem, we design a detail enhancement module involving spatial attention mechanism to enhance the textures and details of images. The output of the proposed module is fed into modified YOLOv4. We introduce Alpha-IoU loss and Weighted-NMS to YOLOv4 to enhance geometric factors in both bounding box regression and Non-Maximum Suppression, leading to notable gains of average precision. The experiment results show that compared with YOLOv4, mAP0.5 and mAP0.5:0.95 of our model are improved by 1.1% and 3.5% respectively, effectively improving the detection accuracy.",
keywords = "attention, infrared images, object detection",
author = "Dan Wang and Huiqian Du and Zhifeng Ma",
note = "Publisher Copyright: {\textcopyright} COPYRIGHT SPIE. Downloading of the abstract is permitted for personal use only.; 14th International Conference on Graphics and Image Processing, ICGIP 2022 ; Conference date: 21-10-2022 Through 23-10-2022",
year = "2023",
doi = "10.1117/12.2680173",
language = "English",
series = "Proceedings of SPIE - The International Society for Optical Engineering",
publisher = "SPIE",
editor = "Liang Xiao and Jianru Xue",
booktitle = "Fourteenth International Conference on Graphics and Image Processing, ICGIP 2022",
address = "United States",
}