无法将视频转换为灰度

Question

无法将视频转换为灰度

3

我正在尝试将从我的摄像头获取的低帧率视频转换为灰度。我已成功获取了视频，现在想将其转换为灰度。

我尝试过基本的opencv操作，但没有成功。当我打开生成的视频文件时，其中没有视频。

import cv2
import time

cap = cv2.VideoCapture('output.avi')
fourcc = cv2.VideoWriter_fourcc(*'XVID')
print(fourcc)
out = cv2.VideoWriter('grey.avi',fourcc, 30.0, (800,600))
while True:    
    ret, frame = cap.read()
    time.sleep(0.1)
    cv2.imshow('frame1',frame)
    frame = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)
    out.write(frame)
    cv2.imwrite('img.jpg',frame)
    cv2.imshow('frame',frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
out.release()
cv2.destroyAllWindows()

- High On Math

最有可能的问题是你的相机不是RGB格式，大多数网络摄像头都是YUV格式，你需要确定相机视频流的格式，以便传递正确的颜色转换枚举值。 - EdChum

我该如何找到它？ - High On Math

你应该设置好你的相机或获取它的功能：cap.get(CV_CAP_PROP_FORMAT)，这将返回当前格式，你需要做更多的研究。 - EdChum

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- nathancy · Accepted Answer

你需要更改cv2.VideoWriter中的isColor标志。当前视频编写器设置为彩色而不是灰度。您正在尝试错误地将3通道彩色图像（OpenCV默认为BGR）保存为灰度图像。

更改

out = cv2.VideoWriter('grey.avi',fourcc, 30.0, (800,600))

为

out = cv2.VideoWriter('grey.avi',fourcc, 30.0, (800,600), isColor=False)

此外，您的总体目标似乎是从流/摄像头捕获视频并以灰度格式保存捕获的视频。以下是一个“一站式”小部件，它从相机流链接（RTSP）读取帧，将每个帧转换为灰度，并将其保存为视频。将video_src更改为您的相机流链接。

from threading import Thread
import cv2

class VideoToGrayscaleWidget(object):
    def __init__(self, src=0):
        # Create a VideoCapture object
        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('X','V','I','D')
        self.output_video = cv2.VideoWriter('output.avi', self.codec, 30, (self.frame_width, self.frame_height), isColor=False)

        # Start the thread to read frames from the video stream
        self.thread = Thread(target=self.update, args=())
        self.thread.daemon = True
        self.thread.start()

    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):
        # Convert to grayscale and display frames
        if self.status:
            self.gray = cv2.cvtColor(self.frame, cv2.COLOR_BGR2GRAY)
            cv2.imshow('grayscale frame', self.gray)

        # 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 grayscale frame into video output file
        self.output_video.write(self.gray)

if __name__ == '__main__':
    video_src = 'Your video stream link!'
    video_stream_widget = VideoToGrayscaleWidget(video_src)
    while True:
        try:
            video_stream_widget.show_frame()
            video_stream_widget.save_frame()
        except AttributeError:
            pass