我也来晒晒图像二值化程序

论坛>计算机科学>软件综合

文号

t17979

阅读

4358

我也来晒晒图像二值化程序

小俊

2009/07/28

软件综合

IP:香港

用的是破93的RGB2L，不过偶是用CUDA写的：

#define THREAD_N    256
#define THRESHOLD    127

__global__ static void binarizeKernel(unsigned char *in, unsigned char *out)
{
    const unsigned long offset = (blockIdx.x * THREAD_N + threadIdx.x);
    unsigned char b = in[offset * 3];
    unsigned char g = in[offset * 3 + 1];
    unsigned char r = in[offset * 3 + 2];

    unsigned char mi = __min(r, __min(g, b));
    unsigned char ma = __max(r, __max(g, b));

    if(((unsigned short)ma + (unsigned short)mi) > THRESHOLD * 2)
        out[offset] = 255;
    else
        out[offset] = 0;
}

extern "C" float binarizeCall(unsigned char *in, unsigned char *out, int w, int h)
{
    unsigned char *device_src = 0;
    unsigned char *device_dest = 0;

    cudaMalloc((void **)&device_src, w * h * 3);
    cudaMalloc((void **)&device_dest, w * h);

    unsigned int timer = 0;
    cutCreateTimer(&timer);

    cudaMemcpy(device_src, in, w * h * 3, cudaMemcpyHostToDevice);

    cutStartTimer(timer);

    binarizeKernel<<<w * h / THREAD_N, THREAD_N>>>(device_src, device_dest);
    cudaThreadSynchronize();

    cutStopTimer(timer);

    cudaMemcpy(out, device_dest, w * h, cudaMemcpyDeviceToHost);

    float ms = cutGetTimerValue(timer);
    cutDeleteTimer(timer);

    cudaFree(device_dest);
    cudaFree(device_src);

    return ms;
}

来自：

计算机科学/软件综合

全部只看作者

小俊

作者

17年6个月前IP:未同步

132796

1楼

初学CUDA练练手用的。。。

找了张3264×2448的BMP，在Quadro FX5800（GT200核心）上跑花了1.73ms，在Geforce 9400M上跑花了74.2ms。

CPU实现的方法（用下面的代码VS2005编译），同样大小的图像在QX6600（2.4GHz）上花了42.6ms。

    for(y = 0; y < h; y ++)
        for(x = 0; x < w; x ++)
        {
            unsigned long offset = y * WIDTH + x;

            b = in[offset * 3];
            g = in[offset * 3 + 1];
            r = in[offset * 3 + 2];
            mi = min(r, min(g, b));
            ma = max(r, max(g, b));

            if(((unsigned short)ma + (unsigned short)mi) > THRESHOLD * 2)
                out[offset] = 255;
            else
                out[offset] = 0;
        }

引用

93°

17年6个月前IP:未同步

132798

2楼

喷血 CUDA就是变态

引用

93°

17年6个月前IP:未同步

132800

3楼

我去写个内存法……

引用

小俊

作者

17年6个月前IP:未同步

132802

4楼

引用第2楼破93于2009-07-28 16:57发表的 :
喷血 CUDA就是变态

CUDA是BT，可也得卡好，至少得卡王级别的，像9400M这种主流低端配置就没啥意思了。

另外很囧的一点就是，把原始数据通过PCI-E copy到显存以及把结果从显存copy回来一共花了将近20ms。

引用

全部只看作者

笔记

{{fromNow(n.toc)}} {{n.status === noteStatus.disabled ? "已屏蔽" : ""}} {{n.status === noteStatus.unknown ? "正在审核" : ""}} {{n.status === noteStatus.deleted ? '已删除' : ''}}

编辑
删除
{{n.status === 'disabled' ? "解除屏蔽" : "屏蔽" }}

我也是有底线的