0

我正在为 iOS 开发一些图像处理应用程序,阈值确实是一个巨大的瓶颈。所以我正在尝试使用 NEON 对其进行优化。这是函数的C版本。有没有办法用 NEON 重写它(不幸的是我完全没有这方面的经验)?

static void thresh_8u( const Image& _src, Image& _dst, uchar thresh, uchar maxval, int type ) {
    int i, j;
    uchar tab[256];
    Size roi = _src.size();
    roi.width *= _src.channels();

    memset(&tab[0], 0, thresh);
    memset(&tab[thresh], maxval, 256-thresh);

    for( i = 0; i < roi.height; i++ ) {
        const uchar* src = (const uchar*)(_src.data + _src.step*i);
        uchar* dst = (uchar*)(_dst.data + _dst.step*i);
        j = 0;

        for(; j <= roi.width; ++j) {
            dst[j] = tab[src[j]];
        }
    }
}
4

1 回答 1

7

如果你能确保你的行总是 16 字节宽的倍数,这实际上很容易,因为编译器 (clang) 具有表示 NEON 向量寄存器的特殊类型,并且知道如何将普通的 C 运算符应用于它们。这是我的小测试功能:

#ifdef __ARM_NEON__

#include <arm_neon.h>

void computeThreshold(void *input, void *output, int count, uint8_t threshold, uint8_t highValue) {
    uint8x16_t thresholdVector = vdupq_n_u8(threshold);
    uint8x16_t highValueVector = vdupq_n_u8(highValue);
    uint8x16_t *__restrict inputVector = (uint8x16_t *)input;
    uint8x16_t *__restrict outputVector = (uint8x16_t *)output;
    for ( ; count > 0; count -= 16, ++inputVector, ++outputVector) {
        *outputVector = (*inputVector > thresholdVector) & highValueVector;
    }
}

#endif

这一次对 16 个字节进行操作。Auint8x16_t是一个向量寄存器,包含 16 个 8 位无符号整数。返回一个填充了其参数的 16 个副本的向量vdupq_n_u8uint8x16_t

>运算符应用于两个uint8x16_t值,在 8 位无符号整数对之间进行 16 次比较。如果左输入大于右输入,则返回 0xff(与普通 C 不同>,后者仅返回 0x01)。当左输入小于或等于右输入时,返回 0。(编译成 VCGT.U8 指令。)

&运算符应用于两个uint8x16_t值,计算 128 对位的布尔 AND。

循环在发布版本中编译为此:

0x6e668:  vldmia r2, {d4, d5}
0x6e66c:  subs   r0, #16
0x6e66e:  vcgt.u8 q10, q10, q8
0x6e672:  adds   r2, #16
0x6e674:  cmp    r0, #0
0x6e676:  vand   q10, q10, q9
0x6e67a:  vstmia r1, {d4, d5}
0x6e67e:  add.w  r1, r1, #16
0x6e682:  bgt    0x6e668
于 2012-07-20T22:54:22.073 回答