1

我正在尝试编写一个需要为多个全分辨率图像计算高斯和拉普拉斯金字塔的 android 应用程序，我用 NDK 在 C++ 上编写了它，代码的最关键部分是将高斯滤波器应用于图像 abd 我正在应用这个滤波器与水平和垂直。

过滤器是 (0.0625, 0.25, 0.375, 0.25, 0.0625) 因为我正在处理整数，所以我正在计算 (1, 4, 6, 4, 1)/16

dst[index] = ( src[index-2] + src[index-1]*4 + src[index]*6+src[index+1]*4+src[index+2])/16;

我做了一些简单的优化，但是它的运行速度仍然比预期的慢，我想知道是否还有其他优化选项我遗漏了。

PS：我应该提一下，我曾尝试使用内联臂组件编写此过滤器部件，但是它的结果会慢 2 倍。

//horizontal  filter
for(unsigned y = 0; y < height;  y++) {
    for(unsigned x = 2; x < width-2;  x++) {
        int index = y*width+x;
            dst[index].r = (src[index-2].r+ src[index+2].r + (src[index-1].r + src[index+1].r)*4 + src[index].r*6)>>4;
            dst[index].g = (src[index-2].g+ src[index+2].g + (src[index-1].g + src[index+1].g)*4 + src[index].g*6)>>4;
            dst[index].b = (src[index-2].b+ src[index+2].b + (src[index-1].b + src[index+1].b)*4 + src[index].b*6)>>4;                
     }
}
//vertical filter
for(unsigned y = 2;  y < height-2;  y++) {
    for(unsigned x = 0;  x < width;  x++) {
        int index = y*width+x;
            dst[index].r = (src[index-2*width].r + src[index+2*width].r  + (src[index-width].r + src[index+width].r)*4 + src[index].r*6)>>4;
            dst[index].g = (src[index-2*width].g + src[index+2*width].g  + (src[index-width].g + src[index+width].g)*4 + src[index].g*6)>>4;
            dst[index].b = (src[index-2*width].b + src[index+2*width].b  + (src[index-width].b + src[index+width].b)*4 + src[index].b*6)>>4;
     }
}
4

3 回答 3

4

index乘法可以排除在内循环之外，因为乘法仅在更改时发生y： 

for (unsigned y ...
{
    int index = y * width;
    for (unsigned int x...

您可以通过在使用变量之前加载它们来获得一些速度。这将使处理器将它们加载到缓存中：

for (unsigned x = ...  
{  
    register YOUR_DATA_TYPE a, b, c, d, e;
    a = src[index - 2].r;
    b = src[index - 1].r;
    c = src[index + 0].r; // The " + 0" is to show a pattern.
    d = src[index + 1].r;
    e = src[index + 2].r;
    dest[index].r = (a + e + (b + d) * 4 + c * 6) >> 4;
    // ...

另一个技巧是“缓存” src 的值，以便每次只添加一个新值，因为 in 中的值src[index+2]最多可以使用 5 次。

所以这里是一个概念的例子： 

//horizontal  filter
for(unsigned y = 0; y < height;  y++)
{
    int index = y*width + 2;
    register YOUR_DATA_TYPE a, b, c, d, e;
    a = src[index - 2].r;
    b = src[index - 1].r;
    c = src[index + 0].r; // The " + 0" is to show a pattern.
    d = src[index + 1].r;
    e = src[index + 2].r;
    for(unsigned x = 2; x < width-2;  x++)
    {
        dest[index - 2 + x].r = (a + e + (b + d) * 4 + c * 6) >> 4;
        a = b;
        b = c;
        c = d;
        d = e;
        e = src[index + x].r;
于 2012-10-24T22:38:22.770 回答
2

我不确定您的编译器将如何优化所有这些，但我倾向于使用指针。假设您的结构是 3 个字节...您可以从正确位置的指针开始（源的过滤器边缘和目标的目标），然后使用常量数组偏移量移动它们。我还在外部循环中添加了一个可选的 OpenMP 指令，因为这也可以改进。

#pragma omp parallel for
for(unsigned y = 0; y < height;  y++) {
    const int rowindex = y * width;
    char * dpos = (char*)&dest[rowindex+2];
    char * spos = (char*)&src[rowindex];
    const char *end = (char*)&src[rowindex+width-2];

    for( ; spos != end;  spos++, dpos++) {
        *dpos = (spos[0] + spos[4] + ((spos[1] + src[3])<<2) + spos[2]*6) >> 4;
    }
}

垂直循环也是如此。

const int scanwidth = width * 3;
const int row1 = scanwidth;
const int row2 = row1+scanwidth;
const int row3 = row2+scanwidth;
const int row4 = row3+scanwidth;

#pragma omp parallel for
for(unsigned y = 2;  y < height-2;  y++) {
    const int rowindex = y * width;
    char * dpos = (char*)&dest[rowindex];
    char * spos = (char*)&src[rowindex-row2];
    const char *end = spos + scanwidth;

    for( ; spos != end;  spos++, dpos++) {
        *dpos = (spos[0] + spos[row4] + ((spos[row1] + src[row3])<<2) + spos[row2]*6) >> 4;
    }
}

无论如何，这就是我做卷积的方式。它稍微牺牲了可读性，而且我从未尝试过测量差异。我只是倾向于从一开始就这样写。看看这是否能让你加速。如果你有一台多核机器，OpenMP 肯定会，而指针的东西可能.

我喜欢关于将 SSE 用于这些操作的评论。

于 2012-10-24T22:31:22.297 回答
1

一些更明显的优化正在利用内核的对称性：

a=*src++;    b=*src++;    c=*src++;    d=*src++;    e=*src++; // init

LOOP (n/5) times:
z=(a+e)+(b+d)<<2+c*6;  *dst++=z>>4;  // then reuse the local variables
a=*src++;
z=(b+a)+(c+e)<<2+d*6;  *dst++=z>>4;  // registers have been read only once...
b=*src++;
z=(c+b)+(d+a)<<2+e*6;  *dst++=z>>4;
e=*src++;

第二件事是可以使用单个整数执行多次加法。当要过滤的值是无符号的时，可以在一个 32 位整数中拟合两个通道（或在 64 位整数中拟合 4 个通道）；这是穷人的SIMD。

a=  0x[0011][0034]  <-- split to two 
b=  0x[0031][008a]
----------------------
sum    0042  00b0
>>4    0004  200b0  <-- mask off
mask   00ff  00ff   
-------------------
       0004  000b   <-- result

（模拟 SIMD 显示一个加法，然后移位 4）

这是一个并行计算 3 个 rgb 操作的内核（在 64 位架构中很容易修改为 6 个 rgb 操作......）

#define MASK (255+(255<<10)+(255<<20))
#define KERNEL(a,b,c,d,e) { \
 a=((a+e+(c<<1))>>2) & MASK; a=(a+b+c+d)>>2 & MASK; *DATA++ = a; a=DATA[4]; }

void calc_5_rgbs(unsigned int *DATA)
{
   register unsigned int a = DATA[0], b=DATA[1], c=DATA[2], d=DATA[3], e=DATA[4];
   KERNEL(a,b,c,d,e);
   KERNEL(b,c,d,e,a);
   KERNEL(c,d,e,a,b);
   KERNEL(d,e,a,b,c);
   KERNEL(e,a,b,c,d);
}

在 ARM 和具有 16 个寄存器的 64 位 IA 上效果最佳……需要大量的汇编程序优化来克服 32 位 IA 中的寄存器短缺问题（例如，使用 ebp 作为 GPR）。正因为如此，它是一个就地算法......

每 8 位数据之间只有 2 个保护位，这足以得到与整数计算完全相同的结果。

顺便说一句：每字节遍历数组字节比通过 r,g,b 元素更快

 unsigned char *s=(unsigned char *) source_array; 
 unsigned char *d=(unsigned char *) dest_array; 
 for (j=0;j<3*N;j++) d[j]=(s[j]+s[j+16]+s[j+8]*6+s[j+4]*4+s[j+12]*4)>>4;
于 2012-10-25T11:17:25.117 回答