Infared ship recognition has many applications in port supervision and management. However, when the imaging distance is long or the target changes are obvious, it is difficult to achieve accurate detection and recognition by traditional methods. In this paper, we designed a single step cascade neural network that consists of three parts: feature extraction module, scale transform module and classification regression module. Firstly, the VGG network is used to extract the different level features of the target images. Then the scale transform module is used to fuse the high-level features and the low-level features to reflect the semantic information and shallow information of the targets more completely. The generated region of interest is input to classification regression module that predicts the targets location and classes. The main contribution of this paper is to combine the specific problems of infrared polymorphic ships detection and recognition. The clustering algorithm is used to generate the appropriate anchors to adapt our targets, and the attention mechanism is introduced into the model training process. Compared with the traditional detection and recognition methods, the proposed single step cascade neural network achieves the better average precision in polymorphic ships.
|