Improvement of real time detection algorithm based on SSD

摘要

Since the convolution neural network models emerged, convolution neural network model is more and more large, which brings the improvement of model effect, but its heavy computational load and huge memory make the model difficult to deploy in the embedded system. In this paper, an improved scheme is proposed based on the Single Shot Detector (SSD) network model. A network with a small amount of parameters, which is named Wide Residual Network (WRN), takes place of the original feature extraction network. What's more, input size of the network is reduced to reduce the computational load. In order to compensate for the loss of accuracy caused by reducing the input size of the network and solve the problem of mismatch between positive and negative samples in training samples, Focal Loss' loss function is adopted in training objectives, which makes the model training more focused on difficult samples. Experiments show that the model achieve mAP 0.781 on VOC0712. At the same time, it reached 89FPS on the GPU K80.