Как явно получить доступ к бэкенду mjpeg для видеозахвата в opencv - PullRequest
Новая
       29

Как явно получить доступ к бэкенду mjpeg для видеозахвата в opencv

2 голосов
/ 14 апреля 2020

Когда я выполняю следующее: 

availableBackends = [cv2.videoio_registry.getBackendName(b) for b in v2.videoio_registry.getBackends()]
print(availableBackends)

Я получаю ['FFMPEG', 'GSTREAMER', 'INTEL_MFX', 'V4L2', 'CV_IMAGES', 'CV_MJPEG'].

Если я сейчас попробую: 

print(cv2.CAP_FFMPEG)
print(cv2.CAP_GSTREAMER)
print(cv2.CAP_INTEL_MFX)
print(cv2.CAP_V4L2)
print(cv2.CAP_IMAGES)
print(cv2.CAP_MJPEG)

вся работа, кроме последней AttributeError: module 'cv2.cv2' has no attribute 'CAP_MJPEG'.

Как явно установить бэкэнд cv2.CAP_MJPEG (cv2.CAP_CV_MJPEG также не работает)?

1 Ответ

2 голосов
/ 14 апреля 2020

Здесь вы можете увидеть все флаги .

Похоже, cv2.CAP_OPENCV_MJPEG - это то, что вы ищете.

Следующий тест создает MJPEG syntheti c AVI видеофайл и считывает видео с использованием cv2.CAP_OPENCV_MJPEG backend: 

import numpy as np
import cv2

#availableBackends = [cv2.videoio_registry.getBackendName(b) for b in cv2.videoio_registry.getBackends()]
#print(availableBackends)

print('cv2.CAP_OPENCV_MJPEG = ' + str(cv2.CAP_OPENCV_MJPEG))


intput_filename = 'vid.avi'

# Generate synthetic video files to be used as input:
###############################################################################
width, height, n_frames = 640, 480, 100  # 100 frames, resolution 640x480

# Use motion JPEG codec (for testing)
synthetic_out = cv2.VideoWriter(intput_filename, cv2.VideoWriter_fourcc(*'MJPG'), 25, (width, height))

for i in range(n_frames):
    img = np.full((height, width, 3), 60, np.uint8)
    cv2.putText(img, str(i+1), (width//2-100*len(str(i+1)), height//2+100), cv2.FONT_HERSHEY_DUPLEX, 10, (30, 255, 30), 20)  # Green number
    synthetic_out.write(img)

synthetic_out.release()
###############################################################################


# Read the video using CV_MJPEG backend
###############################################################################
cap = cv2.VideoCapture(intput_filename, cv2.CAP_OPENCV_MJPEG)

# Keep iterating break
while True:
    ret, frame = cap.read()  # Read frame from first video

    if ret:
        cv2.imshow('frame', frame)  # Display frame for testing
        cv2.waitKey(100) #Wait 100msec (for debugging)
    else:
        break

cap.release() #Release must be inside the outer loop
###############################################################################
...