展会信息港展会大全

数字图像处理算法实现
来源:互联网   发布日期:2011-12-28 10:41:34   浏览:13112次  

导读:摘要: 关于空间域图像处理算法框架,直方图处理,空间域滤波器算法框架的编程心得,使用GDI+(C++) 一,图像文件的读取 初学数字图像处理时,图像文件的读取往往是一件麻烦的事情,我们要面对各种各样的图像文件格式,如果仅用C++的fstream库那就必须了解各种...
摘要: 关于空间域图像处理算法框架,直方图处理,空间域滤波器算法框架的编程心得,使用GDI+(C++)

一,图像文件的读取

    初学数字图像处理时,图像文件的读取往往是一件麻烦的事情,我们要面对各种各样的图像文件格式,如果仅用C++的fstream库那就必须了解各种图像编码格式,这对于初学图像处理是不太现实的,需要一个能帮助轻松读取各类图像文件的库。在Win32平台上GDI+(C++)是不错的选择,不光使用上相对于Win32 GDI要容易得多,而且也容易移植到.Net平台上的GDI+。

    Gdiplus::Bitmap类为我们提供了读取各类图像文件的接口,Bitmap::LockBits方法产生的BitmapData类也为我们提供了高速访问图像文件流的途径。这样我们就可以将精力集中于图像处理算法的实现,而不用关心各种图像编码。具体使用方式请参考MSDN中GDI+文[来源:GameRes.com]档中关于Bitmap类和BitmapData类的说明。另外GDI+仅在Windows XP/2003上获得直接支持,对于Windows 2000必须安装相关DLL,或者安装有Office 2003,Visual Studio 2003 .Net等软件。

二,空间域图像处理算法框架

(1) 在空间域图像处理中,对于一个图像我们往往需要对其逐个像素的进行处理,对每个像素的处理使用相同的算法(或者是图像中的某个矩形部分)。即,对于图像f(x,y),其中0≤x≤M,0≤y≤N,图像为M*N大小,使用算法algo,则f(x,y) = algo(f(x,y))。事先实现一个算法框架,然后再以函数指针或函数对象(functor,即实现operator()的对象)传入算法,可以减轻编程的工作量。

    如下代码便是一例:

#ifndef PROCESSALGO_H

#define PROCESSALGO_H

 

#include <windows.h>

#include <Gdiplus.h>

 

 

namespace nsimgtk

{

         template <typename pixelType, Gdiplus::PixelFormat pixelFormat, class Processor>

    bool ProcessPixelsOneByOne(Gdiplus::Bitmap* const p_bitmap, Processor processor, unsigned int x, unsigned int y,

                                                           unsigned int width, unsigned int height)

    {

                   if (p_bitmap == NULL)

                   {

                            return false;

                   }

 

                   if ((width + x > p_bitmap->GetWidth()) || (height + y >p_bitmap->GetHeight()))

                   {

                            return false;

                   }

 

        Gdiplus::BitmapData bitmapData;

                   Gdiplus::Rect rect(x, y, width,height);

        

        if (p_bitmap->LockBits(&rect, Gdiplus::ImageLockModeWrite, pixelFormat, &bitmapData) != Gdiplus::Ok)

            {

                            return false;

                   }

 

                   pixelType *pixels = (pixelType*)bitmapData.Scan0;

                  

 

        for (unsigned int row=0; row<height; ++row)

                   {

                            for (unsigned int col=0; col<width; ++col)

                            {

                                     processor(&pixels[col+row*bitmapData.Stride/sizeof(pixelType)]);      

                            }

                   }

 

                   if (p_bitmap->UnlockBits(&bitmapData) != Gdiplus::Ok)

                   {

                            return false;

                   }

        

                   return true;

         }

}

 

#endif

ProcessPixelsOneByOne函数可以对图像中从(x,y)位置起始,width*height大小的区域进行处理。模板参数pixelType用于指定像素大小,例如在Win32平台上传入unsigned char即为8位,用于8阶灰度图。模板参数Processor为图像处理算法实现,可以定义类实现void operator(pixelType *)函数,或者传入同样接口的函数指针。

    如下便是一些算法示例(说明见具体注释):

#ifndef SPATIALDOMAIN_H

#define SPATIALDOMAIN_H

#include <cmath>

#include <string>

 

namespace nsimgtk

{

    // 8阶灰度图的灰度反转算法  

         class NegativeGray8

         {

         public:

                   void operator()(unsigned char *const p_value)

                   {

                            *p_value ^= 0xff;

                   }

         };

    

    // 8阶灰度图的Gamma校正算法

         class GammaCorrectGray8

         {

         private:

                   unsigned char d_s[256];

         public:

                   GammaCorrectGray8::GammaCorrectGray8(double c, double gamma);

 

                   void operator()(unsigned char*const p_value)

                   {

                            *p_value = d_s[*p_value];

                   }

         };

 

    // 8阶灰度图的饱和度拉伸算法

         class ContrastStretchingGray8

         {

         private:

                   unsigned char d_s[256];

         public:

                   ContrastStretchingGray8::ContrastStretchingGray8(double a1, double b1, unsigned int x1,

                            double a2, double b2, unsigned int x2, double a3, double b3);

 

                   void operator()(unsigned char*const p_value)

                   {

                            *p_value = d_s[*p_value];

                   }

         };

    

    // 8阶灰度图的位平面分割,构造函数指定位平面号

         class BitPlaneSliceGray8

         {

         private:      

                   unsigned char d_s[256];

         public:

                   BitPlaneSliceGray8(unsigned char bitPlaneNum);

 

                   void operator()(unsigned char* const p_value)

                   {

                            *p_value = d_s[*p_value];

                   }

         };

}

 

#endif

 

// 上述类中各构造函数的实现代码,应该分在另一个文件中,此处为说明方便,一并列出

#include "SpatialDomain/spatialDomain.h"

 

namespace nsimgtk

{

         GammaCorrectGray8::GammaCorrectGray8(double c, double gamma)

         {

                   double temp;

                   for (unsigned int i=0; i<256; ++i)

                   {

                            temp = ceil(c * 255.0 * pow(double(i)/255.0, gamma));

                            d_s[i] = unsigned char(temp);

                   }

         }

 

         ContrastStretchingGray8::ContrastStretchingGray8(double a1, double b1, unsigned int x1,

                            double a2, double b2, unsigned int x2, double a3, double b3)

         {

                   if (x1 > 255 || x2 > 255 || x1 > x1)

                   {

                            for (unsigned int i=0; i<256; ++i)

                                     d_s[i] = i;

                   }

                   else

                   {

                            double tmp;

                            for (unsigned int i=0; i<x1; ++i)

                            {

                                     tmp = ceil(a1*double(i)+b1);

                                     d_s[i] = (unsigned char)tmp;

                            }

 

                            for (unsigned int i=x1; i<x2; ++i)

                            {

                                     tmp = ceil(a2*double(i)+b2);

                                     d_s[i] = (unsigned char)tmp;

                            }

 

                            for (unsigned int i=x2; i<256; ++i)

                            {

                                     tmp = ceil(a3*double(i)+b3);

                                     d_s[i] = (unsigned char)tmp;

                            }

                   }

         }

 

         BitPlaneSliceGray8::BitPlaneSliceGray8(unsigned char bitPlaneNum)

         {

                  unsigned char bitMaskArray[8] =

                   {

                            0x01, 0x02, 0x04, 0x08,

                            0x10, 0x20, 0x40, 0x80

                   };

 

                   for (unsigned int i=0; i<256; ++i)

                   {

                            unsigned char tmp = i;

                            tmp &= bitMaskArray[bitPlaneNum];

                            tmp = (tmp >> bitPlaneNum) * 255;

                            d_s[i] = tmp;

                  }

         }

}

(2) 直方图在GDI+1.0中没有获得支持,我们必须自行实现。直方图相关的处理在数字图像处理中占有重要地位,可以通过它获取图像灰度级的统计信息,且可以通过直方图进行一些重要的图像增强技术,如直方图均衡化,直方图规定化,基本全局门限等。

下面是获取8阶图像直方图的算法实现:

namespace nsimgtk

{

         bool GetHistogramNormalizeGray8(Gdiplus::Bitmap * const p_bitmap, float *histogramArray)

         {

                   if (p_bitmap == NULL || histogramArray == NULL)

                   {

                            return false;

                   }

 

                   Gdiplus::BitmapData bitmapData;

                   Gdiplus::Rect rect(0, 0, p_bitmap->GetWidth(), p_bitmap->GetHeight());

 

                   if (p_bitmap->LockBits(&rect, Gdiplus::ImageLockModeRead, PixelFormat8bppIndexed, &bitmapData) != Gdiplus::Ok)

            {

                            return false;

                   }

 

                   unsigned char *pixels = (unsigned char*)bitmapData.Scan0;

        unsigned int histogram[256];

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

                   {

                            histogram[i] = 0;

                   }

 

                   for (unsigned int row=0; row<p_bitmap->GetWidth(); ++row)

                   {

                            for (unsigned int col=0; col<p_bitmap->GetHeight(); ++col)

                            {

                                     ++histogram[pixels[col+row*bitmapData.Stride]];

                            }

                   }

 

                   const unsigned int totalPixels = p_bitmap->GetWidth() * p_bitmap->GetHeight();

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

                   {

                            histogramArray[i] = float(histogram[i]) / float(totalPixels);

                   }

 

                   if (p_bitmap->UnlockBits(&bitmapData) != Gdiplus::Ok)

                   {

                            return false;

                   }

 

                   return true;

         }

}

在获取直方图后(即上面算法的第二个参数),再将其作为参数传入下面的对象的构造函数,然后以该对象为仿函数传入ProcessPixelsOneByOne即可实现8阶图像直方图均衡化:

#ifndef SPATIALDOMAIN_H

#define SPATIALDOMAIN_H

 

#include <cmath>

#include <string>

 

namespace nsimgtk

{

    // 8阶灰度图的直方图均衡化

         class HistogramEqualizationGray8

         {

         private:

                   unsigned char d_s[256];

         public:

                   HistogramEqualizationGray8(const float *const histogramArray);

                  

                   void operator()(unsigned char *const p_value)

                   {

                            *p_value = d_s[*p_value];

                   }

         };

}

 

#endif        

 

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

#include "SpatialDomain/spatialDomain.h"

 

namespace nsimgtk

{

         HistogramEqualizationGray8::HistogramEqualizationGray8(const float *const histogramArray)

         {

                   if (histogramArray != NULL)

                   {

                            float sum = 0.0;

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

                            {

                                     sum += histogramArray[i];

                                     d_s[i] = unsigned char(sum * 255);

                            }

                   }

         }

}      

(3)空间域滤波器,滤波器是一个m*n大小的掩模,其中m,n均为大于1的奇数。滤波器逐像素地通过图像的全部或部分矩形区域,然后逐像素地对掩模覆盖下的像素使用滤波器算法获得响应,将响应赋值于当前像素即掩模中心像素,另外滤波器算法使用中将会涉及到图像边缘的问题,这可以对边缘部分掩模使用补零法补齐掩模下无像素值的区域,或者掩模的移动范围以不越出图像边缘的方式移动,当然这些处理方法都会给图像边缘部分带来不良效果,但是一般情况下,图像边缘部分往往不是我们关注的部分或者没有重要的信息。

下面的滤波器算法框架SpatialFilterAlgo即以补零法(zero-padding)实现:

#ifndef SPATIALFILTER_H

#define SPATIALFILTER_H

 

#include <vector>

#include <numeric>

#include <algorithm>

#include <gdiplus.h>

#include <fstream>

#include <cmath>

 

namespace nsimgtk

{

    template <typename pixelType, Gdiplus::PixelFormat pixelFormat, class FilterMask>

    bool SpatialFilterAlgo(Gdiplus::Bitmap* const p_bitmap, FilterMask filterMask, unsigned int x, unsigned int y,

                                                           unsigned int width, unsigned int height)

    {

                   if (p_bitmap == NULL)

                   {

                            return false;

                   }

 

                   if ((width + x > p_bitmap->GetWidth()) || (height + y >p_bitmap->GetHeight()))

                   {

                            return false;

                   }

 

        Gdiplus::BitmapData bitmapData;

                   Gdiplus::Rect rect(x, y, width,height);

        

        if (p_bitmap->LockBits(&rect, Gdiplus::ImageLockModeWrite, pixelFormat, &bitmapData) != Gdiplus::Ok)

            {

                            return false;

                   }

 

                   pixelType *pixels = (pixelType*)bitmapData.Scan0;

                  

        const unsigned int m = filterMask.d_m;                                         // mask's width

        const unsigned int n = filterMask.d_n;                                          // mask's height

        std::vector<pixelType> tmpImage((m-1+width)*(n-1+height));   // extend image to use zero-padding

        

        // copy original bitmap to extended image with zero-padding method

        for (unsigned int row=0; row<height; ++row)

                   {

                            for (unsigned int col=0; col<width; ++col)

                            {

                                     tmpImage[(col+m/2)+(row+n/2)*(bitmapData.Stride/sizeof(pixelType)+m-1)] =

                    pixels[col+row*bitmapData.Stride/sizeof(pixelType)];    

                            }

                   }

        

        // process every pixel with filterMask

        for (unsigned int row=0; row<height; ++row)

                   {

                            for (unsigned int col=0; col<width; ++col)

                            {

                // fill the "m*n" mask with the current pixel's neighborhood

                for (unsigned int i=0; i<n; ++i)

                {

                    for (unsigned int j=0; j<m; ++j)

                    {

                        filterMask.d_mask[i*m+j] = tmpImage[(col+j)+(row+i)*(bitmapData.Stride/sizeof(pixelType)+m-1)];

                    }

                }

              

                // replace the current pixel with filter mask's response

                                     pixels[col+row*bitmapData.Stride/sizeof(pixelType)] = filterMask.response();      

                            }

                   }

 

        if (p_bitmap->UnlockBits(&bitmapData) != Gdiplus::Ok)

                   {

                            return false;

                   }

        

                   return true;

         }

}

 

#endif

其中模板参数FilterMask即为滤波掩模算法。通常的滤波算法有均值滤波器,可以模糊化图像,去除图形中的细节部分,使得我们可以关注图像中较为明显的部分,均值滤波器用于周期性噪声。中值滤波器用于图像中存在椒盐噪声也即脉冲噪声的情况下。另外有基于一阶微分的Sobel梯度算子和基于两阶微分的拉普拉斯算子,它们往往被用于边缘检测中。

下面是一些滤波器算法的具体实现,所以滤波器算法都应该实现pixelType response()函数以及有d_mask,d_m,d_n成员,这可以通过继承__filteMask类获得(不需要付出虚函数代价)。

#ifndef SPATIALFILTER_H

#define SPATIALFILTER_H

 

#include <vector>

#include <numeric>

#include <algorithm>

#include <gdiplus.h>

#include <fstream>

#include <cmath>

 

namespace nsimgtk

{

    // 滤波器掩模的基类,提供掩模大小d_m,d_n,掩模覆盖下的m*n个像素值d_mask

    // others filterMask should inherit it

    template <typename pixelType>

    struct __filterMask

    {

        const unsigned int d_m;

        const unsigned int d_n;

 

        // image's pixels under the m*n filter mask

        std::vector<pixelType> d_mask;

 

        // filter mask's width and heigh must be a odd, if not, it will plus one for the width or the height

        __filterMask(unsigned int m, unsigned int n)

            : d_m(m%2 ? m:m+1), d_n(n%2 ? n:n+1), d_mask(d_m*d_n)

        {

        }

    };

 

    // 掩模权值为全1的均值滤波器

    template <typename pixelType>

    class averagingFilterMaskSp

        : public __filterMask<pixelType>

    {

    public:

        averagingFilterMaskSp(unsigned int m, unsigned int n)

            : __filterMask<pixelType>(m, n)

        { }

 

        pixelType response()

        {

            return std::accumulate(d_mask.begin(), d_mask.end(), 0) / (d_m * d_n);

        }

    };

 

    // 可自定义掩模权值的均值滤波器

    template <typename pixelType>

    class averagingFilterMask

        : public __filterMask<pixelType>

    {

    private:

        std::vector<pixelType> d_weight;                 // weights' vector(m*n)

        int d_weight_sum;                                        // all weights' sum

 

    public:

        averagingFilterMask(unsigned int m, unsigned int n, const std::vector<pixelType>& weightVec)

            : __filterMask<pixelType>(m, n), d_weight(weightVec)

        {

            if (weightVec.size() != d_mask.size())

            {

                // if weight's size isn't equal to mask's size, it will change filter mask as a special filter mask

                d_weight.resize(d_mask.size(), 1);

            }

 

            d_weight_sum = std::accumulate(d_weight.begin(), d_weight.end(), 0);

        }

 

        pixelType response()

        {

            return std::inner_product(d_mask.begin(), d_mask.end(), d_weight.begin(), 0) / d_weight_sum;

        }

    };

 

    // 中值滤波器

    template <typename pixelType>

    class medianFilterMask

        : public __filterMask<pixelType>

    {

    public:

        medianFilterMask(unsigned int m, unsigned int n)

            : __filterMask<pixelType>(m, n)

        { }

 

        pixelType response()

        {

            std::sort(d_mask.begin(), d_mask.end());

            return d_mask[d_mask.size()/2];

        }

    };

 

    // 3*3拉普拉斯滤波器

    // the mask is:  [0 1 0           [0 -1 0

    //             1 -5 1     or    -1 5 -1

    //             0 1 0]          0 -1 0]

    // if pixel's brightness is less than min, set it to min

    // if pixel's brightness is larger than max, set it to max

    template <typename pixelType, pixelType min, pixelType max>

    class laplacianFilter

        : public __filterMask<pixelType>

    {

    public:

        laplacianFilter()

            : __filterMask<pixelType>(3, 3)

        {  }

 

        pixelType response()

        {

            int ret = (int)(5*(int)d_mask[4]) -  ((int)d_mask[5]+d_mask[3]+d_mask[1]+d_mask[7]);

            if (ret < min)

                ret = min;

            if (ret > max)

                ret = max;

            return ret;

        }

    };

 

    // 3*3Sobel滤波器

    // the mask is: [-1 -2 -1            [-1 0 1

    //            0 0 0    and       -2 0 2

    //            1 2 1]             -1 0 1]

    // if pixel's brightness is larger than max, set it to max

    template <typename pixelType, pixelType max>

    class sobelFilter

        : public __filterMask<pixelType>

    {

    public:

        sobelFilter()

            : __filterMask<pixelType>(3, 3)

        {  }

 

        pixelType response()

        {

            int ret = ::abs(d_mask[6]+2*d_mask[7]+d_mask[8]-d_mask[0]-2*d_mask[1]-d_mask[2])

                + ::abs(d_mask[2]+2*d_mask[5]+d_mask[8]-d_mask[0]-2*d_mask[3]-d_mask[6]);

            

            if (ret > max)

                ret = max;

            return ret;

        }

    };

}

 

#endif

 

 

赞助本站

人工智能实验室
AiLab云推荐
展开

热门栏目HotCates

Copyright © 2010-2024 AiLab Team. 人工智能实验室 版权所有    关于我们 | 联系我们 | 广告服务 | 公司动态 | 免责声明 | 隐私条款 | 工作机会 | 展会港