繁体   English   中英

ffmpeg / c ++使用ffmpeg编码视频帧的其他信息

[英]ffmpeg/c++ Encode additional information of video frame with ffmpeg

我是ffmpeg和视频编码的新手,在此页面上查找了一些相关问题之后,我发现了这篇帖子,对于理解ffmpeg的概述过程非常有用。

但是,我的工作不仅需要使用Mat帧进行操作,还需要从视频中提取重要信息(提取边缘,边缘块的位置,每个边缘块的类型,块编号,运动矢量),还必须对其进行编码并将其发送给客户端。 我试图找到这部分的示例代码,但似乎以前没有人做过。

我的问题是如何将这些附加信息与视频帧一起编码,然后将两者都发送给客户端。 我了解了可以帮助无损压缩的霍夫曼编码,但是有可能在使用ffmpeg编码视频帧时使用霍夫曼编码对边缘和运动数据进行编码吗? 我正在使用udp协议进行实验。

我找不到有关此的任何信息。 我在ffmpeg中读了元数据和附带信息,但这不是我想要的。

我希望您能给我一些建议或指导,以便对此领域进行研究,以便我能够理解并尝试实施。 如果有这种情况的示例代码,非常感谢您的分享。

非常感谢。

下面是服务器端的编码器部分:

int encode(Mat& input_frame, EncodedCallback callback, void* userdata = nullptr) {
    AVPacket pkt;
    /* encode 1 second of video */
    av_init_packet(&pkt);
    pkt.data = NULL; // packet data will be allocated by the encoder
    pkt.size = 0;
    int size = 0;
    fflush(stdout);
    cvtFrame2AVFrameYUV420(input_frame, &frame);

    static int time; 
    frame->pts = time++;
    /* encode the image */
    ret = avcodec_send_frame(c, frame);
    if (ret < 0) {
        fprintf(stderr, "Error avcodec_send_frame\n");
        exit(1);
    }
    nbFramesEncoded++;
    ret = avcodec_receive_packet(c, &pkt);
    if (!isFirstFrameEmmited) {
        nbNeededFramesInBuffer++;

        printf("nbNeededFramesInBuffer: %d\n", nbNeededFramesInBuffer);
    }
    if (ret < 0) {
        if (ret == -EAGAIN) {
            //output is not available, we must send more input
        } else {
            fprintf(stderr, "Error avcodec_receive_packet %d\n", ret);
            exit(1);
        }
    } else {

        if (callback) {
            callback(pkt, userdata);
        }
        size = pkt.size + 4;
        av_packet_unref(&pkt);
    }

    return size;
}

下面是处理帧处理的代码(目前,我们检查并发送运动块给客户端)

void updateFrame(Mat& frame) {
    //Get all Streams ready
    bool isReady = true;
    if (!frameStreamer->encoder->isFirstFrameEmmited) {
        frameStreamer->sendFrame(frame);
        isReady = false;
    }
    for (int yidx = 0; yidx < gridSize.height; yidx++) {
        for (int xidx = 0; xidx < gridSize.width; xidx++) {
            StreamPtr& stream = streamGrid[yidx][xidx];
            if (!stream->encoder->isFirstFrameEmmited) {
                Mat block = frame(stream->irect);
                stream->sendFrame(block);
                isReady = false;
            }
        }
    }
    if (isReady == false) {
        return;
    }

    if (pGray.empty()) {

        frameStreamer->sendFrame(frame);
        frameStreamer->sendFrame(frame);

        cvtColor(frame, pGray, CV_BGR2GRAY);
        return;
    }

    //Motion Detection
    Mat gray;
    cvtColor(frame, gray, CV_BGR2GRAY);
    Mat diff;
    absdiff(gray, pGray, diff);
    threshold(diff, diff, NOISE_THRESHOLD, 255, CV_THRESH_BINARY);
    if (HEAT_IMAGE) {
        gray.copyTo(diff, diff);
        imshow("Gray", gray);
        threshold(diff, diff, HEAT_THRESH, 255, CV_THRESH_TOZERO);
    }
    if (USE_MORPH_NOISE) {
        Morph_Noise(diff);
    }

    Mat motionImg = Mat::zeros(frameSize, CV_8UC3);
    //Block Classification
    int nbModifiedBlocks = 0;
    for (int yidx = 0; yidx < gridSize.height; yidx++) {
        for (int xidx = 0; xidx < gridSize.width; xidx++) {
            Rect irect(xidx * blockSize.width, yidx * blockSize.height,
                    blockSize.width, blockSize.height);
            int blockDiff = sum(diff(irect))[0];
            if (blockDiff > BLOCK_THRESHOLD * 255) {
                this->blockCls.at<uchar>(yidx, xidx) = MODI_BLOCK;
                nbModifiedBlocks++;
            } else {
                this->blockCls.at<uchar>(yidx, xidx) = SKIP_BLOCK;
            }
        }
    }

    //Send
    if (nbModifiedBlocks > this->nbBlocksThresh) {
        nbSentBytes += this->frameStreamer->sendFrame(frame);
    } else {
        for (int yidx = 0; yidx < gridSize.height; yidx++) {
            for (int xidx = 0; xidx < gridSize.width; xidx++) {
                uchar cls = this->blockCls.at<uchar>(yidx, xidx);
                StreamPtr& stream = streamGrid[yidx][xidx];
                bool send = false;
                if (cls == MODI_BLOCK) {
                    if (DEBUG_NETWORK) {
                        printf("Normal (%d, %d): ", xidx, yidx);
                    }
                    send = true;
                    stream->encoder->nbFramesBuffered = stream->encoder->nbNeededFramesInBuffer;

                    rectangle(motionImg, stream->irect, Scalar(0, 0, 255), CV_FILLED);

                } else if (stream->encoder->nbFramesBuffered > 0) {
                    if (DEBUG_NETWORK) {
                        printf("Extra (%d, %d): ", xidx, yidx);
                    }
                    send = true;
                    stream->encoder->nbFramesBuffered--;
                    stream->encoder->nbFlushFrames++;
                    rectangle(motionImg, stream->irect, Scalar(0, 255, 0), CV_FILLED);
                }

                if (send) {
                    Mat block = frame(stream->irect);
                    nbSentBytes += stream->sendFrame(block);
                    gray(stream->irect).copyTo(pGray(stream->irect));
                }
            }
        }
    }

}

我可能不太了解您的要求,我想您要求将其他流添加到可能包含一些自定义数据的输出中。 这是添加data流的方法,请在add video stream之后添加此data add video stream

AVCodecParameters *par;
AVStream *st;

st = avformat_new_stream(ctx->oc, NULL); // ctx->oc is AVFormatContext *oc same you used where you add video stream
if (st == NULL)
{
    fprintf(stderr, "Error: avformat_new_stream() failed.\n");
    exit(EXIT_FAILURE);
}

par = st->codecpar;
par->codec_id = AV_CODEC_ID_TEXT; // raw UTF-8 text
par->codec_type = AVMEDIA_TYPE_DATA; // Opaque data information usually continuous

之后,您可以将“数据”放入pkt.data并在关联的视频帧输出之后使用av_interleaved_write_frame发送。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM