Abstract:Target detection is the basis for automatic target recognition system of infrared imaging guidance to complete subsequent tasks such as recognition and tracking. Existing systems have not the autonomous learning ability of target feature, and it will be powerless once the task environment exceeds the pre-planned condition. The single-stage target detection based on deep learning has the ability of autonomous learning and high computational efficiency, which is an effective way to solve the problem of infrared imaging guidance target detection in complex environment. SSD (Single Shot MultiBox Detector) is a classical single-stage detection model, however, the convolution layer with strong semantic information in SSD has low resolution, which is not conducive to small target detection. In addition, the location loss of SSD does not consider the impact of target scale change. Therefore, this paper puts forward two improvement ideas in view of SSD: (1) Starting from FPN (Feature Pyramid Network), feature channel’s importance is distinguished through efficient channel attention mechanism, the contribution of each feature layer to the fusion output is described based on the learnable weight, and the feature weighted fusion of bidirectional multi-scale is realized between the feature layer which has low resolution and strong semantics and the feature layer which has high resolution and weak semantics. (2) Starting from IoU (Intersection over Union) and considering non overlapping parts and geometric relationship between the predicted box and the ground-truth box, the location loss of SSD that remains invariable to the target scale change is constructed to improve the sensitivity of the detection model to the locating error of small target. The experimental results show that, for 300 × 300 input, the presented method achieves 84.7% mAP (mean Average Precision) on VOC2007 test and for 512 × 512 input, it reaches 86.6%. On the self-built infrared aircraft data set, the proposed method achieves 81.1% mAP and can detect more small targets. Without affecting detection speed, the presented method on experimental results outperforms some comparable state-of-the-art models such as YOLOv3 (You Only Look Once), DSSD (Deconvolutional Single Shot Multibox Detector), RSSD (Rainbow Single Shot Multibox Detector) and FSSD (Fusion Single Shot Multibox Detector).

Object Detection Based on the Improved Single Shot MultiBox Detector

Object Detection Using Deep Learning: Single Shot Detector with a Refined Feature-fusion Structure

Improved single shot multibox detector target detection method based on deep feature fusion

Small Object Detection Algorithm Based on Feature Pyramid-Enhanced Fusion SSD

Feature-fused SSD: fast detection for small objects

Detecting Small Objects in Thermal Images Using Single-Shot Detector

A Balanced Feature Fusion SSD for Object Detection

Improvements Based on Feature Fusion Single Shot Multibox Detector

Object Detection Based on Faster R-CNN Algorithm with Skip Pooling and Fusion of Contextual Information

TDFSSD: Top-Down Feature Fusion Single Shot MultiBox Detector

Attention Based Single Shot Multibox Detector

An efficient single shot detector with weight-based feature fusion for small object detection

SSD Optimization Model Based on Shallow Feature Fusion

FSSD: Feature Fusion Single Shot Multibox Detector

Multi-scales Feature Integration Single Shot Multi-Box Detector on Small Object Detection

MDSSD: Multi-scale Deconvolutional Single Shot Detector for Small Objects

Single-Shot Object Detection via Feature Enhancement and Channel Attention

A Small Object Detection Algorithm Based on Improved Faster RCNN

SSD: Single Shot MultiBox Detector

Small-objectness Sensitive Detection Based on Shifted Single Shot Detector

An improved SSD method for infrared target detection based on convolutional neural network