void HomoFilter(cv::Mat srcImg, cv::Mat &dst)
{
srcImg.convertTo(srcImg, CV_64FC1);
dst.convertTo(dst, CV_64FC1);
//第一步,取对数
for (int i = 0; i < srcImg.rows; i++)
{
double* srcdata = srcImg.ptr<double>(i);
double* logdata = srcImg.ptr<double>(i);
for (int j = 0; j < srcImg.cols; j++)
{
logdata[j] = log(srcdata[j] + 1);
}
}
//第二步,傅里叶变换
cv::Mat mat_dct = cv::Mat::zeros(srcImg.rows, srcImg.cols, CV_64FC1);
dct(srcImg, mat_dct);
cv::namedWindow("dct", cv::WINDOW_FREERATIO);
cv::imshow("dct", mat_dct);
//第三步,频域滤波
cv::Mat H_u_v;
int n1 = floor(srcImg.rows / 2);
int n2 = floor(srcImg.cols / 2);
double gammaH = 5;
double gammaL = 0.5;
double C = 3;
double d0 = 9.0;
double d2 = 0;
H_u_v = cv::Mat::zeros(srcImg.rows, srcImg.cols, CV_64FC1);
double totalWeight = 0.0;
for (int i = 0; i < srcImg.rows; i++)
{
double * dataH_u_v = H_u_v.ptr<double>(i);
for (int j = 0; j < srcImg.cols; j++)
{
d2 = pow(i - n1, 2.0) + pow(j - n2, 2.0);
dataH_u_v[j] = (gammaH - gammaL)*exp(C*(-d2 / (d0*d0))) + gammaL;
}
}
H_u_v.ptr<double>(0)[0] = 1.5;
mat_dct = mat_dct.mul(H_u_v);
//第四步,傅里叶逆变换
idct(mat_dct, dst);
//第五步,取指数运算
for (int i = 0; i < srcImg.rows; i++)
{
double* srcdata = dst.ptr<double>(i);
double* dstdata = dst.ptr<double>(i);
for (int j = 0; j < srcImg.cols; j++)
{
dstdata[j] = exp(srcdata[j]);
}
}
dst.convertTo(dst, CV_8UC1);
}