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

Question

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

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

非常感谢您的建议；）

Answer 1

在这里（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()