二维卷积c代码

二维卷积c代码

二维信号的卷积原理请参考另外一篇文章：http://blog.csdn.net/carson2005/article/details/43702241

这里直接给出参考代码：

[cpp] view plaincopy





void Conv2(int** filter, int** arr, int** res, int filterW, int filterH, int arrW, int arrH)

{

int temp;

for (int i=0; i<filterH+arrH-1; i++)

{

for (int j=0; j<filterW+arrW-1; j++)

{

temp = 0;

for (int m=0; m<filterH; m++)

{

for (int n=0; n<filterW; n++)

{

if ((i-m)>=0 && (i-m)<arrH && (j-n)>=0 && (j-n)<arrW)

{

temp += filter[m]
*arr[i-m][j-n];

}

}

}

res[i][j] = temp;

}

}

}

void Conv2Test()

{

int filterW = 3;

int filterH = 3;

int arrW = 5;

int arrH = 5;

int resW = filterW + arrW - 1;

int resH = filterH + arrH - 1;

int** pFilter = new int*[filterH];

for (int i=0; i<filterH; i++)

{

pFilter[i] = new int[filterW];

}

int** arr = new int*[arrH];

for (int i=0; i<arrH; i++)

{

arr[i] = new int[arrW];

}

int** res = new int*[resH];

for (int i=0; i<resH; i++)

{

res[i] = new int[resW];

}

pFilter[0][0] = 1;

pFilter[0][1] = 3;

pFilter[0][2] = 1;

pFilter[1][0] = 0;

pFilter[1][1] = 5;

pFilter[1][2] = 0;

pFilter[2][0] = 2;

pFilter[2][1] = 1;

pFilter[2][2] = 2;

arr[0][0] = 17;

arr[0][1] = 24;

arr[0][2] = 1;

arr[0][3] = 8;

arr[0][4] = 15;

arr[1][0] = 23;

arr[1][1] = 5;

arr[1][2] = 7;

arr[1][3] = 14;

arr[1][4] = 16;

arr[2][0] = 4;

arr[2][1] = 6;

arr[2][2] = 13;

arr[2][3] = 20;

arr[2][4] = 22;

arr[3][0] = 10;

arr[3][1] = 12;

arr[3][2] = 19;

arr[3][3] = 21;

arr[3][4] = 3;

arr[4][0] = 11;

arr[4][1] = 18;

arr[4][2] = 25;

arr[4][3] = 2;

arr[4][4] = 9;

printf("pFilter: \n");

for (int i=0; i<filterH; i++)

{

for (int j=0; j<filterW; j++)

{

printf("%d ", pFilter[i][j]);

}

printf("\n");

}

printf("*********************************************** \n");

printf("arr: \n");

for (int i=0; i<arrH; i++)

{

for (int j=0; j<arrW; j++)

{

printf("%d ", arr[i][j]);

}

printf("\n");

}

printf("*********************************************** \n");

Conv2(pFilter, arr, res, filterW, filterH, arrW, arrH);

printf("res: \n");

for (int i=0; i<resH; i++)

{

for (int j=0; j<resW; j++)

{

printf("%d ", res[i][j]);

}

printf("\n");

}

printf("*********************************************** \n");

}