Multi-scale Selection Pyramid Networks for Small-Sample Target Detection Algorithms

Target detection is to detect the specified target in the image. This technology has been widely used in automatic driving, face recognition and other fields, and has become a major research hotspot in the field of computer vision at home and abroad. Traditional target detection often requires a large number of annotated datasets, so it is a challenge to detect targets with only a small number of annotated samples. To address this problem, this paper proposes a multi-scale selection pyramid network algorithm for small sample target detection so that detection no longer relies on large-scale labeled datasets. Firstly, this paper designs a multi-scale selection pyramid network for small sample target detection, which consists of three components: context layer attention module, feature scale enhancement module, and feature scale selection module. Secondly, this paper performs feature fusion after the RoI features generated by the RPN network using maximum pooling and average pooling to improve the correlation between features. This paper uses feature subtraction to highlight the category information in the features, which can improve the sensitivity to new class parameters while maintaining the stability of the model to the sample parameters. Finally, the orthogonal mapping loss function is used to constrain the features before the classification layer, which can well measure the similarity between features even in the case of a small number of samples.