YOLO应用化之保存视频

最新推荐文章于 2025-06-13 09:50:54 发布

hrsstudy

最新推荐文章于 2025-06-13 09:50:54 发布

阅读量1.1w

点赞数

CC 4.0 BY-SA版权

分类专栏：机器学习文章标签： YOLO

本文链接：https://blog.youkuaiyun.com/hrsstudy/article/details/60876451

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

使用命令

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

可以打开camera进行实时检测MSCOCO的８０种物体。（附加参数-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/videoio/videoio_c.h"
 image get_image_from_stream(CvCapture *cap);

+#ifdef SAVEVIDEO
+stati