Research on infrared small target segmentation algorithm based on improved mask R-CNN

Abstract

The task of segmenting small infrared targets, which have few pixels and weak features, has been a difficult problem in the field of small target image processing. Small targets not only exist in general images, but also widely exist in drone cameras, communication base station cameras, rescue cameras, and vehicle cameras. Research on small target segmentation algorithm is very important for analyzing and utilizing these images, and it also has important application value in security, traffic, and rescue. Traditional small target segmentation algorithms are able to segment objects with simple target contour edges and large differences in signal strength. The traditional algorithm often has a high false detection rate and missed detection rate when facing several targets with weak signal strength. It does not perform well in complex scenes. In this paper, we introduce an infrared small target segmentation scheme facing multiple types and numbers of targets. This paper also produces an infrared UAV and pedestrian dataset for validation. Experiments show that the optimized algorithm has a better detection effect on targets in complex scenes.