VideoWriter выводит поврежденный видеофайл

Question

Это мой код для сохранения потоковой передачи в web_cam. Работает, но проблема с выходным видеофайлом.

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 1

Вот простой код для сохранения кадров в видеофайл. Я рекомендую создать другой поток для получения фреймов, так как cv2.VideoCapture.read() блокирует. Это может быть дорого и вызывать задержку, так как основной поток должен ждать, пока он не получит кадр. Помещая эту операцию в отдельный поток, который просто фокусируется на захвате кадров и обработке / сохранении кадров в главном потоке, это значительно повышает производительность благодаря уменьшению задержки ввода-вывода. Вы также можете экспериментировать с другими кодеками, но использование MJPG должно быть безопасным, поскольку оно встроено в OpenCV.

from threading import Thread
import cv2

class WebcamVideoWriter(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('M','J','P','G')
        self.output_video = cv2.VideoWriter('output.avi', 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()

    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('frame', 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)

if __name__ == '__main__':
    webcam_videowriter = WebcamVideoWriter()
    while True:
        try:
            webcam_videowriter.show_frame()
            webcam_videowriter.save_frame()
        except AttributeError:
            pass

Answer 2

Я добавил codec parameter к функции cv2.videowriter.

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))
fps = cap.get(cv2.CAP_PROP_FPS)
width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)   # float
height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
codec = cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')
out = cv2.VideoWriter('output.avi',codec,fps, (int(width),\
                                                    int (height)))

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

Я надеюсь, вы можете увидеть, что отличается в моем коде и вашем коде. Шахта работает сейчас. Используя MJPG код c для расширения .avi, отступ немного запутан, прошу прощения, потому что я очень первый раз. Файл video больше не поврежден. Я получил информацию от: Ссылка

Answer 3

Выходной файл поврежден из-за неправильной частоты кадров и разрешения кадров. Используя этот код:

out = cv2.VideoWriter('output.avi', -1, 20.0, (640,480))

Мы устанавливаем частоту кадров / кадров в секунду 20. Что было не правильно. Кроме того, ширина и высота рамы были неправильными. Я решил, получив fps, width, height из захваченного профиля web_cam.

cap = cv2.VideoCapture(0)  #web-cam capture

fps = cap.get(cv2.CAP_PROP_FPS)
width  = cap.get(cv2.CAP_PROP_FRAME_WIDTH)   # float
height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)  # float
out = cv2.VideoWriter('output.avi', -1,fps, (int(width), int(height)))