深度學習之目標檢測（四）—— SSD算法

SSD算法

SSD, Single Shot MultiBox Detector(one-stage)，Wei Liu在ECCV 2016上提出的一種目標檢測算法

算法特點

• 直接回歸目標類別和位置
• 不同尺度的特徵圖上進行預測
• 端到端的訓練
• 圖像分別率比較低，也能保證檢測的精度

 

 

SSD系列應用場景

 

SSD網絡結構

主幹網絡：VGGNet

最後通過NMS算法對檢測結果進行合併和篩選

 

主幹網絡介紹

原作者採用的是vgg，但我們也可以採用resnet、mobilenet等網絡結構

FC層即全連接層

 

多尺度FeatureMap預測

 

Prior Box Layer

得到6個不同的長寬比

 

 

樣本構造

 

數據增強

 

損失函數

 

SSD算法性能

 

SSD算法性能增強的原因