python中的openCV视频保存

Question

我正在尝试保存视频，但无法正常工作。 我按照 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()

怎么了？

Answer 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()

Answer 2

https://github.com/ContinuumIO/anaconda-issues/issues/223的 jveitchmichaelis 提供了详尽的答案。 在这里，我复制了他的答案：

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 文件夹添加到我的路径中，尽管这不会有所作为，因为它已融入 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 将它们变成有用的视频。

Answer 3

就我而言，我发现 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))

Answer 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()

Answer 5

我也遇到了同样的问题，但是当我使用“MJPG”而不是 'XVID'

我用了

fourcc = cv2.VideoWriter_fourcc(*'MJPG')

代替

fourcc = cv2.VideoWriter_fourcc(*'XVID')

Answer 6

请确保设置正确的宽度和高度。 你可以像下面这样设置

cv2.VideoWriter('output.avi', fourcc, 20.0, (int(cap.get(3)), int(cap.get(4))))

Answer 7

这是一个仅在 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()

Answer 8

我遇到了同样的问题，然后我尝试了这个：

frame = cv2.flip(frame,180) 

代替

frame= cv2.flip(frame,0) 

它正在工作。

Answer 9

我没有像上面的答案那样遇到编解码器问题或尺寸问题。 相反，我的问题是因为我的输出帧是灰度的。

我必须使用参数 isColor=False 创建一个 VideoWriter

out = cv2.VideoWriter(output_path,
                      cv2.VideoWriter_fourcc(*'mp4v'),
                      30,
                      (INPUT_VIDEO_WIDTH,INPUT_VIDEO_HEIGHT),
                      isColor=False
                      )

在API Docs中，它错误地表示该标志目前仅在 Windows 上受支持。 我已经在 Ubuntu 20.04 上使用 opencv-python==4.2.0.34 进行了测试，它最终正确写入文件。

Answer 10

这个答案涵盖了变量方面的内容，重要的是，输入帧和视频大小的输出大小应该相同。

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()

Answer 11

其他答案适用于保存单个视频提要。 但是，如果您有多个视频源，例如大量 IP 摄像机源、RTSP 流或网络摄像头，您可能希望一次录制它们。 这是一种使用多线程同时保存多个视频源的方法。 这个想法是为每个视频源设置两个线程：1）专用于仅从流中读取帧，2）专用于处理帧（显示和保存）。

由于cv2.VideoCapture.read()是阻塞操作，我们必须将读取帧与保存帧分开。 我们可以在其自己的独立线程中读取帧，以通过减少 I/O 操作引起的延迟来提高性能。 通过将帧捕获专用于它自己的线程，总会有一个准备好处理的帧，而不必等待 I/O 操作完成并返回一个新的帧。 第二个线程专用于处理帧并将其保存到输出文件。 我们可以将所有这些封装到一个对象中，无论同时视频流的数量如何，都可以对其进行缩放。 请务必将 video 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、视频、线程和多处理帖子

1. 来自相机的 Python OpenCV 流式传输 - 多线程、时间戳

2. 使用 OpenCV cv2.VideoCapture 在 Python 中从 IP 摄像机进行视频流传输

3. 如何使用 OpenCV 捕获多个摄像机流？

4. OpenCV 实时流式视频捕获很慢。 如何丢帧或实时同步？

5. 使用 OpenCV VideoWriter 将 RTSP 流存储为视频文件

6. OpenCV 视频保存

7. Python OpenCV 多处理 cv2.VideoCapture mp4

Answer 12

Nuru 答案实际上有效，唯一的事情是在if ret==True:循环下删除这条线frame = cv2.flip(frame,0) ，它将输出视频文件而不翻转

Answer 13

举个例子 ：

fourcc = cv2.VideoWriter_fourcc(*'MJPG')
out_corner = cv2.VideoWriter('img_corner_1.avi',fourcc, 20.0, (640, 480))

在那个地方，必须将 X,Y 定义为宽度和高度

但是，当您创建图像（例如空白图像）时，您必须将 Y,X 定义为 height 和 width ：

    img_corner = np.zeros((480, 640, 3), np.uint8)

Answer 14

正如@ปรีดา ตั้งนภากร所说：Writer的大小必须与相机或文件的框架相匹配。

您可以使用此类代码检查您的相机是否为 (640, 480)：

print(int(cap.get(3)), int(cap.get(4)))

对于我自己，我发现我的相机是 (1280, 720) 并用 (1280, 720) 替换 (640, 480)。 然后它可以保存视频。

Answer 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()

这可以正常工作，但是视频尺寸相对非常小的问题意味着没有捕获任何内容。 因此，请确保视频的高度和宽度与您要录制的图像相同。 如果您在捕获视频后使用某些操作，则必须确认大小（之前和之后）。 希望它能节省一些时间

Answer 16

我想写一个灰度图像，但对我没有任何作用，我必须在 VideoWriter 中添加一个0

out = cv2.VideoWriter(outfilename, fourcc, fps, (width, height), 0)

Answer 17

我有点晚了，但是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以获得更高级的应用程序和功能。

Answer 18

您必须根据图像馈送的大小指定视频的宽度和高度。 否则，它将创建大约 5.5KB 的文件大小。

result = cv2.VideoWriter(name,cv2.VideoWriter_fourcc(*'MJPG'), 10, (**960, 540**))