話不多說,記錄一下
目標檢測:SSD 與 YOLO的實現,附訓練好的模型,python
(注:程序裏選用了GPU,若用CPU則選擇在選項中更改一下)
SSD:
paper:https://arxiv.org/pdf/1512.02325.pdf
code:https://download.csdn.net/download/u014426939/10697935
SSD的代碼稍微改了一下形式,採用opencv的圖片數據結構,本來需要在jupyter notebook上運行,這裏直接python運行即可,基於圖片或者視頻流的目標檢測
3分》
SSD檢測效果:對小物體檢測不好,但檢測定位精度粗略看比yolo高一點,格柵也對檢測存在一定的影響。
YOLO:
paper:https://arxiv.org/abs/1506.02640
code: https://pan.baidu.com/s/1DTv4onHAzc2KYJEfLO_hnA 密碼:8xk5
yolo檢測效果:對小物體檢測效果比ssd好一點,但實際上,這兩個網絡對大場景中的小物體檢測其實還是不太好。
檢測定位精度比ssd差一點。