Python Подпроцесс ffmpeg: Сломанный канал - PullRequest
1 голос
/ 03 февраля 2020

Следующий скрипт читает видео с OpenCV, применяет преобразование к каждому кадру и пытается записать его с помощью ffmpeg. Моя проблема в том, что у меня не работает ffmpeg с модулем subprocess. Я всегда получаю ошибку BrokenPipeError: [Errno 32] Broken pipe в строке, где я пытаюсь написать в стандартный ввод. Почему, что я делаю не так?

# Open input video with OpenCV
video_in = cv.VideoCapture(src_video_path)
frame_width = int(video_in.get(cv.CAP_PROP_FRAME_WIDTH))
frame_height = int(video_in.get(cv.CAP_PROP_FRAME_HEIGHT))
fps = video_in.get(cv.CAP_PROP_FPS)
frame_count = int(video_in.get(cv.CAP_PROP_FRAME_COUNT))
bitrate = bitrate * 4096 * 2160 / (frame_width * frame_height)

# Process video in ffmpeg pipe
# See http://zulko.github.io/blog/2013/09/27/read-and-write-video-frames-in-python-using-ffmpeg/
command = ['ffmpeg',
           '-loglevel', 'error',
           '-y',
           # Input
           '-f', 'rawvideo',
           '-vcodec', 'rawvideo'
           '-pix_fmt', 'bgr24',
           '-s', str(frame_width) + 'x' + str(frame_height),
           '-r', str(fps),
           # Output
           '-i', '-',
           '-an',
           '-vcodec', 'h264',
           '-r', str(fps),
           '-b:v', str(bitrate) + 'M',
           '-pix_fmt', 'bgr24',
           dst_video_path
           ]
pipe = sp.Popen(command, stdin=sp.PIPE)

for i_frame in range(frame_count):
    ret, frame = video_in.read()
    if ret:
        warped_frame = cv.warpPerspective(frame, homography, (frame_width, frame_height))
        pipe.stdin.write(warped_frame.astype(np.uint8).tobytes())
    else:
        print('Stopped early.')
        break
print('Done!')

1 Ответ

2 голосов
/ 03 февраля 2020

Существует пропущенная запятая после '-vcodec', 'rawvideo' !!!

Мне потребовалось около часа, чтобы заметить ...

Вы также должны закрыть stdin и ждать до print('Done!'):

pipe.stdin.close()
pipe.wait()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...