我提供了一種通過灰度圖像和底肥索貝爾濾波器去的碼:(索貝爾類似於你的鄰居(濾波器A,B,C,d)函數)
#define QUANTUM_TYPE short
__global__ void sobel_gpu(QUANTUM_TYPE *img_out, QUANTUM_TYPE *img_in, int WIDTH, int HEIGHT){
int x,y;
x=blockDim.x*blockIdx.x+threadIdx.x;
y=blockDim.y*blockIdx.y+threadIdx.y;
QUANTUM_TYPE LUp,LCnt,LDw,RUp,RCnt,RDw;
int pixel;
if(x<WIDTH && y<HEIGHT){
LUp = (x-1>=0 && y-1>=0)? img_in[(x-1)+(y-1)*WIDTH]:0;
LCnt= (x-1>=0)? img_in[(x-1)+y*WIDTH]:0;
LDw = (x-1>=0 && y+1<HEIGHT)? img_in[(x-1)+(y+1)*WIDTH]:0;
RUp = (x+1<WIDTH && y-1>=0)? img_in[(x+1)+(y-1)*WIDTH]:0;
RCnt= (x+1<WIDTH)? img_in[(x+1)+y*WIDTH]:0;
RDw = (x+1<WIDTH && y+1<HEIGHT)? img_in[(x+1)+(y+1)*WIDTH]:0;
pixel = -1*LUp + 1*RUp +
-2*LCnt + 2*RCnt +
-1*LDw + 1*RDw;
pixel=(pixel<0)?0:pixel;
pixel=(pixel>MAXRGB)?MAXRGB:pixel;
img_out[x+y*WIDTH]=pixel;
}
}
該代碼是工作用於全局內存並安全地處理邊界。我的完整代碼讀取一個BMP圖像並在其上應用過濾器,並將生成的BMP存儲回磁盤。它可在here(CPU和GPU實現集成,無論是Linux和Windows)。
你可以用一點工作把它變成共享內存風格。首先,你應該決定你給每個區塊多少任務。然後把這個任務分解成多個共享內存sink/drain。 CUDA SDK中的Matrix Multiply示例爲您提供了一個完美的想法。
對於圖像處理應用程序,共享內存使用情況有點複雜。我會建議你使用CUDA紋理。紋理讀取被緩存,處理邊界情況非常簡單。紋理最適合於社區操作。 – sgarizvi
哪種計算能力具有您的圖形設備? – pQB