python - OpenCV2 : grayscaled video displays fine, 但不保存

Question

我正在尝试熟悉 OpenCV2 以进行基本的图像和视频处理。现在我无法保存灰度视频帧。以下代码(改编自 here )旨在保存我拍摄的凯旋门 ( arc.avi ) 的 6 秒视频的三个版本 - 一个未修改，一个完全灰度，一个仅在某些帧期间灰度，而在其他帧全彩.

import cv2

capture = cv2.VideoCapture('arc.avi')

fourcc = cv2.VideoWriter_fourcc(*'DIVX')
outgray = cv2.VideoWriter('vidgray.avi',fourcc, 30.0, (1280,720))
outorig = cv2.VideoWriter('vidorig.avi',fourcc, 30.0, (1280,720))
outhalf = cv2.VideoWriter('vidhalf.avi',fourcc, 30.0, (1280,720))

framecounter = 0
while(True):

    ret, frame = capture.read()

    grayFrame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    outorig.write(frame)
    outgray.write(grayFrame)

    #cv2.imshow('video gray', grayFrame)
    #cv2.imshow('video original', frame)

    if framecounter//40%2 == 1:
        outhalf.write(frame)
        cv2.imshow('video half', frame)
    else:
        outhalf.write(grayFrame)
        cv2.imshow('video half', grayFrame)

    if cv2.waitKey(1) == 27:
        break

    framecounter += 1

capture.release()
outgray.release()
outorig.release()
outhalf.release()
cv2.destroyAllWindows()

(我不知道如何附上我的视频文件，但如果你想重现这个问题，你应该能够在你选择的任何 .avi 文件上自己使用代码)。
vidorig.avi 保存得很好，并且在播放时看起来与原版完全一样(正如它的本意那样)。 vidgray.avi 是一个零帧的空视频文件(当我尝试播放它时，让我的 VLC 播放器感到困惑)。 vidhalf.avi 仅包含全彩帧，并简单地跳过它应该拥有的所有灰度帧。这导致视频更短，并且在帧之间出现奇怪的抖动。

请注意，问题仅在于保存的视频文件，而不是(据我所知)灰度过程本身。虽然我确实收到一条错误消息:

OpenCV Error: Assertion failed (scn == 3 || scn == 4) in cv::ipp_cvtColor, file ..\..\..\modules\imgproc\src\color.cpp, line 7456
Traceback (most recent call last):
  File "grayvid.py", line 15, in <module>
    grayFrame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
cv2.error: ..\..\..\modules\imgproc\src\color.cpp:7456: error: (-215) scn == 3 || scn == 4 in function cv::ipp_cvtColor

使用 cv2.imshow 在运行时所有三个都可以正常显示。只有保存的文件有问题。

我在 Windows 8 上，在 Python 3.5.4 中使用 OpenCV2。

请注意，这不是 Save grayscale video in OpenCV? 的副本——我没有得到 Nithin 的错误，建议的答案也不起作用(我的代码从一开始就实现了它们)。

Answer 1

writer VideoWriter 的功能类需要(通常)具有 RGB 格式的图像(参见 OpenCV Docu )。如果图像数据仅包含 1 个 channel ，则编码器很可能会误解图像数据。要存储灰度值图像，您可以通过将灰度 channel 复制到 R、G 和 B 来将 1 channel 灰度图像扩展为 3 channel 图像。这将由 cv2.COLOR_GRAY2BGR 完成。

在这种情况下，您的代码将如下所示:

ret, frame = capture.read()

grayFrame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
// expand dimension on gray valued image
grayFrame = cv2.cvtColor(grayFrame, cv2.COLOR_GRAY2BGR)
outorig.write(frame)
outgray.write(grayFrame)