• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    公众号

Matlab 快速多通道积分图计算函数

原作者: [db:作者] 来自: [db:来源] 收藏 邀请

所谓快速多通道积分图计算,其实就是 cumsum2D。
我写了一个比较快的版本(比 VLFeat 的快),用 mex 编译一下就能用了。

代码

#include <string.h>
#include <mex.h>
#include <stdio.h>
#include <stdint.h>

// compute integral image
template <typename T>
void integral(const T* in, T* out, mwSize h, mwSize w, mwSize ch)
{
    mwSize iw = w + 1;
    mwSize ih = h + 1;
    out += 1;
    memset(out, 0, ih * iw * ch * sizeof(T));
    for (mwSize c = 0; c < ch; ++c)
    {
        // offset one row and one column
        out += ih;
        // from first to last column
        for (mwSize x = 0; x < w; ++x, in += h, out += ih)
        {
            T s = 0.0f;
            // from first to last column
            for (mwSize y = 0; y < h; ++y)
            {
                s += in[y];
                out[y] = out[y - ih] + s;
            }
        }
    }
}

// itg = integralChannels(channels)
void mexFunction (int nlhs, mxArray* plhs[], int nrhs, const mxArray* prhs[])
{
    mxClassID classId = mxGetClassID(prhs[0]);
    mwSize nDims = mxGetNumberOfDimensions(prhs[0]);
    const mwSize *dims = mxGetDimensions(prhs[0]);
    mwSize *iDims = new mwSize[nDims];
    
    if (nDims != 3 && nDims != 2)
    {
        mexErrMsgTxt("integralChannels() can only integrate single or multichannel image.");
    }
    mwSize h = dims[0];
    mwSize w = dims[1];
    mwSize ch = 1;
        
    iDims[0] = h + 1;
    iDims[1] = w + 1;
    if (nDims == 3)
    {
        ch = dims[2];
        iDims[2] = ch;
    }
    plhs[0] = mxCreateNumericArray(nDims, iDims, classId, mxREAL);
    delete [] iDims;
    switch (classId)
    {
        case mxSINGLE_CLASS:
            integral<float>((float *)mxGetData(prhs[0]), 
                    (float *)mxGetData(plhs[0]), h, w, ch);
            break;
        case mxDOUBLE_CLASS:
            integral<double>((double *)mxGetData(prhs[0]), 
                    (double *)mxGetData(plhs[0]), h, w, ch);
            break;
        case mxUINT32_CLASS:
            integral<uint32_t>((uint32_t *)mxGetData(prhs[0]), 
                    (uint32_t *)mxGetData(plhs[0]), h, w, ch);
            break;
        case mxINT32_CLASS:
            integral<int32_t>((int32_t *)mxGetData(prhs[0]), 
                    (int32_t *)mxGetData(plhs[0]), h, w, ch);
            break;
        default:
            mexErrMsgTxt("Illegal Class ID.");
    }
    return;
}

性能测试

# Elapsed time is 0.062636 seconds.
tic; integralChannels(ones(1000, 1000, 9)); toc;
# Elapsed time is 0.085479 seconds.
tic; vl_imintegral(ones(1000, 1000, 9)); toc;
# Elapsed time is 0.128849 seconds.
tic; cumsum2D(ones(1000, 1000, 9)); toc;

鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap