Small Object Detection Algorithm Based on Weighted Network

Abstract

For the observation of a picture, people may instinctly pay more attention to the eye-catching objects in the picture. Usually such objects tend to occupy a larger proportion in the picture, which leads to small targets being ignored. Because the area where the small target is located is often a weak detection area, and the features that can be extracted in the process of extracting features by the detector are few and are easily lost in the process of feature information transmission after the feature is extracted, the effect of small target detection is not good. Therefore, on the basis of the single-order detector, this paper adds a cross-channel interaction mechanism to ensure the integrity of the information between layers, adopts target enhancement of training samples and designs a general loss function. Apart from this, this paper improves the sample weighting on the basis of the loss function to predict weight of samples. The mAP of this paper framework UWN (unified weighted network) on the VOC public dataset is 81.2% and the mAP on the self-made small target aerial photography dataset is 82.3%. Compared with the FSSD algorithm, some speed is sacrificed, and the accuracy is greatly improved.

Keywords

• |small target detection|target enhancement|cross-channel interaction|weighted network