Small aircraft detection in infrared aerial imagery based on deep neural network

Abstract

Detection of aerial target is an important part of infrared image processing. Both neural network method and traditional method can be used in infrared object detection. Neural network method has many advantages such as high accuracy and good portability compared with traditional object detection method. Since the features extracted by neural network method can change over detection target, automatic feature extraction makes neural network based detection method more effective. In recent years deep learning method has been also found wide use for object detection in images. In this paper, an object detection model based on the deep learning network YOLO is constructed for solving the infrared aircraft detection problem. We construct the dataset used for training and testing with recognized features being iteratively learned. The task of infrared object detection is sensitive to model size and detection speed. There is a requirement of using quantization method to reduce the storage space and the computation complexity. We propose a quantized model with appropriate accuracy for infrared object detection task. To solve the detection task for multiple extremely small aircrafts, model adjustment and quantization are used in proposed model and it gets a better performance. Experimental results on the constructed dataset show that the storage space for weight after quantization shrinks to a quarter, and there is no precision loss for extremely small aircrafts compared to the original model. The optimized YOLO-based deep learning model is effective to detect the small aircraft target in infrared aerial imagery.