可以看我这一篇:
【论文阅读笔记】Pelee: A Real-Time Object Detection System on Mobile Devices
这里简单复述一下:
作者在 2016 年提出了一个基于 DenseNet 的叫做 PeleeNet 的轻量级图像识别网络,实现了比 MobileNet 更高的准确率,并在此基础上提出了可以使用 PeleeNet 作为目标检测网络(论文中基于 SSD)的 Backbone,来解决大量使用 DW Conv 造成的缺乏特征有效利用的问题;
源码地址:https://github.com/Sharpiless/Easy-Pelee-Detection/tree/master
首先打开网页点一个⭐:
然后点击下载:
下载后解压:
然后下载训练好的模型:
链接:https://pan.baidu.com/s/16ytCh6KRfeNdSRIEtqKTRA
提取码:ys4q
放到路径下面:
打开 detector.py 文件夹,将这里改成你的视频路径:
运行即可~