Проблема с реализацией DCT

Я должен реализовать алгоритм DCT в C ++, вот мой настоящий код:

// dct: computes the discrete cosinus tranform of a 8x8 block
template<typename Tin=uchar,typename Tout=float>
inline cv::Mat_<Tout> dct(const cv::Mat_<Tin>& oBlock) {

int indexNumber;
float pi = 3.14159265359;
float fcoscos, fxy, cos1, cos2, forCos1, forCos2;

cv::Mat_<Tout> resultBloc(8, 8);

for (int u = 0; u < oBlock.rows; u++){
for (int v = 0; v < oBlock.cols; v++){

float cu=0, cv=0, Result=0;

// calcul c(u)
if (u == 0){
cu = (float)sqrt((float)1 / (float)oBlock.rows);
}
else {
cu = (float)sqrt((float)2 / (float)oBlock.rows);
}

// calcul c(v)
if (v == 0){
cv = (float)sqrt((float)1 / (float)oBlock.cols);
}
else {
cv = (float)sqrt((float)2 / (float)oBlock.cols);
}

float sums = 0;

for (int x = 0; x < oBlock.rows; x++){
for (int y = 0; y < oBlock.cols; y++){

indexNumber = x * oBlock.rows + y;
fxy = (int)oBlock.data[indexNumber];

forCos1 = (pi*((2 * x) + 1)*u) / (2 * oBlock.rows);
forCos2 = (pi*((2 * y) + 1)*v) / (2 * oBlock.cols);

cos1 = cos(forCos1);
cos2 = cos(forCos2);

fcoscos = fxy * cos1 * cos2;

sums += fcoscos;

}
}

// calcul total
Result = sums*cu*cv;

indexNumber = u * oBlock.rows + v;
resultBloc.data[indexNumber] = Result;

}
}

return resultBloc;

}

Я сравнил результат с алгоритмом cv DCT следующим образом:

cv::Mat_<float> tempImage(8,8);

for (int i = 0; i < vecImageCut[0].cols*vecImageCut[0].rows; i++){
tempImage.data[i] = (int)vecImageCut[0].data[i];
}
cv::Mat_<float> dctCV;
cv::dct(tempImage, dctCV);
for (int i = 0; i < blocksAfterDCT[0].cols*blocksAfterDCT[0].rows; i++){
std::cerr << "Difference DCT for pixel " << i << " : " << dctCV.data[i] - blocksAfterDCT[0].data[i] << std::endl;
}

Результаты между моим DCT и cv DCT очень разные, поэтому я предполагаю, что мой алгоритм DCT неверен, но я искал часы и не могу найти свою ошибку, может кто-нибудь сказать мне, где я сделал что-то не так?

1

Решение

Ваши расчеты индекса неверны. В indexNumber = x * oBlock.rows + y;, поскольку x подсчитывает строки, которые нужно умножить на количество столбцов:

indexNumber = x * oBlock.cols + y;

То же самое для indexNumber = u * oBlock.rows + v;

indexNumber = u * oBlock.cols + v;
0

Другие решения

Других решений пока нет …

По вопросам рекламы [email protected]