在python中保存openCV视频
- 2025-04-17 09:02:00
- admin 原创
- 14
问题描述:
我尝试保存视频,但失败了。我按照 OpenCV 文档中的说明操作了。
import numpy as np
import cv2
cap = cv2.VideoCapture(0)
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi', fourcc, 20.0, (640,480))
while(cap.isOpened()):
ret, frame = cap.read()
if ret==True:
frame = cv2.flip(frame,0)
out.write(frame)
cv2.imshow('frame',frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
else:
break
cap.release()
out.release()
cv2.destroyAllWindows()
怎么了?
解决方案 1:
试试这个。我使用的是 Windows 10,效果不错。
import numpy as np
import cv2
cap = cv2.VideoCapture(0)
# Define the codec and create VideoWriter object
#fourcc = cv2.cv.CV_FOURCC(*'DIVX')
#out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))
out = cv2.VideoWriter('output.avi', -1, 20.0, (640,480))
while(cap.isOpened()):
ret, frame = cap.read()
if ret==True:
frame = cv2.flip(frame,0)
# write the flipped frame
out.write(frame)
cv2.imshow('frame',frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
else:
break
# Release everything if job is finished
cap.release()
out.release()
cv2.destroyAllWindows()
解决方案 2:
jveitchmichaelis 在https://github.com/ContinuumIO/anaconda-issues/issues/223上给出了详尽的答案。我在这里复制了他的答案:
OpenCV 的文档(隐藏部分)指出,只能使用 OpenCV3 写入 avi 文件。我无法确定这是否属实,但我一直无法写入其他文件。
然而,OpenCV 主要是一个计算机视觉库,而不是一个视频流、编解码器和编写库。因此,开发人员试图使这部分尽可能简单。因此,OpenCV 的视频容器仅支持 avi 扩展,这是其第一个版本。
出处:http://docs.opencv.org/3.1.0/d7/d9e/tutorial_video_write.html
我的设置:我使用 MSVC 2015 从源代码构建了 OpenCV 3,包含 ffmpeg。我还从 Cisco 下载并安装了 XVID 和 openh264,并将其添加到我的 PATH 环境变量中。我正在运行 Anaconda Python 3。我还下载了 ffmpeg 的最新版本,并将 bin 文件夹添加到我的 PATH 环境变量中,不过这应该没什么影响,因为它已经内置在 OpenCV 中了。
我在 Win 10 64 位中运行。
这段代码在我的电脑上似乎运行良好。它会生成一段包含随机静态内容的视频:
writer = cv2.VideoWriter("output.avi", cv2.VideoWriter_fourcc(*"MJPG"), 30,(640,480)) for frame in range(1000): writer.write(np.random.randint(0, 255, (480,640,3)).astype('uint8')) writer.release()
我通过反复尝试学到了一些东西:
仅使用“.avi”,它只是一个容器,编解码器才是最重要的。
指定帧大小时要小心。在构造函数中,你需要以 (列,行) 的形式传递帧大小,例如 640x480。但是,你传入的数组的索引是 (行,列) 的。看看上面的例子,它是如何切换的?
如果您输入的图像与 VideoWriter 的大小不同,它将会失败(通常是无声的)
仅传递 8 位图像,如果需要,请手动转换数组(.astype('uint8'))
其实没关系,只要一直转换就行。即使你用 cv2.imread 加载图像,也需要转换为 uint8 类型……
如果您不传入 3 通道、8 位图像,MJPG 将会失败。至少我遇到过这种情况,断言失败。
XVID 也需要 3 通道图像,但如果不这样做,则会悄无声息地失败。
H264 似乎可以处理单通道图像
如果需要原始输出,例如来自机器视觉相机的原始输出,可以使用“DIB”。“RAW”或空编解码器有时也能正常工作。奇怪的是,如果我使用 DIB,会收到 ffmpeg 错误,但视频保存正常。如果我使用 RAW,则不会出现错误,但 Windows 视频播放器无法打开它。在 VLC 中一切正常。
最后,我认为关键在于 OpenCV 并非设计为视频捕捉库——它甚至不支持音频。VideoWriter 很有用,但 99% 的情况下,你最好将所有图像保存到一个文件夹中,然后使用 ffmpeg 将它们转换成有用的视频。
解决方案 3:
就我而言,我发现 Writer 的大小必须与相机或文件的帧大小匹配。因此,我首先读取帧大小,然后将其应用于 Writer 设置,如下所示。
(grabbed, frame) = camera.read()
fshape = frame.shape
fheight = fshape[0]
fwidth = fshape[1]
print fwidth , fheight
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc, 20.0, (fwidth,fheight))
解决方案 4:
您需要获取捕获的确切尺寸,如下所示:
import cv2
cap = cv2.VideoCapture(0)
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH) + 0.5)
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT) + 0.5)
size = (width, height)
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('your_video.avi', fourcc, 20.0, size)
while(True):
_, frame = cap.read()
cv2.imshow('Recording...', frame)
out.write(frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
out.release()
cv2.destroyAllWindows()
解决方案 5:
我也遇到了同样的问题,但当我使用“MJPG”而不是'XVID'
我用过
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
而不是
fourcc = cv2.VideoWriter_fourcc(*'XVID')
解决方案 6:
请确保设置正确的宽度和高度。您可以像下面这样设置
cv2.VideoWriter('output.avi', fourcc, 20.0, (int(cap.get(3)), int(cap.get(4))))
解决方案 7:
这个答案涵盖了变量方面的内容,重要的是,输出大小应与输入帧和视频大小相同。
import cv2
save_name = "output.mp4"
fps = 10
width = 600
height = 480
output_size = (width, height)
out = cv2.VideoWriter(save_name,cv2.VideoWriter_fourcc('M','J','P','G'), fps , output_size )
cap = cv2.VideoCapture(0) # 0 for webcam or you can put in videopath
while(True):
_, frame = cap.read()
cv2.imshow('Video Frame', frame)
out.write(cv2.resize(frame, output_size ))
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
out.release()
cv2.destroyAllWindows()
解决方案 8:
这个答案仅在 MacOS 中测试过,但它可能也适用于 Linux 和 Windows。
import numpy as np
import cv2
cap = cv2.VideoCapture(0)
# Get the Default resolutions
frame_width = int(cap.get(3))
frame_height = int(cap.get(4))
# Define the codec and filename.
out = cv2.VideoWriter('output.avi',cv2.VideoWriter_fourcc('M','J','P','G'), 10, (frame_width,frame_height))
while(cap.isOpened()):
ret, frame = cap.read()
if ret==True:
# write the frame
out.write(frame)
cv2.imshow('frame',frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
else:
break
# Release everything if job is finished
cap.release()
out.release()
cv2.destroyAllWindows()
解决方案 9:
其他答案适用于保存单个视频源。但如果您有多个视频源,例如多个 IP 摄像头源、RTSP 流或网络摄像头,您可能需要一次性录制所有视频源。这里有一个使用多线程同时保存多个视频源的方法。其思路是为每个视频源设置两个线程:1)专用于从流中读取帧;2)专用于处理帧(显示和保存)。
由于cv2.VideoCapture.read()
这是一个阻塞操作,我们必须将读取帧与保存帧分开。我们可以在其独立的线程中读取帧,从而减少 I/O 操作造成的延迟,从而提高性能。通过将帧捕获专门用于其自己的线程,始终会有帧可供处理,而不必等待 I/O 操作完成并返回新帧。第二个线程专门用于处理帧并将其保存到输出文件。我们可以将所有这些封装到一个对象中,无论同时录制的视频流数量有多少,都可以在该对象中进行缩放。请务必将视频src
参数更改为您自己的视频源。以下是同时录制三个视频流的示例。
from threading import Thread
import cv2
import time
class VideoWriterWidget(object):
def __init__(self, video_file_name, src=0):
# Create a VideoCapture object
self.frame_name = str(src)
self.video_file = video_file_name
self.video_file_name = video_file_name + '.avi'
self.capture = cv2.VideoCapture(src)
# Default resolutions of the frame are obtained (system dependent)
self.frame_width = int(self.capture.get(3))
self.frame_height = int(self.capture.get(4))
# Set up codec and output video settings
self.codec = cv2.VideoWriter_fourcc('M','J','P','G')
self.output_video = cv2.VideoWriter(self.video_file_name, self.codec, 30, (self.frame_width, self.frame_height))
# Start the thread to read frames from the video stream
self.thread = Thread(target=self.update, args=())
self.thread.daemon = True
self.thread.start()
# Start another thread to show/save frames
self.start_recording()
print('initialized {}'.format(self.video_file))
def update(self):
# Read the next frame from the stream in a different thread
while True:
if self.capture.isOpened():
(self.status, self.frame) = self.capture.read()
def show_frame(self):
# Display frames in main program
if self.status:
cv2.imshow(self.frame_name, self.frame)
# Press Q on keyboard to stop recording
key = cv2.waitKey(1)
if key == ord('q'):
self.capture.release()
self.output_video.release()
cv2.destroyAllWindows()
exit(1)
def save_frame(self):
# Save obtained frame into video output file
self.output_video.write(self.frame)
def start_recording(self):
# Create another thread to show/save frames
def start_recording_thread():
while True:
try:
self.show_frame()
self.save_frame()
except AttributeError:
pass
self.recording_thread = Thread(target=start_recording_thread, args=())
self.recording_thread.daemon = True
self.recording_thread.start()
if __name__ == '__main__':
src1 = 'Your link1'
video_writer_widget1 = VideoWriterWidget('Camera 1', src1)
src2 = 'Your link2'
video_writer_widget2 = VideoWriterWidget('Camera 2', src2)
src3 = 'Your link3'
video_writer_widget3 = VideoWriterWidget('Camera 3', src3)
# Since each video player is in its own thread, we need to keep the main thread alive.
# Keep spinning using time.sleep() so the background threads keep running
# Threads are set to daemon=True so they will automatically die
# when the main thread dies
while True:
time.sleep(5)
相关摄像头/IP/RTSP/流媒体、FPS、视频、线程和多处理帖子
Python OpenCV 从相机流式传输 - 多线程、时间戳
使用 OpenCV cv2.VideoCapture 在 Python 中从 IP 摄像机进行视频流传输
如何使用 OpenCV 捕获多个摄像机流?
OpenCV 实时流视频捕获速度很慢。如何解决丢帧问题或使其与实时同步?
使用 OpenCV VideoWriter 将 RTSP 流存储为视频文件
OpenCV视频保存
Python OpenCV 多处理 cv2.VideoCapture mp4
解决方案 10:
我遇到了同样的问题然后我尝试了这个:
frame = cv2.flip(frame,180)
而不是
frame= cv2.flip(frame,0)
并且它正在发挥作用。
解决方案 11:
我遇到的不是上述问题的编解码器问题或尺寸问题。相反,我的问题是由于我的输出帧是灰度的。
我必须创建一个带有参数 isColor=False 的 VideoWriter
out = cv2.VideoWriter(output_path,
cv2.VideoWriter_fourcc(*'mp4v'),
30,
(INPUT_VIDEO_WIDTH,INPUT_VIDEO_HEIGHT),
isColor=False
)
API 文档错误地指出该标志目前仅在 Windows 上受支持。我已经在 Ubuntu 20.04 上进行了测试,并使用了 opencv-python==4.2.0.34,它最终正确地写入了文件。
解决方案 12:
我有点晚了,但是VidGear
Python 库的WriteGear API 可以在任何平台上自动将 OpenCV 帧实时传输到 FFmpeg 中,并支持硬件编码器,同时提供相同的 opencv-python 语法。这是一个基本的 Python 示例:
# import libraries
from vidgear.gears import WriteGear
import cv2
output_params = {"-vcodec":"libx264", "-crf": 0, "-preset": "fast"} #define (Codec,CRF,preset) FFmpeg tweak parameters for writer
stream = cv2.VideoCapture(0) #Open live webcam video stream on first index(i.e. 0) device
writer = WriteGear(output_filename = 'Output.mp4', compression_mode = True, logging = True, **output_params) #Define writer with output filename 'Output.mp4'
# infinite loop
while True:
(grabbed, frame) = stream.read()
# read frames
# check if frame empty
if not is grabbed:
#if True break the infinite loop
break
# {do something with frame here}
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# write a modified frame to writer
writer.write(gray)
# Show output window
cv2.imshow("Output Frame", frame)
key = cv2.waitKey(1) & 0xFF
# check for 'q' key-press
if key == ord("q"):
#if 'q' key-pressed break out
break
cv2.destroyAllWindows()
# close output window
stream.release()
# safely close video stream
writer.close()
# safely close writer
来源:https ://abhitronix.github.io/vidgear/latest/gears/writegear/compression/usage/#using-compression-mode-with-opencv
您可以查看VidGear Docs以了解更多高级应用程序和功能。
解决方案 13:
Nuru 答案确实有效,唯一的事情是删除循环frame = cv2.flip(frame,0)
下的 这一行if ret==True:
,它将输出视频文件而不翻转
解决方案 14:
举个例子:
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
out_corner = cv2.VideoWriter('img_corner_1.avi',fourcc, 20.0, (640, 480))
在那个地方,必须将 X、Y 定义为宽度和高度
但是,当您创建图像(例如空白图像)时,您必须将 Y、X 定义为高度和宽度:
img_corner = np.zeros((480, 640, 3), np.uint8)
解决方案 15:
import cv2
cap = cv2.VideoCapture(0)
fourcc = cv2.VideoWriter_fourcc('X','V','I','D')
frame_width = int(cap.get(3))
frame_height = int(cap.get(4))
out = cv2.VideoWriter('output.mp4', fourcc, 20,(frame_width,frame_height),True )
print(int(cap.get(3)))
print(int(cap.get(4)))
while(cap.isOpened()):
ret,frame = cap.read()
if ret == True:
print(frame.shape)
out.write(frame)
cv2.imshow('Frame', frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
else:
break
cap.release()
out.release()`enter code here`
cv2.destroyAllWindows()
这种方法效果不错,但问题是视频尺寸相对较小,导致什么都拍不出来。所以请确保视频和要录制的图像的高度和宽度一致。如果在录制视频后进行了一些处理,则必须确认尺寸(录制前后)。希望这能节省一些人的时间。
解决方案 16:
正如@ปรีดาตั้งนภากร所说:Writer的尺寸必须与相机或文件的帧相匹配。
您可以使用这样的代码来检查您的相机是否是(640,480):
print(int(cap.get(3)), int(cap.get(4)))
我自己发现我的相机分辨率是(1280, 720),就把(640, 480)换成了(1280, 720)。这样就能保存视频了。
解决方案 17:
我想写一个灰度图像,但没有什么效果,我不得不在 VideoWriter 中添加一个0
out = cv2.VideoWriter(outfilename, fourcc, fps, (width, height), 0)
解决方案 18:
您必须根据图片源的大小指定视频的宽度和高度。否则,它将创建一个大约 5.5KB 的文件。
result = cv2.VideoWriter(name,cv2.VideoWriter_fourcc(*'MJPG'), 10, (**960, 540**))
扫码咨询,免费领取项目管理大礼包!