如何使用python从ffmpeg管道输出？

Question

我正在尝试在 Python 中管道输出来自 FFmpeg 的输出。我正在从视频采集卡中读取图像，并且成功地使用 dshow 从命令行将其读取到输出文件中。我正在尝试将图像从卡中抓取到我的 OpenCv 代码中，以便能够进一步使用数据。不幸的是，当我输出图像时，我只得到了视频的显示，如链接所示：

链接：s000.tinyupload.com/?file_id=15940665795196022618。

我使用的代码如下图：

import cv2
import subprocess as sp
import numpy
import sys
import os
old_stdout=sys.stdout
log_file=open("message.log","w")
sys.stdout=log_file

FFMPEG_BIN = "C:/ffmpeg/bin/ffmpeg.exe"
command = [ FFMPEG_BIN, '-y',
            '-f', 'dshow', '-rtbufsize', '100M',
            '-i', 'video=Datapath VisionAV Video 01' ,
             '-video_size', '640x480',
              '-pix_fmt', 'bgr24', '-r','25',  
          '-f', 'image2pipe', '-' ]    
pipe = sp.Popen(command, stdout = sp.PIPE, bufsize=10**8)

while True:
    # Capture frame-by-frame
    raw_image = pipe.stdout.read(640*480*3)
      # transform the byte read into a numpy array
    image =  numpy.frombuffer(raw_image, dtype='uint8')
    print(image)
    image = image.reshape((480,640,3))          
    if image is not None:
        cv2.imshow('Video', image)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
    pipe.stdout.flush()
sys.stdout=old_stdout
log_file.close()
cv2.destroyAllWindows()

请为我提供一些解决此问题的建议。非常感谢您的帮助。

Answer 1

您拨打sp.Popen后，您已经与它通信。您可以使用以下代码：

try:    
    pipe = sp.Popen(command, stdout=sp.PIPE, stderr=sp.STDOUT, universal_newlines=True)`

    ffmpeg_output, _ = pipe.communicate()

except  sp.CalledProcessError as err:
     print("FFmpeg stdout output on error:\n" + err.output)

最后，您可以打印输出以确保上述命令有效：

print(ffmpeg_output)

上述语句将显示与进程通信返回的输出。

Answer 2

我在控制台应用程序 FFmpeg 上挣扎了更长时间，最后放弃了。 这个扩展更容易：

pip install ffmpeg-python

Karl Kroening 在这里发布了 FFmpeg 与 Python 的非常好的集成。 有了这些例子，一个解决方案应该是可能的： https://github.com/kkroening/ffmpeg-python

Answer 3

这对我有用

import subprocess as sp
import json
import os
import numpy
import PIL
from imutils.video import FPS
import cv2 


def video_frames_ffmpeg():
    width = 640
    height = 360
    iterator = 0
    cmd = ['ffmpeg', '-loglevel', 'quiet',
           '-f', 'dshow',
           '-i', 'video=HD USB Camera',
           #'-vf','scale=%d:%d,smartblur'%(width,height),
           '-preset' ,'ultrafast', '-tune', 'zerolatency',
           '-f', 'rawvideo',
           '-pix_fmt','bgr24',
           '-']
    p = sp.Popen(cmd, stdout=sp.PIPE)

    while True:
        arr = numpy.frombuffer(p.stdout.read(width*height*3), dtype=numpy.uint8)
        iterator += 1 
        if len(arr) == 0:
            p.wait()
            print("awaiting")
            #return
        if iterator >= 1000:
            break

        frame = arr.reshape((height, width,3))
        cv2.putText(frame, "frame{}".format(iterator), (75, 70),
                cv2.cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255,255,255), 2)
        im = Image.fromarray(frame)
        im.save("ffmpeg_test/test%d.jpeg" % iterator)

        yield arr


from PIL import Image
from imutils.video import FPS
for i, frame in enumerate(video_frames_ffmpeg()):
    if i == 0:
        fps = FPS().start()
    else: fps.update()
fps.stop()
print("[INFO] approx. FPS: {:.2f}".format(fps.fps()))
cv2.destroyAllWindows()

Answer 4

试试：

command = [ FFMPEG_BIN,
            '-rtbufsize', '100M',
            '-i', '/dev/video0' , #change to here to your camera device
            '-video_size', '640x480',
            '-pix_fmt', 'bgr24',# '-r','25',
            '-f', 'image2pipe', #'-'
            '-vcodec', 'rawvideo', '-an', '-'
            ]

我不知道'-vcodec', 'rawvideo', '-an', '-' 这有什么帮助， 如果没有它，我的代码将无法工作。