[英]How to process images of a video, frame by frame, in video streaming using OpenCV and Python
我是 OpenCV 的初学者。 我想对上传到我的服务器的视频的帧进行一些图像处理。 我只想读取可用的帧并将它们写入目录中。 然后,等待视频的另一部分上传并将帧写入目录。 并且,我应该等待每一帧完全上传然后将其写入文件。
你能告诉我如何用 OpenCV (Python) 做到这一点吗?
编辑 1:我编写了这段代码,用于从文件中捕获视频,同时将新数据附加到文件末尾。 换句话说, out.mp4
文件不是一个完整的视频,另一个程序正在其上写入新的帧。 我要做的是,等待其他程序写入新的帧,然后读取它们并显示它们。
这是我的代码:
import cv2
cap = cv2.VideoCapture("./out.mp4")
while True:
if cap.grab():
flag, frame = cap.retrieve()
if not flag:
continue
else:
cv2.imshow('video', frame)
if cv2.waitKey(10) == 27:
break
所以问题是cap.grab()
调用! 当没有框架时,它将返回False
! 即使我等了很长时间,它也不会再捕获帧了。
阅读VideoCapture
的文档后。 我发现您可以告诉VideoCapture
,下次我们调用VideoCapture.read()
(或VideoCapture.grab()
)时要处理哪个帧。
问题在于,当您想要read()
未准备好的帧时, VideoCapture
对象会卡在该帧上并且永远不会继续。 所以你必须强制它从前一帧重新开始。
这是代码
import cv2
cap = cv2.VideoCapture("./out.mp4")
while not cap.isOpened():
cap = cv2.VideoCapture("./out.mp4")
cv2.waitKey(1000)
print "Wait for the header"
pos_frame = cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES)
while True:
flag, frame = cap.read()
if flag:
# The frame is ready and already captured
cv2.imshow('video', frame)
pos_frame = cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES)
print str(pos_frame)+" frames"
else:
# The next frame is not ready, so we try to read it again
cap.set(cv2.cv.CV_CAP_PROP_POS_FRAMES, pos_frame-1)
print "frame is not ready"
# It is better to wait for a while for the next frame to be ready
cv2.waitKey(1000)
if cv2.waitKey(10) == 27:
break
if cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES) == cap.get(cv2.cv.CV_CAP_PROP_FRAME_COUNT):
# If the number of captured frames is equal to the total number of frames,
# we stop
break
用这个:
import cv2
cap = cv2.VideoCapture('path to video file')
count = 0
while cap.isOpened():
ret,frame = cap.read()
cv2.imshow('window-name', frame)
cv2.imwrite("frame%d.jpg" % count, frame)
count = count + 1
if cv2.waitKey(10) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows() # destroy all opened windows
根据 OpenCV 3.0 及更高版本的最新更新,您需要在 Mehran 的代码中更改属性标识符,如下所示:
cv2.cv.CV_CAP_PROP_POS_FRAMES
至
cv2.CAP_PROP_POS_FRAMES
同样适用于cv2.CAP_PROP_POS_FRAME_COUNT
。
希望能帮助到你。
在 openCV 的文档中有一个逐帧获取视频的示例。 它是用 C++ 编写的,但很容易将示例移植到 python - 您可以搜索每个功能文档以查看如何在 python 中调用它们。
#include "opencv2/opencv.hpp"
using namespace cv;
int main(int, char**)
{
VideoCapture cap(0); // open the default camera
if(!cap.isOpened()) // check if we succeeded
return -1;
Mat edges;
namedWindow("edges",1);
for(;;)
{
Mat frame;
cap >> frame; // get a new frame from camera
cvtColor(frame, edges, CV_BGR2GRAY);
GaussianBlur(edges, edges, Size(7,7), 1.5, 1.5);
Canny(edges, edges, 0, 30, 3);
imshow("edges", edges);
if(waitKey(30) >= 0) break;
}
// the camera will be deinitialized automatically in VideoCapture destructor
return 0;
}
我找到的唯一解决方案不是将索引设置为前一帧并等待(然后 OpenCV 停止读取帧,无论如何),而是再次初始化捕获。 所以,它看起来像这样:
cap = cv2.VideoCapture(camera_url)
while True:
ret, frame = cap.read()
if not ret:
cap = cv.VideoCapture(camera_url)
continue
# do your processing here
它完美无缺!
这是一个示例,其中 a.mp4 是视频文件
import cv2
cap = cv2.VideoCapture('a.mp4')
count = 0
while cap.isOpened():
ret,frame = cap.read()
if not ret:
continue
cv2.imshow('window-name', frame)
#cv2.imwrite("frame%d.jpg" % count, frame)
count = count + 1
if cv2.waitKey(10) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.