生成一个opencv rtmp推流程序

以下是一个使用 OpenCV 推流 RTMP 的示例程序。在运行程序之前，需要安装 OpenCV 和 librtmp 库。

#include <iostream>
#include <opencv2/opencv.hpp>
#include <librtmp/rtmp.h>

using namespace std;
using namespace cv;

int main() {
    // 打开 RTMP 流
    RTMP* rtmp = RTMP_Alloc();
    RTMP_Init(rtmp);
    RTMP_SetupURL(rtmp, "rtmp://your_rtmp_server_url/your_stream_key");
    RTMP_EnableWrite(rtmp);

    if (!RTMP_Connect(rtmp, NULL) || !RTMP_ConnectStream(rtmp, 0)) {
        cerr << "Failed to connect RTMP stream" << endl;
        return -1;
    }

    // 打开摄像头
    VideoCapture cap(0);
    if (!cap.isOpened()) {
        cerr << "Failed to open camera" << endl;
        return -1;
    }

    namedWindow("RTMP Stream", WINDOW_AUTOSIZE);

    // 推流主循环
    while (true) {
        Mat frame;
        cap.read(frame);
        if (frame.empty()) {
            cerr << "Failed to capture frame" << endl;
            break;
        }

        imshow("RTMP Stream", frame);
        waitKey(30);

        // 将帧转换成 RTMP 数据并推送
        RTMPPacket packet;
        RTMPPacket_Alloc(&packet, frame.total() * 3);
        packet.m_packetType = RTMP_PACKET_TYPE_VIDEO;
        packet.m_nBodySize = frame.total() * 3;
        packet.m_nTimeStamp = RTMP_GetTime();
        packet.m_nChannel = 0x04;
        packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
        packet.m_nInfoField2 = rtmp->m_stream_id;

        memcpy(packet.m_body, frame.data, frame.total() * 3);
        if (RTMP_SendPacket(rtmp, &packet, TRUE) == -1) {
            cerr << "Failed to send RTMP packet" << endl;
            break;
        }
        RTMPPacket_Free(&packet);
    }

    // 关闭摄像头和 RTMP 流
    cap.release();
    RTMP_Close(rtmp);
    RTMP_Free(rtmp);

    return 0;
}

你需要将 `"rtmp://your_rtmp_server_url/your_stream_key"` 替换为你自己的 RTMP 流地址。此外，还需要在程序中添加 librtmp 库的链接选项，例如：

g++ -o rtmp_push rtmp_push.cpp -lopencv_core -lopencv_highgui -lrtmp

运行程序后，即可从摄像头捕获视频并推送到 RTMP 流中。