基于yolov4-tiny-pytorch轻量级框架的目标检测

1.官网下载框架https://github.com/bubbliiiing/yolov4-tiny-pytorch
基于yolov4-tiny-pytorch轻量级框架的目标检测_第1张图片
2.下载权重yolov4_tiny_voc.pth,https://pan.baidu.com/s/1W6hKHXyvKoq57qyxfYNDww,密码,gcb7,放在如图位置
基于yolov4-tiny-pytorch轻量级框架的目标检测_第2张图片
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

4.图片检测,运行predict.py
基于yolov4-tiny-pytorch轻量级框架的目标检测_第3张图片
5.视频流检测,运行video.py
基于yolov4-tiny-pytorch轻量级框架的目标检测_第4张图片
6.修改视频流的地方,可以放入本地视频地址和网络rtmp流地址
基于yolov4-tiny-pytorch轻量级框架的目标检测_第5张图片
7.运行成功效果
基于yolov4-tiny-pytorch轻量级框架的目标检测_第6张图片
8.优点:速度极快,缺点:识别效果稍低,综合来说,此框架很强,再会

你可能感兴趣的:(人工智能算法,python,人工智能,深度学习)