天天看點

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高一點,格栅也對檢測存在一定的影響。

tensorflow 實作SSD+YOLO

YOLO:

paper:https://arxiv.org/abs/1506.02640

code:   https://pan.baidu.com/s/1DTv4onHAzc2KYJEfLO_hnA 密碼:8xk5

yolo檢測效果:對小物體檢測效果比ssd好一點,但實際上,這兩個網絡對大場景中的小物體檢測其實還是不太好。

檢測定位精度比ssd差一點。

tensorflow 實作SSD+YOLO

繼續閱讀