YOLO应用化之保存视频

YOLO原生代码不支持保存视频,但可以通过修改代码将检测结果整合成视频。介绍如何利用YOLO检测视频或相机中的物体,并提供了一种方法将检测的每一帧保存为图片。已分享相关代码更改于Github,涉及`demo.c`和`image.c`两个文件。
摘要由CSDN通过智能技术生成

YOLO的原生代码没有保存视频的功能,但我们往往需要一个视频文件作为检测结果。

使用命令

./darknet detector demo cfg/coco.data cfg/yolo.cfg yolo.weights

可以打开camera进行实时检测MSCOCO的80种物体。(附加参数-c camera_index 可以选择camera)

如果没有camera也可以使用命令

./darknet detector demo cfg/coco.data cfg/yolo.cfg yolo.weights input.avi

检测视频文件中的物体。

通过附加参数-prefix 可以将每一帧输出保存为jpg图片。

我们需要修改下代码将检测结果以视频文件形式保存。
已将代码提交到Github,先放链接
https://github.com/PaulChongPeng/darknet/commit/0366ba1fc0f8a640a11d300a3ef16382fbc530ef

共需要修改两个文件

demo.c

@@ -10,13 +10,18 @@
 #include <sys/time.h>

 #define FRAMES 3
+#define SAVEVIDEO

 #ifdef OPENCV
 #include "opencv2/highgui/highgui_c.h"
 #include "opencv2/imgproc/imgproc_c.h"
 #include "opencv2/vi
评论 20
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值