1.官网下载框架https://github.com/bubbliiiing/yolov4-tiny-pytorch
文章图片
2.下载权重yolov4_tiny_voc.pth,https://pan.baidu.com/s/1W6hKHXyvKoq57qyxfYNDww,密码,gcb7,放在如图位置
文章图片
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
文章图片
5.视频流检测,运行video.py
文章图片
6.修改视频流的地方,可以放入本地视频地址和网络rtmp流地址
文章图片
7.运行成功效果
文章图片
8.优点:速度极快,缺点:识别效果稍低,综合来说,此框架很强,再会
推荐阅读
- 动手学深度学习(pytorch|动手学深度学习(四十一)——深度循环神经网络(Deep-RNN))
- 动手学深度学习(pytorch|动手学深度学习(四)——用numpy和torch实现单层神经网络对比)
- 深度学习|深度学习典型应用之超分重建
- transformer|Transformer模型结构详解
- 当人工智能遇上安全|[当人工智能遇上安全] 7.基于机器学习的安全数据集总结
- Python从零到壹|[Python从零到壹] 四十三.图像增强及运算篇之图像点运算和图像灰度化处理
- transformer|【总结】Transformer预训练任务总结
- GUI猜数字游戏,直接开玩...
- 工具|StreamLit构造Web应用