人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测

1.官网下载框架https://github.com/bubbliiiing/yolov4-tiny-pytorch
人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测
文章图片

2.下载权重yolov4_tiny_voc.pth,https://pan.baidu.com/s/1W6hKHXyvKoq57qyxfYNDww,密码,gcb7,放在如图位置
人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测
文章图片

3.需要的关键几个python库,numpy1.16.2,torch1.2,torchvision==0.4.0,下载代码如下:

pip install numpy==1.16.2 pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

【人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测】4.图片检测,运行predict.py
人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测
文章图片

5.视频流检测,运行video.py
人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测
文章图片

6.修改视频流的地方,可以放入本地视频地址和网络rtmp流地址
人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测
文章图片

7.运行成功效果
人工智能算法|基于yolov4-tiny-pytorch轻量级框架的目标检测
文章图片

8.优点:速度极快,缺点:识别效果稍低,综合来说,此框架很强,再会

    推荐阅读