如何使用子进程(Python)将Picamera视频传输到FFMPEG

时间:2017-07-20 21:14:22

标签: python ffmpeg subprocess raspberry-pi3

我看到大量关于将生动的流直接传输到FFMPEG以进行编码,复用和休息的信息,但这些用例主要来自bash;类似于:

raspivid -n -w 480 -h 320 -b 300000 -fps 15 -t 0 -o - | ffmpeg -i - -f mpegts udp://192.168.1.2:8090ffmpeg

我希望利用Picamera库的功能,这样我就可以使用OpenCV进行并发处理,同时还可以使用FFMPEG进行流式处理。但我无法弄清楚如何正确打开FFMPEG作为子进程并将视频数据传输给它。我已经看到了很多尝试,unanswered posts和人claiming to have done it,但似乎没有一个尝试在我的Pi上工作。

我应该使用Picamera创建视频缓冲区并将原始视频传输到FFMPEG吗?我可以使用camera.capture_continuous()并将我用于OpenCV计算的bgr24图像传递给FFMPEG吗?

我尝试了各种各样的变化,我不确定我是否只是误解了如何使用子进程模块,FFMPEG,或者我只是缺少一些设置。我知道原始流不会有任何元数据,但我不完全确定我需要为FFMPEG提供哪些设置才能理解我提供的内容。

我有一台Wowza服务器我最终将要流式传输,但我目前正在通过流式传输到笔记本电脑上的VLC服务器进行测试。我目前试过这个:

import subprocess as sp
import picamera
import picamera.array
import numpy as np

npimage = np.empty(
        (480, 640, 3),
        dtype=np.uint8)
with picamera.PiCamera() as camera:
    camera.resolution = (640, 480)
    camera.framerate = 24

    camera.start_recording('/dev/null', format='h264')
    command = [
        'ffmpeg',
        '-y',
        '-f', 'rawvideo',
        '-video_size', '640x480',
        '-pix_fmt', 'bgr24',
        '-framerate', '24',
        '-an',
        '-i', '-',
        '-f', 'mpegts', 'udp://192.168.1.54:1234']
    pipe = sp.Popen(command, stdin=sp.PIPE,
                    stdout=sp.PIPE, stderr=sp.PIPE, bufsize=10**8)
    if pipe.returncode != 0:
        output, error = pipe.communicate()
        print('Pipe failed: %d %s %s' % (pipe.returncode, output, error))
        raise sp.CalledProcessError(pipe.returncode, command)

    while True:
        camera.wait_recording(0)
        for i, image in enumerate(
                        camera.capture_continuous(
                            npimage,
                            format='bgr24',
                            use_video_port=True)):
            pipe.stdout.write(npimage.tostring())
    camera.stop_recording()

我还尝试将流写入类似文件的对象,该对象只是创建FFMPEG子进程并写入它的stdin(在初始化picam时,camera.start_recording()可以被赋予这样的对象):

class PipeClass():
    """Start pipes and load ffmpeg."""

    def __init__(self):
        """Create FFMPEG subprocess."""
        self.size = 0
        command = [
            'ffmpeg',
            '-f', 'rawvideo',
            '-s', '640x480',
            '-r', '24',
            '-i', '-',
            '-an',
            '-f', 'mpegts', 'udp://192.168.1.54:1234']

        self.pipe = sp.Popen(command, stdin=sp.PIPE,
                         stdout=sp.PIPE, stderr=sp.PIPE)

        if self.pipe.returncode != 0:
            raise sp.CalledProcessError(self.pipe.returncode, command)

    def write(self, s):
        """Write to the pipe."""
        self.pipe.stdin.write(s)

    def flush(self):
        """Flush pipe."""
        print("Flushed")

usage:
(...)
with picamera.PiCamera() as camera:
    p = PipeClass()
    camera.start_recording(p, format='h264')
(...)

对此的任何帮助都会很棒!

3 个答案:

答案 0 :(得分:2)

我已经能够使用以下内容将PiCamera输出流式传输到ffmpeg:

import picamera
import subprocess

# start the ffmpeg process with a pipe for stdin
# I'm just copying to a file, but you could stream to somewhere else
ffmpeg = subprocess.Popen([
    'ffmpeg', '-i', '-',
    '-vcodec', 'copy',
    '-an', '/home/pi/test.mpg',
    ], stdin=subprocess.PIPE)

# initialize the camera
camera = picamera.PiCamera(resolution=(800, 480), framerate=25)

# start recording to ffmpeg's stdin
camera.start_recording(ffmpeg.stdin, format='h264', bitrate=2000000)

或者那不是你想要的吗?

答案 1 :(得分:0)

乍一看我遇到的两个问题:

  1. 在您的第一个示例中,您将数据写入子流程的stdout而不是stdin。这肯定不起作用,可能会导致挂起。

  2. 在这两个示例中,您将使用stdin=sp.PIPE, stderr=sp.PIPE启动该过程,然后从不读取这些管道。这意味着只要ffmpeg写入足够的输出来填充管道缓冲区,它就会阻塞并且你将遇到死锁。使用默认的stdout=None, stderr=None让ffmpeg的输出转到进程的stdout和stderr,或者将它们连接到打开到/dev/null的文件句柄以丢弃输出。或者使用communicate方法在每次写入输入时获取输出,并使用它做一些有用的事情(比如监视流的状态)。

答案 2 :(得分:0)

您好,您可以使用opencv& ffmpeg&把它重新传给wowza或者其他。

这是一个opencv&& amp; FFMPEG

    int main(int argc, char* argv[])
{
    if (argc < 4){
        cout << "eksik parametre" << endl;
        return -1;
    }
    int fps = 1;                        //fps degeri varsayilan 1
    char *input_adress = argv[1];       //goruntunun alinacagi dosya yada adres bilgisi
    char *output_adress = argv[3];      //ciktinin gonderilecegi dosya yada adres bilgisi
    sscanf(argv[2], "%d", &fps);        //fps degeri okundu
    VideoCapture video(input_adress);   //kamera acildi
    if (!video.isOpened()){
        cout << "Yayin acilamadi!!!" << endl;
        getchar();
        return -1;
    }
    Mat frame;//frame ornegi
    FILE *pipe;//pipe icin acilan process in input streami
    char *cmd = (char*)calloc(100 + sizeof(output_adress), sizeof(char));//komut icin alan alindi
    sprintf(cmd, "ffmpeg -y -f image2pipe -vcodec mjpeg -r %d -i -  -r %d -vcodec libx264 -f flv %s", fps, fps, output_adress);//ffmpeg komutu
    //ffmpeg komutu aciliyor
    if (!(pipe = _popen(cmd, "wb"))){
        cout << "Acilamadi!!!" << endl;
        return 1;
    }
    float wait_time = 1000.0f / (float)fps;//kac milisaniye bekletilecek
    while (true)
    {
        try {
            //videodan siradaki frame okunuyor
            video >> frame;
            if (frame.empty())              //eger bos frame ise video bitmis demektir
                break;
            adjust_brightness(frame);       //parlaklik ayarlamasini yapıyoruz
            write_jpeg(frame, pipe);        //jpeg formatina cevirip pipe a yazıyoruz
            if (waitKey(wait_time) >= 0) break;
        }
        catch (Exception ex){
        }
    }
    video.release();
    return 0;
}

在write_jpeg moethod fwrite和fflush中调用。

这将使用

进行调用

testOutput.exe ORIGINAL_SOURCE RTMP_OUTPUT