微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

OpenCL 生成 YUV420P 黑色像素

如何解决OpenCL 生成 YUV420P 黑色像素

我正在开发一个 FFmpeg OpenCL 过滤器,用于在 Google EAC 投影文件中转换 gopro Max .360 文件。 为此,我需要将 2 个输入流“堆叠”到两倍高度的输出。 它以实时速度运行得非常好。 (工作代码在注释中)

为了更进一步,我需要用一些特定的颜色替换一些像素。 我写了一些宏来进行 RGB->YUV 转换。但我只得到一些绿色或粉红色像素(只有灰色是可以的)。 这是我的测试代码(注释中有堆叠)

#define Y(R,G,B) 0.299 * R + 0.587 * G + 0.114 * B
#define U(R,B) -0.147 * R - 0.289 * G + 0.436 * B
#define V(R,B) 0.615 * R - 0.515 * G - 0.100 * B
#define YUV(R,B) (float4)(Y(R,B),U(R,V(R,0)

__kernel void gopromax_stack(__write_only image2d_t dst,__read_only  image2d_t gopromax_front,__read_only  image2d_t gopromax_rear)
{
    const sampler_t sampler = (CLK_norMALIZED_COORDS_FALSE |
                               CLK_FILTER_NEAREST);
    
    float4 val;
    int2 dst_size = get_image_dim(dst);
    int2 loc = (int2)(get_global_id(0),get_global_id(1));
    int split_loc = dst_size.y/2;

        if (loc.y < split_loc)
        {
          // working code for stacking
          //  val = read_imagef(gopromax_front,sampler,(int2)(loc.x,loc.y));

          // testing to put grey (working)
            val = YUV(0.5f,0.5f,0.5f);
        }
        else
        {
          // working code for stacking
          //  val = read_imagef(gopromax_rear,loc.y-split_loc));

          // testing to put black (gives green !)
            val = YUV(0,0);
        }

    if ((loc.x<dst_size.x) && (loc.y<dst_size.y))
        write_imagef(dst,loc,val);
}

我尝试了很多认为我无法成功生成黑色或灰色以外的任何东西的方法。 我做错了什么? 我假设我的像素是 YUV,因为我在过滤器中指定了 yuv420p 作为格式:

-filter_complex '[0:0]format=yuv420p,hwupload[a],[0:4]format=yuv420p,hwupload[b],[a][b]gopromax_opencl,hwdownload,format=yuv420p'

源流位于 hevc / nv12。

感谢大家的帮助。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。