• 大小: 8KB
    文件类型: .cpp
    金币: 1
    下载: 0 次
    发布日期: 2021-06-15
  • 语言: C/C++
  • 标签: wavelet  opencv  

资源简介

基于vs2010与opencv实现的第二代提升小波,代码经调试可用,阅读性强,思路清晰。

资源截图

代码片段和文件信息

// DWT.cpp : 定义控制台应用程序的入口点。
//
//#include “stdafx.h“

#include
#include “cv.h“
#include “highgui.h“
// 二维离散小波变换(单通道浮点图像)

using namespace std;
void DWT(IplImage *pImage int nlayer)
{
// 执行条件
if (pImage)
{
if (pImage->nChannels == 1 &&pImage->depth == IPL_DEPTH_32F &&
((pImage->width >> nlayer) << nlayer) == pImage->width &&
((pImage->height >> nlayer) << nlayer) == pImage->height)
{
int     i x y n;
float   fValue   = 0;
float   fRadius  = sqrt(2.0f);
int     nWidth   = pImage->width;
int     nHeight  = pImage->height;
int     nHalfW   = nWidth / 2;
int     nHalfH   = nHeight / 2;
float **pData    = new float*[pImage->height];//生成二维数组用于对图像逐像素进行运算
float  *pRow     = new float[pImage->width];
float  *pColumn  = new float[pImage->height];
for (i = 0; i < pImage->height; i++)
{
//pData每一个指针头指向输入图像pImage中的每一行,这样就可对输入图像进行关联,从而进行操作
pData[i] = (float*) (pImage->imageData + pImage->widthStep * i);
}
// 多层小波变换
for (n = 0; n < nlayer; n++ nWidth /= 2 nHeight /= 2 nHalfW /= 2 nHalfH /= 2)
{
// 水平变换
for (y = 0; y < nHeight; y++)
{
// 奇偶分离

memcpy(pRow pData[y] sizeof(float) * nWidth);//每次将数据中的一行拿出来
for (i = 0; i < nHalfW; i++)
{
x = i * 2;
pData[y][i] = pRow[x];//pData前半段为偶数位
pData[y][nHalfW + i] = pRow[x + 1];//pData后半段为奇数位
}
// 提升小波变换
for (i = 0; i < nHalfW - 1; i++)
{
fValue = (pData[y][i] + pData[y][i + 1]) / 2;
pData[y][nHalfW + i] -= fValue;
}
fValue = (pData[y][nHalfW - 1] + pData[y][nHalfW - 2]) / 2;
pData[y][nWidth - 1] -= fValue;

fValue = (pData[y][nHalfW] + pData[y][nHalfW + 1]) / 4;
pData[y][0] += fValue;
for (i = 1; i < nHalfW; i++)
{
fValue = (pData[y][nHalfW + i] + pData[y][nHalfW + i - 1]) / 4;
pData[y][i] += fValue;
}
// 频带系数
for (i = 0; i < nHalfW; i++)
{
pData[y][i] *= fRadius;
pData[y][nHalfW + i] /= fRadius;
}
}


// 垂直变换
for (x = 0; x < nWidth; x++)
{
// 奇偶分离
for (i = 0; i < nHalfH; i++)
{
y = i * 2;
pColumn[i] = pData[y][x];
pColumn[nHalfH + i] = pData[y + 1][x];
}
for (i = 0; i < nHeight; i++)
{
pData[i][x] = pColumn[i];
}
// 提升小波变换
for (i = 0; i < nHalfH - 1; i++)
{
fValue = (pData[i][x] + pData[i + 1][x]) / 2;
pData[nHalfH + i][x] -= fValue;
}
fValue = (pData[nHalfH - 1][x] + pData[nHalfH - 2][x]) / 2;
pData[nHeight - 1][x] -= fValue;
fValue = (pData[nHalfH][x] + pData[nHalfH + 1][x]) / 4;
pData[0][x] += fValue;
for (i = 1; i < nHalfH; i++)
{
fValue = (pData[nHalfH + i][x] + pData[nHalfH + i - 1][x]) / 4;
pData[i][x] += fValue;
}
// 频带系数
for (i = 0; i < nHalfH; i++)
{
pData[i][x] *= fRadius;
pData[nHalfH + i][x] /

评论

共有 条评论