python + tensorflow + yolov4视频物体识别demo

之前做的一个小项目，整理了一下源码，有需要的朋友可以参考

5.1更新

更新了视频演示： bilibili

gitee仓库

YOLOv4_on-video

功能介绍：

• 支持YOLOv4和YOLOv4-tiny模型
• 支持视频输入以及摄像头输入
• 支持视频保存

使用方法：

• 首先下载模型文件，放入根目录中，即./checkpoints

  链接: https://pan.baidu.com/s/1kLJsJ590lVxA8CP9FsJuCw 密码: 9od3

• 直接运行YOLOv4-tiny模型：

python detect_video.py

• 更改输入视频文件

python detect_video.py --video ./data/video/video.mp4

• 使用摄像头作为输入

ls /dev/video*
python detect_video.py --video 0

• 更改输出的结果视频路径

python detect_video.py --output out_video.mp4
python detect_video.py --output None

• 运行YOLOv4模型：

python detect_video.py --weights ./checkpoints/yolov4-416 --tiny False

样例