NV12格式转RGB的CUDA实现
Posted 逍遥一度,恣情江湖
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了NV12格式转RGB的CUDA实现相关的知识,希望对你有一定的参考价值。
NV12格式是yuv420格式的一种,NV12格式的u,v排布顺序为交错排布,假如一幅图像尺寸为W*H,则先Y分量有W*H个,然后U分量和V分量交错排布,U分量和V分量各有W*H/4个,U,V加起来总数是Y分量的一半。
下面是CUDA实现的NV12格式到BGR格式的转换代码。StepY,StepUV分别为ffmpeg解码出的源数据中的Y分量一行的宽度和UV分量一行的宽度,比实际的图像宽度要大。
__global__ void YCrCb2RGBConver(uchar *pYdata, uchar *pUVdata,int stepY, int stepUV, uchar *pImgData, int width, int height, int channels) { const int tidx = blockIdx.x * blockDim.x + threadIdx.x; const int tidy = blockIdx.y * blockDim.y + threadIdx.y; if (tidx < width && tidy < height) { int indexY, indexU, indexV; uchar Y, U, V; indexY = tidy * stepY + tidx; Y = pYdata[indexY]; if (tidx % 2 == 0) { indexU = tidy / 2 * stepUV + tidx; indexV = tidy / 2 * stepUV + tidx + 1; U = pUVdata[indexU]; V = pUVdata[indexV]; } else if (tidx % 2 == 1) { indexV = tidy / 2 * stepUV + tidx; indexU = tidy / 2 * stepUV + tidx - 1; U = pUVdata[indexU]; V = pUVdata[indexV]; } pImgData[(tidy*width + tidx) * channels + 2] = uchar (Y + 1.402 * (V - 128)); pImgData[(tidy*width + tidx) * channels + 1] = uchar (Y - 0.34413 * (U - 128) - 0.71414*(V - 128)); pImgData[(tidy*width + tidx) * channels + 0] = uchar (Y + 1.772*(U - 128)); } }
以上是关于NV12格式转RGB的CUDA实现的主要内容,如果未能解决你的问题,请参考以下文章
NV12 到 RGB32 与 Microsoft Media Foundation
DirectX:以 DXGI_FORMAT_NV12 格式从 ID3D11Texture2D 获取 RGB 数据的最佳方法?