KSSD: single-stage multi-object detection algorithm with higher accuracy

What is it about?

Considering that the single shot multibox detector (SSD) algorithm will be missed or even false when is used to detect the small- and medium-sized objects, in this study, Kullback–Leibler single shot multibox detection (KSSD) object detection algorithm is proposed to improve the accuracy of small- and medium-sized objects detection. Firstly, the details in the detection process are visualised with gradient-weighted class activation mapping technology, and the details of each detection layer are shown in the form of class activation maps. Then it is noted that the phenomenon of the false or missed detection of the objects to be detected on small- and medium-sized objects in the SSD algorithm is related to the regression loss function. Accordingly, Kullback–Leibler border regression loss strategy is adopted and non-maximum suppression algorithm is used to output the final prediction boxes. Experimental results show that compared with the existed detection algorithms, the improved algorithm in this study has higher accuracy and stability, and can significantly improve the detection effect on small- and medium-sized objects.

Featured Image

Photo by Hitesh Choudhary on Unsplash

Photo by Hitesh Choudhary on Unsplash

Why is it important?

(i) The SSD object detection algorithm is visualised to analyse the detection process of this algorithm. (ii) Aiming at the shortcomings of the SSD algorithm, the augmentation strategy of the input image is improved. (iii) Based on the analysis of the visualisation results, the bounding box regression loss of the SSD object detection algorithm is optimised. (iv) The improved NMS algorithm is used to output the prediction bounding box of the object.

Perspectives