【小白目標檢測】手把手教你做視頻中的實時目標檢測(基於Pelee算法)

手把手教你做視頻中的實時目標檢測(基於Pelee算法)

0. 先看效果:

在這裏插入圖片描述
在這裏插入圖片描述

1. 算法詳解:

可以看我這一篇:

【論文閱讀筆記】Pelee: A Real-Time Object Detection System on Mobile Devices

這裏簡單複述一下:

作者在 2016 年提出了一個基於 DenseNet 的叫做 PeleeNet 的輕量級圖像識別網絡,實現了比 MobileNet 更高的準確率,並在此基礎上提出了可以使用 PeleeNet 作爲目標檢測網絡(論文中基於 SSD)的 Backbone,來解決大量使用 DW Conv 造成的缺乏特徵有效利用的問題;

在這裏插入圖片描述

2. 下載源碼:

源碼地址:https://github.com/Sharpiless/Easy-Pelee-Detection/tree/master

首先打開網頁點一個⭐:

在這裏插入圖片描述
然後點擊下載:

在這裏插入圖片描述

下載後解壓:
在這裏插入圖片描述
然後下載訓練好的模型:

鏈接:https://pan.baidu.com/s/16ytCh6KRfeNdSRIEtqKTRA
提取碼:ys4q

放到路徑下面:

在這裏插入圖片描述

3. 運行檢測:

打開 detector.py 文件夾,將這裏改成你的視頻路徑:
在這裏插入圖片描述

運行即可~

在這裏插入圖片描述

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