QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配

From: http://www.cnblogs.com/qytan36/archive/2010/04/04/1704226.html

实验环境:
1,Linux操作系统
2,QT3编程开发环境
3,C++编程语言

傅立叶变换和傅立叶反变换

1.1. 主要源代码

readImage() 从图像中读取数据
writeImage() 往图像中写入数据
fft() 快速傅立叶变换
ifft() 快速傅立叶反变换
adjustImageSize() 调整图像大小
fourier() 傅立叶变换
ifourier() 傅立叶反变换

1.1.1 从图像中读取数据

void ImageProcess::readImage(complex data[], const QImage &srcImage)

{

byte *pImageBytes = srcImage.bits(); //数据首地址

int depth = srcImage.depth(); //每个像素的bit数

int lineBytes = srcImage.bytesPerLine(); //每行的字节数

int w = srcImage.width(); //宽

int h = srcImage.height(); //高

byte *pByte;

//遍历读取每个像素,并转换为灰度值

int i, j;

for(i = 0; i < h; i++)

{

for(j = 0; j < w; j++)

{

if(8 == depth) //采用了256色调色板,8位颜色索引

{

pByte = pImageBytes + i * lineBytes + j;

data[i * w + j] = complex( *pByte, 0);

}

else if(32 == depth)//32位表示,数据格式为0xFFBBGGRR或0xAABBGGRR

{

pByte = pImageBytes + i * lineBytes + j * 4;

//根据RGB模式转化成YIQ色彩模式的方式,取Y作为灰度值

byte pixelValue = (byte)(0.299 * (float)pByte[0] + 0.587 * (float)pByte[1]

+ 0.114 * (float)pByte[2]);

data[i * w + j] = complex( pixelValue, 0);

}

else

{

cout << "invalid format. depth = " << depth << "\n";

return;

}

}

}

}

1.1.2 将数据写入图像

//coef为比例系数,主要用来调整灰度值以便于观察

void ImageProcess::writeImage(QImage &destImage, const complex data[], double coef)

{

int lineBytes = destImage.bytesPerLine();

int depth = destImage.depth();

int w = destImage.width();

int h = destImage.height();

byte *pImageBytes = destImage.bits();

byte *pByte;

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

{

for(int j = 0; j < w; j++)

{

double spectral = abs(data[i * w + j]) * coef; //灰度值调整

spectral = spectral > 255 ? 255 : spectral;

//根据图像格式写数据

if(8 == depth)

{

pByte = pImageBytes + i * lineBytes + j;

*pByte = spectral;

}

else if(32 == depth)

{

pByte = pImageBytes + i * lineBytes + j * 4;

pByte[0] = pByte[1] = pByte[2] = spectral;

}

else

{

return;

}

}

}

}

1.1.3 递归形式的快速傅立叶变换

//数组a为输入,数组y为输出,2的power次方为数组的长度

void ImageProcess::fft(const complex a[], complex y[], int power)

{

if(0 == power)

{

y[0] = a[0];

return;

}

int n = 1 << power;

double angle = 2 * PI / n;

complex wn(cos(angle), sin(angle));

complex w(1, 0);

complex *a0 = new complex[n / 2];

complex *a1 = new complex[n / 2];

complex *y0 = new complex[n / 2];

complex *y1 = new complex[n / 2];

for(int i = 0; i < n / 2; i ++)

{

a0[i] = a[2 * i];

a1[i] = a[2 * i + 1];

}

//分开成两个子fft过程

fft(a0, y0, power - 1);

fft(a1, y1, power - 1);

complex u;

for(int k = 0; k < n / 2; k++) //蝶形算法

{

u = w * y1[k];

y[k] = y0[k] + u;

y[k + n / 2] = y0[k] - u;

w = w * wn;

}

delete[] a0;

delete[] a1;

delete[] y0;

delete[] y1;

}

1.1.4 快速傅立叶反变换

//y为输入,a为输出,2的power次方为数组的长度

void ImageProcess::ifft(const complex y[], complex a[], int power)

{

int count = 1 << power;

complex *x = new complex[count];

memcpy(x, y, sizeof(complex) * count);

int i;

for(i = 0; i < count; i++)

{

x[i] = complex(x[i].real(), -x[i].imag()); //共轭复数

}

fft(x, a, power); //调用快速傅立叶变换算法

for(i = 0; i < count; i++)

{

a[i] = complex(a[i].real() / count, -a[i].imag() / count); //共轭复数

}

delete[] x;

}

1.1.5 调整图像的大小

//宽和高都截取为2的指数倍

void ImageProcess::adjustImageSize(QImage &image)

{

int w = 1;

int h = 1;

int width = image.width();

int height = image.height();

wp = 0, hp = 0;

while(w * 2 <= width) { w *= 2; wp++; }

while(h * 2 <= height) {h *= 2; hp++;}

QImage adjustedImage(w, h, image.depth(), image.numColors(), image.bitOrder());

byte *destBytes = adjustedImage.bits();

byte *srcBytes = image.bits();

int lineBytes = image.bytesPerLine();

int bytesPerPixel = image.depth() / 8; //每个象素的字节数

for(int i = 0; i < h; i++) //拷贝数据

{

memcpy(destBytes + i * w * bytesPerPixel, srcBytes + i * lineBytes,

sizeof(byte) * w * bytesPerPixel);

}

image = adjustedImage; //更新图像

}

1.1.6 傅立叶变换的主过程

void ImageProcess::fourier()

{

int w = currentImage.width();

int h = currentImage.height();

if(needAdjust) //调整图像的大小为2的幂次以便于快速傅立叶变换

{

adjustImageSize(currentImage); //调整大小

needAdjust = false;

if(currentImageData)

{

delete[] currentImageData;

}

currentImageData = new complex[w * h];

readImage(currentImageData, currentImage); //读取数据

}

else if(NULL == currentImageData)

{

currentImageData = new complex[w * h];

readImage(currentImageData, currentImage); //读取数据

}

w = currentImage.width(); //更新宽和高

h = currentImage.height();

complex *TD = currentImageData; //当前读取的数据为时域

complex *FD = new complex[w * h]; //申请空间保存变换结果

int i, j;

for(i = 0; i < h; i++) //在x方向上对按行进行快速傅立叶变换

{

fft(&TD[w * i], &FD[w * i], wp);

}

memcpy(TD, FD, sizeof(complex) * w * h);

complex *columnt = new complex[h];

complex *columnf = new complex[h];

for(i = 0; i < w; i++) //调整行列数据,在y方向上按列进行快速傅立叶变换

{

for(j = 0; j < h; j++)

{

columnt[j] = TD[j * w + i];

}

fft(columnt, columnf, hp);

for(j = 0; j < h; j++)

{

FD[j * w + i] = columnf[j];

}

}

delete[] columnt;

delete[] columnf;

writeImage(currentImage, FD, 0.02); //写入数据

delete[] currentImageData;

currentImageData = FD;

pDispLabel->setPixmap(QPixmap(currentImage));

}

1.1.7 傅立叶反变换

傅立叶反变换的思想与傅立叶变化相似,只是时域和频域互换,然后调用快速傅立叶反变换ifft而不是快速傅立叶变换fft。

1.2. 运行截图

1.2.1 正方形

输入一个256*256的图形,背景为白色,中间有一黑色的正方形,如图1-1所示。经过傅立叶变换后的结果如图1-2所示(注:没有采用平移到中心的方法)。

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第1张图片

图1-1

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第2张图片

图1-2

1.2.2 旋转45度

将图1-1旋转45度后的输入如图1-3所示。其傅立叶变换结果如图1-4所示。

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第3张图片

图1-3

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第4张图片

图1-4

1.2.3 输入长方形图像

输入图像如图1-5所示。傅立叶变换结果如图1-6所示。

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第5张图片

图1-5

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第6张图片

图1-6

1.2.4 傅立叶反变换

对傅立叶变换结果图1-2进行傅立叶反变换,其结果与原图1-1相同,如图1-7所示:

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第7张图片

图1-7


图像增强

图像增强是一种很重要的图像处理技术,为了方便人们观察以及机器处理而去处理给定的一幅图像。有很多图像增强的方法,以下这部分实现了其中的平滑和锐化这两种方法。

2.1. 主要源码

2.1.1 平滑

平滑采用的模板是clip_image016, 实现如下:

void ImageProcess::smooth()

{

int w = currentImage.width();

int h = currentImage.height();

if(NULL == currentImageData) //判断是否需要重新读取数据

{

currentImageData = new complex[w * h];

readImage(currentImageData, currentImage);

}

//拷贝一份数据便于计算

complex *buffer = new complex[w * h];

memcpy(buffer, currentImageData, sizeof(complex) * w * h);

//根据模板进行计算

//为了简化编码忽略了图像边界(i =0 or h, j =0 or w),对于整体效果没有影响

int i, j;

for(i = 1; i < h - 1; i++)

{

for(j = 1; j < w - 1; j++)

{

complex k;

k = buffer[(i - 1) * w + j - 1];

k += buffer[(i - 1) * w + j];

k += buffer[(i - 1) * w + j + 1];

k += buffer[i * w + j - 1];

k += buffer[i * w + j];

k += buffer[i * w + j + 1];

k += buffer[(i + 1) * w + j - 1];

k += buffer[(i + 1) * w + j];

k += buffer[(i + 1) * w + j + 1];

k = complex(k.real() / 9, 0);

currentImageData[i * w + j] = k;

}

}

writeImage(currentImage, currentImageData);

pDispLabel->setPixmap(QPixmap(currentImage));

}

2.1.2 锐化

采用拉普拉斯锐化,其模板为clip_image018,其实现如下:

void ImageProcess::sharp()

{

int w = currentImage.width();

int h = currentImage.height();

if(NULL == currentImageData) //判断是否需要读取数据

{

currentImageData = new complex[w * h];

readImage(currentImageData, currentImage);

}

//拷贝一份数据便于计算

complex *buffer = new complex[w * h];

memcpy(buffer, currentImageData, sizeof(complex) * w * h);

//根据模板进行计算

//为了简化编码忽略了图像边界(i =0 or h, j =0 or w),对于整体效果没有影响

int i, j;

complex k;

for(i = 1; i < h - 1; i++)

{

for(j = 1; j < w - 1; j++)

{

k = buffer[i * w + j];

k = complex(k.real() * 5, 0);

k -= buffer[(i - 1) * w + j];

k -= buffer[i * w + j - 1];

k -= buffer[i * w + j + 1];

k -= buffer[(i + 1) * w + j];

currentImageData[i * w + j] = k;

}

}

writeImage(currentImage, currentImageData);

pDispLabel->setPixmap(QPixmap(currentImage));

}

2.2. 运行截图

输入图像2-1,其平滑结果为图2-2,锐化结果为 图2-3。

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第8张图片

图2-1原来的图像

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第9张图片

图2-2 平滑后的图像

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第10张图片

图2-3 锐化后的图像


图像分析

这部分主要实现了图像的模板匹配。模板匹配是一种非常原始的模式识别方法。有很多模板匹配的算法。这里采用的算法是计算二者之间的相似度,在目标图像中选取一个坐标,将以该坐标为左上角选定一块区域,计算该区域与模板的相似度,相似度最大的点即为匹配之处。通过二者之间的差异度来判断其相似程度,差异度的计算:m = clip_image026。即将累加其像素之间的差值,为了提高计算速度,可以设置阀值,当m大于阀值时,认定该块区域不匹配,继续寻找下一区域。

3.1. 主要源码

void ImageProcess::match()

{

//让用户选取模板

QString fileName = QFileDialog::getOpenFileName("/home/tanqiyu", "Images (*.png *.xpm

.jpg)", this, "open file dialog", "Choose a model image");

if(QString::null == fileName)

{

return;

}

//读取模板数据

QImage modelImage(fileName);

int mw = modelImage.width();

int mh = modelImage.height();

complex *modelImageData = new complex[mw * mh];

readImage(modelImageData, modelImage);

unsigned long t = mw * mh * 8; //根据匹配模板的大小设置一定的阀值

unsigned long m = t; //初始差异度

int ri = -1; //z左上角坐标(ri, rj)

int rj = -1;

int w = currentImage.width();

int h = currentImage.height();

if(NULL == currentImageData) //判断是否需要读取目标图像数据

{

currentImageData = new complex[w * h];

readImage(currentImageData, currentImage);

}

//遍历目标图像,选取左上角坐标,考虑到模板图像的大小注意不要越界

int i, j;

for(i = 0; i < h - mh + 1; i++ )

{

for(j = 0; j < w - mw + 1; j++)

{

//下面开始对点(i, j)为左上角的mw * mh区域进行匹配

bool overFlag = false;

unsigned long k = 0; //差异值的累加和

int u, v;

for(u = 0; u < mh && !overFlag; u++)

{

for(v = 0; v < mw && !overFlag; v++)

{

k += abs(currentImageData[(i + u) * w + j + v].real()

- modelImageData[u * mw + v].real()); //计算差值并累加

if(k >= t) //判断是否大于阀值

{

overFlag = true;

}

}

}

if(k < m) //判断是否找到更加匹配的区域

{

ri = i;

rj = j;

m = k;

}

}

}

//找到匹配区域,则将目标图像匹配区域之外的点置成白色以便于观察结果

if(ri != -1)

{

for(i = 0; i < h; i++)

{

for(j = 0; j < w; j++)

{

if(i < ri || j < rj || i > ri + mh - 1 || j > rj + mw - 1)

{

currentImageData[i * w + j] = complex(255, 0);

}

}

}

}

writeImage(currentImage, currentImageData);

pDispLabel->setPixmap(QPixmap(currentImage));

}

3.2. 运行截图

目标图像为图3-1,图3-2位匹配模板,匹配结果为图3-3。

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第11张图片

图3-1 目标图像

clip_image028

图3-2 匹配模板

QT实现图像处理-傅立叶变换、傅立叶反变换、平滑、锐化与模板匹配_第12张图片

匹配结果


你可能感兴趣的:(QT)