【小白目标检测】手把手教你做视频中的实时目标检测(基于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 文件夹,将这里改成你的视频路径:
在这里插入图片描述

运行即可~

在这里插入图片描述