使用ffmpeg在Python实现OpenCV Image推流至RTMP服务器

该方案成功打通了Python端以fps近30的速度real-time传输单张Image至RTMP服务器的过程。为例如图像识别,图像处理等应用的前后端图像数据传输提供了解决方案。
摘要由CSDN通过智能技术生成

使用ffmpeg在Python实现OpenCV Image推流至RTMP服务器

在实现Real time的机器学习推理过程中,如果想要将Image实时显示到前端,需要和后端的推理服务进行图像传输的通信。在这里提供一种自测可用的方案。

源码

源码在此文档的基础上有所修改,更多细节请参考原文

代码修改

command = ['ffmpeg',
        '-y',
        '-f', 'rawvideo',
        '-vcodec','rawvideo',
        
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
要在C++中实现OpencvFFmpeg推流RTMP,需要使用FFmpeg的API和Opencv的VideoCapture类。 首先,需要初始化FFmpeg的网络库和注册所有的组件。可以使用如下代码: ``` av_register_all(); avformat_network_init(); ``` 然后,需要打开视频文件或者摄像头,并将其转换为FFmpeg的AVFormatContext结构体。可以使用如下代码: ``` AVFormatContext *pFormatContext = nullptr; avformat_open_input(&pFormatContext, "video.mp4", nullptr, nullptr); if (avformat_find_stream_info(pFormatContext, nullptr) < 0) { // 处理错误 } ``` 接下来,需要找到视频流和音频流的索引,并打开它们。可以使用如下代码: ``` int videoStreamIndex = -1; int audioStreamIndex = -1; for (int i = 0; i < pFormatContext->nb_streams; i++) { AVStream *pStream = pFormatContext->streams[i]; if (pStream->codec->codec_type == AVMEDIA_TYPE_VIDEO) { videoStreamIndex = i; } if (pStream->codec->codec_type == AVMEDIA_TYPE_AUDIO) { audioStreamIndex = i; } } AVCodecContext *pVideoCodecContext = pFormatContext->streams[videoStreamIndex]->codec; AVCodecContext *pAudioCodecContext = pFormatContext->streams[audioStreamIndex]->codec; AVCodec *pVideoCodec = avcodec_find_decoder(pVideoCodecContext->codec_id); AVCodec *pAudioCodec = avcodec_find_decoder(pAudioCodecContext->codec_id); if (pVideoCodec == nullptr || pAudioCodec == nullptr) { // 处理错误 } if (avcodec_open2(pVideoCodecContext, pVideoCodec, nullptr) < 0 || avcodec_open2(pAudioCodecContext, pAudioCodec, nullptr) < 0) { // 处理错误 } ``` 然后,需要创建一个FFmpeg的AVOutputFormat结构体来表示输出格式,并打开输出文件。可以使用如下代码: ``` AVOutputFormat *pOutputFormat = av_guess_format("flv", nullptr, nullptr); AVFormatContext *pOutputContext = nullptr; if (avformat_alloc_output_context2(&pOutputContext, pOutputFormat, nullptr, "rtmp://127.0.0.1/live/test") < 0) { // 处理错误 } if (avio_open(&pOutputContext->pb, "rtmp://127.0.0.1/live/test", AVIO_FLAG_WRITE) < 0) { // 处理错误 } ``` 接着,需要创建视频流和音频流的AVStream结构体,并设置它们的编码器和参数。可以使用如下代码: ``` AVStream *pVideoStream = avformat_new_stream(pOutputContext, pVideoCodec); AVStream *pAudioStream = avformat_new_stream(pOutputContext, pAudioCodec); // 设置编码器参数 avcodec_parameters_from_context(pVideoStream->codecpar, pVideoCodecContext); avcodec_parameters_from_context(pAudioStream->codecpar, pAudioCodecContext); ``` 最后,需要循环读取视频帧和音频帧,并将它们写入输出流。可以使用如下代码: ``` AVPacket packet; av_init_packet(&packet); while (av_read_frame(pFormatContext, &packet) >= 0) { if (packet.stream_index == videoStreamIndex) { // 处理视频帧 av_interleaved_write_frame(pOutputContext, &packet); } if (packet.stream_index == audioStreamIndex) { // 处理音频帧 av_interleaved_write_frame(pOutputContext, &packet); } av_packet_unref(&packet); } ``` 完整的代码示例可以参考[这里](https://github.com/charles-wangkai/opencv-ffmpeg-rtmp)。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值