| Popis: |
In order to further improve the accuracy and real-time performance of the traditional Single Shot Multibox Detector (SSD) object detection model, an improved SSD multi-object detection model is proposed. Firstly, aiming at the defect of weak correlation between prediction object score and positioning accuracy in the traditional SSD model, the improved model enhanced the correlation between the two by adding Intersection Over Union(IoU) prediction loss branch. Secondly, in order to reduce the spatial redundancy of traditional SSD model, a multi-frequency feature component convolution module is designed, which greatly reduces the calculation overhead and hardware overhead of the traditional model. Finally, in order to accelerate the convergence speed of the improved model, the Adaptive and Momental Bound (AdaMod) optimizer is introduced to modify the adaptive learning rate of the improved model which is too large in the training process. Experimental results show that the improved model has stronger detection capabilities, better overall detection results, and improved detection accuracy and real-time detection. |
