tensorflow 實現SSD+YOLO

話不多說,記錄一下

目標檢測: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差一點。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章