如何使用Neon将8位YUV420图像转换为RGB？

问题描述 #

我是Neon新手，想要编写代码通过Neon实现YUV420到RGB的转换。像素为8位深度，需转换为int32_t或float类型（但仍限制在0-255范围内）。不过我未找到对应的Neon指令，另外似乎没有支持int16与float这类不同类型相乘的指令，是否必须通过C代码进行数据类型转换以适配Neon的要求？

以下是我用C代码实现的转换算法（参数范围为[0, 255]）：

inline std::uint8_t clamp(std::int32_t n) {
    n &= -(n >= 0);
    return n | ((255 - n) >> 31);
}

inline void yuv2rgb_i32(std::int32_t y, std::int32_t u, std::int32_t v, std::int32_t &r, std::int32_t &g, std::int32_t &b) {
    y -= 16;
    u -= 128;
    v -= 128;

    r = clamp((std::int32_t)(1.164 * y + 1.596 * v));
    g = clamp((std::int32_t)(1.164 * y - 0.392 * u - 0.813 * v));
    b = clamp((std::int32_t)(1.164 * y + 2.017 * u));
}

核心解答 #

1. 无需单独用C做类型转换，Neon自带类型转换指令 #

Neon原生支持不同宽度、类型的向量转换，常用指令包括：

• vmovl_u8：将uint8向量零扩展为uint16向量
• vmovl_s16：将int16向量符号扩展为int32向量
• vcvtq_s32_f32：将float32向量转换为int32向量
  这些指令可直接在Neon内联汇编或C语言Neon intrinsics中使用，不需要单独用C代码处理类型转换。

2. 优先用定点运算替代浮点，避免跨类型相乘问题 #

Neon处理定点运算的效率远高于浮点，还能规避整数与浮点相乘的适配问题。可以将浮点系数转换为定点格式（比如Q10或Q31），用整数运算完成计算：

• 例如：1.164 ≈ 1.164×2¹⁰ = 1192（Q10格式），1.596≈1635，0.392≈401，0.813≈832，2.017≈2065
• 计算时用int32向量运算，最后右移对应位数还原结果，全程用整数类型，无需浮点参与。

3. Neon实现YUV420转RGB的关键步骤（以内联intrinsics为例） #

步骤1：加载YUV数据并扩展类型

YUV420中U/V是半采样，需先对U/V做插值（示例用简单复制插值，实际推荐双线性插值），再将8位数据扩展为int32向量：

// 加载8位Y向量并扩展为int32
int32x4_t y_vec = vmovl_u8(vld1_u8(y_ptr));
// 加载8位U向量，插值后扩展为int32
int32x4_t u_vec = vmovl_u8(vdup_lane_u8(vld1_u8(u_ptr), 0));
// 加载8位V向量，插值后扩展为int32
int32x4_t v_vec = vmovl_u8(vdup_lane_u8(vld1_u8(v_ptr), 0));

步骤2：执行偏移调整

对应C代码中y-16、u/v-128的操作：

y_vec = vsubq_s32(y_vec, vdupq_n_s32(16));
u_vec = vsubq_s32(u_vec, vdupq_n_s32(128));
v_vec = vsubq_s32(v_vec, vdupq_n_s32(128));

步骤3：用定点系数计算RGB

以Q10定点格式为例，计算后右移10位还原：

// 预定义Q10格式的定点系数
const int32x4_t coeff_1164 = vdupq_n_s32(1192);
const int32x4_t coeff_1596 = vdupq_n_s32(1635);
const int32x4_t coeff_0392 = vdupq_n_s32(401);
const int32x4_t coeff_0813 = vdupq_n_s32(832);
const int32x4_t coeff_2017 = vdupq_n_s32(2065);

// 计算R = 1.164Y + 1.596V
int32x4_t r_vec = vmlaq_s32(vmulq_s32(y_vec, coeff_1164), v_vec, coeff_1596);
r_vec = vshrq_n_s32(r_vec, 10);

// 计算G = 1.164Y - 0.392U - 0.813V
int32x4_t g_vec = vmlaq_s32(vmulq_s32(y_vec, coeff_1164), u_vec, vnegq_s32(coeff_0392));
g_vec = vmlaq_s32(g_vec, v_vec, vnegq_s32(coeff_0813));
g_vec = vshrq_n_s32(g_vec, 10);

// 计算B = 1.164Y + 2.017U
int32x4_t b_vec = vmlaq_s32(vmulq_s32(y_vec, coeff_1164), u_vec, coeff_2017);
b_vec = vshrq_n_s32(b_vec, 10);

步骤4：Clamp到0-255并转换为uint8

Neon有专门的clamp指令，无需调用C的clamp函数：

// 将int32向量钳位到0-255，转换为uint8向量
uint8x4_t r_u8 = vqmovun_s32(r_vec);
uint8x4_t g_u8 = vqmovun_s32(g_vec);
uint8x4_t b_u8 = vqmovun_s32(b_vec);
// 打包为RGB888格式的向量
uint8x12_t rgb_vec = vcombine_u8(vcombine_u8(r_u8, g_u8), b_u8);
// 存储到内存
vst1q_u8(rgb_ptr, rgb_vec);

补充说明 #

• 如果坚持使用浮点运算，Neon也支持float32向量运算：先将8位像素转换为float32向量，用浮点系数计算后再转成int32并钳位，但效率远低于定点运算。
• YUV420的U/V插值是性能关键，可利用Neon的vpadd、vzip等指令高效实现双线性插值，避免逐像素处理。

内容的提问来源于stack exchange，提问作者zuguorui