parent
c2e67dd6ee
commit
34242ab741
@ -0,0 +1,100 @@
|
||||
#include <stdio.h>
|
||||
#include <ctime>
|
||||
#include <stdlib.h>
|
||||
#include<arm_neon.h>
|
||||
|
||||
#define H 5
|
||||
#define W 500
|
||||
void advanced_applySeparableGaussianBlur(float src[H][W], float dst[H][W], int h, int w, float kx[3], float ky[3])
|
||||
{ float buf[H][W-1]={0};
|
||||
float32x4_t kx_vec=vld1q_f32(kx);
|
||||
float32x4_t ky_vec=vld1q_f32(ky);
|
||||
for(int k = 0; k< 2; ++k)
|
||||
{
|
||||
for (int j = 1; j < w - 1; j+=4)
|
||||
{
|
||||
float32x4_t left =vld1q_f32(&src[k][j-1]);
|
||||
float32x4_t mid =vld1q_f32(&src[k][j]);
|
||||
float32x4_t right =vld1q_f32(&src[k][j+1]);
|
||||
|
||||
float32x4_t result =vmulq_lane_f32(left,vget_low_f32(kx_vec),0);
|
||||
result=vmlaq_lane_f32(result,mid,vget_low_f32(kx_vec),1);
|
||||
result=vmlaq_lane_f32(result,right,vget_high_f32(kx_vec),0);
|
||||
|
||||
vst1q_f32(&buf[k][j],result);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
//开始进行可分离卷积
|
||||
for (int i = 1; i < h - 1; ++i)
|
||||
{
|
||||
//进行行间的卷积得到最终像素值
|
||||
for (int j = 1; j < w - 1; j+=4)
|
||||
{
|
||||
float32x4_t left =vld1q_f32(&src[i+1][j-1]);
|
||||
float32x4_t mid =vld1q_f32(&src[i+1][j]);
|
||||
float32x4_t right =vld1q_f32(&src[i+1][j+1]);
|
||||
|
||||
float32x4_t result =vmulq_lane_f32(left,vget_low_f32(kx_vec),0);
|
||||
result=vmlaq_lane_f32(result,mid,vget_low_f32(kx_vec),1);
|
||||
result=vmlaq_lane_f32(result,right,vget_high_f32(kx_vec),0);
|
||||
|
||||
vst1q_f32(&buf[IDX(i + 1)][j],result);
|
||||
|
||||
|
||||
float32x4_t left1 =vld1q_f32(&buf[IDX(i - 1)][j]);
|
||||
float32x4_t mid1 =vld1q_f32(&buf[IDX(i )][j]);
|
||||
float32x4_t right1 =vld1q_f32(&buf[IDX(i + 1)][j]);
|
||||
|
||||
float32x4_t result1 =vmulq_lane_f32(left1,vget_low_f32(kx_vec),0);
|
||||
result1=vmlaq_lane_f32(result1,mid1,vget_low_f32(kx_vec),1);
|
||||
result1=vmlaq_lane_f32(result1,right1,vget_high_f32(kx_vec),0);
|
||||
|
||||
vst1q_f32(&dst[i][j],result1);
|
||||
}
|
||||
}
|
||||
}
|
||||
int main()
|
||||
{
|
||||
float inputImage[H][W]={0};
|
||||
float dst[H][W]={0};
|
||||
float a[H][W]={0};
|
||||
|
||||
for(int i=0;i<H;i++)
|
||||
{
|
||||
for(int j=0;j<W;j++)
|
||||
{
|
||||
inputImage[i][j]=W*i+j+1;
|
||||
dst[i][j]=W*i+j+1;
|
||||
a[i][j]=W*i+j+1;
|
||||
}
|
||||
}
|
||||
|
||||
float kx[3]={0.25,0.5,0.25};
|
||||
float ky[3]={0.25,0.5,0.25};
|
||||
|
||||
clock_t start = clock();
|
||||
advanced_applySeparableGaussianBlur(inputImage, dst, H, W, kx, ky);
|
||||
clock_t end = clock();
|
||||
double time_spent = double(end - start) / CLOCKS_PER_SEC;
|
||||
printf("运行时间:%lf秒\ndst矩阵结果为\n",time_spent);
|
||||
for(int i=0;i<H;i++)
|
||||
{
|
||||
for(int j=0;j<W;j++)
|
||||
{
|
||||
if(i==0||i==H-1||j==0||j==W-1)
|
||||
{
|
||||
printf("%.1f ",a[i][j]);
|
||||
}else
|
||||
{
|
||||
printf("%.1f ",dst[i][j]);
|
||||
|
||||
}
|
||||
}
|
||||
printf("\n");
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
Loading…
Reference in new issue