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