Сохранение потока RTSP в виде видеофайла с помощью OpenCV VideoWriter

Я разрабатываю модуль Python с OpenCV, который подключается к потоку RTSP для выполнения некоторой предварительной обработки видео (в основном, уменьшения частоты кадров и разрешения), а затем сохраняет его в файловой системе.

Но, даже перепробовав несколько кодеков, поискав похожие разработки... Всегда получаю пустое видео. Я видел этот другой поток ( cv::VideoWriter дает нечитаемое видео ), который может быть похожим, но был разработан на C ++.

Кто-нибудь работал над этим? Обычно я использую образец потока RTSP в качестве эталона, например rtsp://freja.hiof.no:1935/rtplive/definst/ hessdalen03.stream, и могу правильно получать и даже просматривать поток из VLC.

Я видел довольно много тем, в которых обсуждалось, как захватывать видео из потока RTSP или как работать с классами и видеофайлами VideoWriters и VideoReaders, но почти ничего не объединяло их.

Любая помощь будет высоко оценена:) Спасибо!!

Редактировать 1: пример кода, используемый для хранения кадров.

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import cv2
import numpy
# Test frame.
width, height = 400, 300
width_2, height_2 = int(width / 2), int(height / 2)
frame = numpy.zeros((height, width, 3), numpy.uint8)
cv2.rectangle(frame, (0, 0), (width_2, height_2), (255, 0, 0), cv2.FILLED)
cv2.rectangle(frame, (width_2, height_2), (width, height), (0, 255, 0), cv2.FILLED)
frames = [frame for _ in range(100)]
fps = 25
# Define the codec.
#fourcc = cv2.VideoWriter_fourcc(*'X264')
#fourcc = cv2.VideoWriter_fourcc(*'XVID')
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
# Create VideoWriter object
out = cv2.VideoWriter(filename='video.avi',
  fourcc=fourcc,
  apiPreference=cv2.CAP_FFMPEG,
  fps=float(fps),
  frameSize=(width, height),
  isColor=True)
result = 0
for frame in frames:
 result += 0 if out.write(frame) is None else 1
print(result)
out.release()

Редактировать 2: Решение

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import cv2
import numpy
# Test frame.
width, height = 400, 300
width_2, height_2 = int(width / 2), int(height / 2)
frame1 = numpy.zeros((height, width, 3), numpy.uint8)
cv2.rectangle(frame1, (0, 0), (width_2, height_2), (255, 0, 0), cv2.FILLED)
cv2.rectangle(frame1, (width_2, height_2), (width, height), (0, 255, 0), cv2.FILLED)
cv2.imwrite('frame1.jpg', frame1)
frame2 = numpy.zeros((height, width, 3), numpy.uint8)
cv2.rectangle(frame2, (width_2, 0), (width, height_2), (255, 0, 0), cv2.FILLED)
cv2.rectangle(frame2, (0, height_2), (width_2, height), (0, 255, 0), cv2.FILLED)
cv2.imwrite('frame2.jpg', frame2)
range1 = [frame1 for _ in range(10)]
range2 = [frame2 for _ in range(10)]
frames = range1 + range2 + range1 + range2 + range1
fps = 2
# Define the codec.
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
# Create VideoWriter object
out = cv2.VideoWriter('video.avi', fourcc, float(fps), (width, height))
for frame in frames:
 out.write(frame)
out.release()

Решение проблемы

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

from threading import Thread
import cv2
class RTSPVideoWriterObject(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__':
 rtsp_stream_link = 'your stream link!'
 video_stream_widget = RTSPVideoWriterObject(rtsp_stream_link)
 while True:
 try:
 video_stream_widget.show_frame()
 video_stream_widget.save_frame()
 except AttributeError:
 pass