为什么 python 屏幕录像机不创建 .avi 文件？

问题描述

我正在尝试制作输出 AVI 文件的 python 屏幕录像机。按“q”后循环应该停止，但这不起作用。当我用 Ctrl + C 中断程序时，它也不输出任何内容。我使用的是 Mac 操作系统。

import cv2
import numpy as np
import pyautogui

SCREEN_SIZE = (1440,900)
fourcc = cv2.VideoWriter_fourcc(*"XVID")
out = cv2.VideoWriter("output2.avi",fourcc,20.0,(SCREEN_SIZE))

while True:
    img = pyautogui.screenshot()
    frame = np.array(img)
    frame = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
    out.write(frame)
    cv2.imshow("screenshot",frame)
    if cv2.waitKey(1) == ord("q"):
        break

cv2.destroyAllWindows()
out.release()

非常感谢您的建议;)

解决方法

在这里（OpenCV videowrite doesn't write video）我发现最常见的问题是如果分辨率错误，它不会保存。

为确保您使用的屏幕分辨率正确，请使用 SCREEN_SIZE = pyautogui.size()

完整的代码是：

import cv2 as cv2
import numpy as np
import pyautogui

SCREEN_SIZE = pyautogui.size() # changed this line

fourcc = cv2.VideoWriter_fourcc(*"XVID")
out = cv2.VideoWriter("output2.avi",fourcc,20.0,SCREEN_SIZE)

while True:
    img = pyautogui.screenshot()
    frame = np.array(img)
    frame = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
    out.write(frame)
    cv2.imshow("screenshot",frame)
    if cv2.waitKey(1) == ord("q"):
        break

cv2.destroyAllWindows()
out.release()