2

我有一个输入图像 A 和一个以 YUV420 格式存储的大小为 800x600 的结果图像 B,我需要将图像 A 缩放为 100x100 大小并在某个点(x = 100,y = 100)将其放入结果图像 B 中。为了减少内存和 CPU 使用率,我将 swscale 结果直接放入最终的 B 图像中。

这是一个代码片段(非常简单):

//here we a creating sws context for scaling into 100x100
sws_ctx = sws_getCachedContext(sws_ctx, frame.hdr.width, frame.hdr.height, AV_PIX_FMT_YUV420P,
                               100, 100, AV_PIX_FMT_YUV420P, SWS_BILINEAR, nullptr, nullptr, nullptr);

接下来我们创建相应的切片和步幅来描述图像 A

    int src_y_plane_sz = frame.hdr.width * frame.hdr.height;
    int src_uv_plane_sz = src_y_plane_sz / 2;
    std::int32_t src_stride[] = {
        frame.hdr.width,
        frame.hdr.width / 2,
        frame.hdr.width / 2,
        0};

    const uint8_t* const src_slice[] = {
        &frame.raw_frame[0],
        &frame.raw_frame[0] + src_y_plane_sz,
        &frame.raw_frame[0] + src_y_plane_sz + src_uv_plane_sz,
        nullptr};

现在对目标 B 图像执行相同操作

    std::int32_t dst_stride[] = {
        current_frame.hdr.width,
        current_frame.hdr.width /2,
        current_frame.hdr.width /2,
        0
    };

    std::int32_t y_plane_sz = current_frame.hdr.width * current_frame.hdr.height;
    std::int32_t uv_plane_sz = y_plane_sz / 2;

    //calculate offset in slices for x=100, y=100 position
    std::int32_t y_offset = current_frame.hdr.width * 100 + 100;

    uint8_t* const dst_slice[] = {
        &current_frame.raw_frame[0] + y_offset,
        &current_frame.raw_frame[0] + y_plane_sz + y_offset / 2,
        &current_frame.raw_frame[0] + y_plane_sz + uv_plane_sz + y_offset / 2,
        nullptr};

毕竟 - 调用 swscale

    int ret = sws_scale(sws_ctx, src_slice, src_stride, 0, frame.hdr.height,
                        dst_slice, dst_stride);

使用测试序列后,我遇到了一些无效结果,并出现以下问题:

  1. Y 组件有一些填充线
  2. UV 组件放错了位置 - 它们比原始 Y 组件低一点。

人工制品

有人对 swscale 函数有同样的问题吗?我对这个 FFmpeg 库集合非常陌生,所以我愿意接受任何关于如何正确执行此任务的意见。

FFmpeg 版本使用 3.3

4

2 回答 2

1

YUV420格式将图像的宽度和高度都缩放两倍。也就是说,每个色平面比亮度平面小 4 倍:

int src_uv_plane_sz = src_y_plane_sz / 4;

另外我不确定计算的步幅值是否正确。通常步幅!=宽度。

于 2018-08-11T14:24:48.383 回答
1

感谢@VTT 指出可能的问题 - 我已将目标切片指针计算固定为以下内容:

    int dest_x = 200, dest_y = 70;

    //into 100x100 position
    std::int32_t y_offset = current_frame.hdr.width * dest_y + dest_x;
    std::int32_t u_offset = ( current_frame.hdr.width * dest_y )  / 4 + dest_x /2;
    std::int32_t v_offset = u_offset + y_plane_sz / 4;

    uint8_t* const dst_slice[] = {
        &current_frame.raw_frame[0] + y_offset,
        &current_frame.raw_frame[0] + y_plane_sz + u_offset,
        &current_frame.raw_frame[0] + y_plane_sz + v_offset,
        nullptr};

并且“线伪影”的第二个问题是通过使用缩放尺寸大小因子 8 来解决的。

为目标切片指针的正确位置计算的另一项补充 - 必须根据当前 Y 平面指向重新调整 y 坐标,因为每两个 Y 步幅只有一个 U 或 V 步幅。例如(参见adjusted_uv_y 变量):

std::int32_t adjusted_uv_y = dest_y % 2 == 0 ? dest_y : dest_y - 1;
std::int32_t y_offset = current_frame.hdr.width * dest_y + dest_x;
std::int32_t u_offset = ( current_frame.hdr.width * adjusted_uv_y )  / 4 + dest_x /2;
std::int32_t v_offset = u_offset + y_plane_sz / 4;
于 2018-08-13T14:53:00.770 回答