質問
Cで前方および逆離散コサイン変換(DCT)を実装しようとしています。コードは、dct()関数を介して変換マトリックスにピクセルの単一の入力ブロックをトランス変更し、IDCTを介して元のピクセル値に戻ることです( ) 関数。添付のコードをご覧ください。私の出力形式IDCTは、244、116、244、116などの連続した値です。IDCT値の外観から、私のプログラムが機能しているように見えません。各機能の後に期待する必要がありますか?明らかにIDCTの後、私は元の入力マトリックスに近づくようになるはずです。
ありがとう
# include <stdio.h>
# define PI 3.14
void dct(float [][]); // Function prototypes
void idct(float [][]); // Function prototypes
void dct(float inMatrix[8][8]){
double dct,
Cu,
sum,
Cv;
int i,
j,
u,
h = 0,
v;
FILE * fp = fopen("mydata.csv", "w");
float dctMatrix[8][8],
greyLevel;
for (u = 0; u < 8; ++u) {
for (v = 0; v < 8; ++v) {
if (u == 0) {
Cu = 1.0 / sqrt(2.0);
} else {
Cu = 1.0;
}
if (v == 0) {
Cv = 1.0 / sqrt(2.0);
} else {
Cu = (1.0);
}
sum = 0.0;
for (i = 0; i < 8; i++) {
for (j = 0; j < 8; j++) {
// Level around 0
greyLevel = inMatrix[i][j];
dct = greyLevel * cos((2 * i + 1) * u * PI / 16.0) *
cos((2 * j + 1) * v * PI / 16.0);
sum += dct;
}
}
dctMatrix[u][v] = 0.25 * Cu * Cv * sum;
fprintf(fp, "\n %f", dctMatrix[u][v]);
}
fprintf(fp, "\n");
}
idct(dctMatrix);
}
void idct(float dctMatrix[8][8]){
double idct,
Cu,
sum,
Cv;
int i,
j,
u,
v;
float idctMatrix[8][8],
greyLevel;
FILE * fp = fopen("mydata.csv", "a");
fprintf(fp, "\n Inverse DCT");
for (i = 0; i < 8; ++i) {
for (j = 0; j < 8; ++j) {
sum = 0.0;
for (u = 0; u < 8; u++) {
for (v = 0; v < 8; v++) {
if (u == 0) {
Cu = 1.0 / sqrt(2.0);
} else {
Cu = 1.0;
}
if (v == 0) {
Cv = 1.0 / sqrt(2.0);
} else {
Cu = (1.0);
}
// Level around 0
greyLevel = dctMatrix[u][v];
idct = (greyLevel * cos((2 * i + 1) * u * PI / 16.0) *
cos((2 * j + 1) * v * PI / 16.0));
sum += idct;
}
}
idctMatrix[i][j] = 0.25 * Cu * Cv * sum;
fprintf(fp, "\n %f", idctMatrix[i][j]);
}
fprintf(fp, "\n");
}
}
int main() {
float
testBlockA[8][8] = { {255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255} },
testBlockB[8][8] = {{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255},
{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255},
{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255},
{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255} };
dct(testBlockB);
}
解決
IFステートメントには、CV定数割り当てには少なくとも2つのタイプミスがあります。
if (v == 0) {
Cv = 1.0 / sqrt(2.0);
} else {
Cu = (1.0); // << this should be Cv = 1.0
}
しかし、あまり適切にチェックしませんでした。を使用して ドイツのウィキペディア Cosine Transformについては、コードの動作に続きます...私は、どの変換定数を定義したかを理解するのに時間を費やしたくありませんでした。正しい定数と逆関数を使用することを確認する必要があると思います。
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
void dct(float **DCTMatrix, float **Matrix, int N, int M);
void write_mat(FILE *fp, float **testRes, int N, int M);
void idct(float **Matrix, float **DCTMatrix, int N, int M);
float **calloc_mat(int dimX, int dimY);
void free_mat(float **p);
float **calloc_mat(int dimX, int dimY){
float **m = calloc(dimX, sizeof(float*));
float *p = calloc(dimX*dimY, sizeof(float));
int i;
for(i=0; i <dimX;i++){
m[i] = &p[i*dimY];
}
return m;
}
void free_mat(float **m){
free(m[0]);
free(m);
}
void write_mat(FILE *fp, float **m, int N, int M){
int i, j;
for(i =0; i< N; i++){
fprintf(fp, "%f", m[i][0]);
for(j = 1; j < M; j++){
fprintf(fp, "\t%f", m[i][j]);
}
fprintf(fp, "\n");
}
fprintf(fp, "\n");
}
void dct(float **DCTMatrix, float **Matrix, int N, int M){
int i, j, u, v;
for (u = 0; u < N; ++u) {
for (v = 0; v < M; ++v) {
DCTMatrix[u][v] = 0;
for (i = 0; i < N; i++) {
for (j = 0; j < M; j++) {
DCTMatrix[u][v] += Matrix[i][j] * cos(M_PI/((float)N)*(i+1./2.)*u)*cos(M_PI/((float)M)*(j+1./2.)*v);
}
}
}
}
}
void idct(float **Matrix, float **DCTMatrix, int N, int M){
int i, j, u, v;
for (u = 0; u < N; ++u) {
for (v = 0; v < M; ++v) {
Matrix[u][v] = 1/4.*DCTMatrix[0][0];
for(i = 1; i < N; i++){
Matrix[u][v] += 1/2.*DCTMatrix[i][0];
}
for(j = 1; j < M; j++){
Matrix[u][v] += 1/2.*DCTMatrix[0][j];
}
for (i = 1; i < N; i++) {
for (j = 1; j < M; j++) {
Matrix[u][v] += DCTMatrix[i][j] * cos(M_PI/((float)N)*(u+1./2.)*i)*cos(M_PI/((float)M)*(v+1./2.)*j);
}
}
Matrix[u][v] *= 2./((float)N)*2./((float)M);
}
}
}
int main() {
float
testBlockA[8][8] = { {255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255},
{255, 255, 255, 255, 255, 255, 255, 255} },
testBlockB[8][8] = {{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255},
{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255},
{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255},
{255, 0, 255, 0, 255, 0, 255, 0},
{0, 255, 0, 255, 0, 255, 0, 255} };
FILE * fp = fopen("mydata.csv", "w");
int dimX = 8, dimY = 8;
int i, j;
float **testBlock = calloc_mat(dimX, dimY);
float **testDCT = calloc_mat(dimX, dimY);
float **testiDCT = calloc_mat(dimX, dimY);
for(i = 0; i<dimX; i++){
for(j = 0; j<dimY; j++){
testBlock[i][j] = testBlockB[i][j];
}
}
dct(testDCT, testBlock, dimX, dimY);
write_mat(fp, testDCT, dimX, dimY);
idct(testiDCT, testDCT, dimX, dimY);
write_mat(fp, testiDCT, dimX, dimY);
fclose(fp);
free_mat(testBlock);
free_mat(testDCT);
free_mat(testiDCT);
return 0;
}
編集DCTは、Wikiの式DCT-IIの交差製品に基づいています。 IDCTは、次元ごとに正規化係数2/nを備えた式DCT-IIIの交差製品に基づいています(これは、テキストに記載されているDCT-IIの逆であるため)。編集逆DCTの因子は、バージョンで1/SQRT(2)ではなくSQRT(2)ではないことをかなり確信しています。
他のヒント
あなたはそうしない
#include <math.h>
それは、コンパイラが真実ではない数学関数についてのことを前提としていることを意味するかもしれません。int
. 。呼び出すすべての機能はどこかで宣言する必要があり、Cには「組み込み」がありません。 sin()
組み込みよりももう printf()
(後者については、正しく含まれています stdin.h
, 、 もちろん)。
また、使用できます M_PI
含めたら <math.h>
.
cv constant(dct()とidct()関数の両方でのタイプミスに関する以前の回答に加えて、使用しました 逆DCT式(2番目) 間違っています。ループで毎回CVとCUを掛ける必要がありました。したがって、idct()の正しいコードは次のとおりです。
void idct(float dctMatrix[8][8]){
double idct,
Cu,
sum,
Cv;
int i,
j,
u,
v;
float idctMatrix[8][8],
greyLevel;
FILE * fp = fopen("mydata.csv", "a");
fprintf(fp, "\n Inverse DCT");
for (i = 0; i < 8; ++i) {
for (j = 0; j < 8; ++j) {
sum = 0.0;
for (u = 0; u < 8; u++) {
for (v = 0; v < 8; v++) {
if (u == 0) {
Cu = 1.0 / sqrt(2.0);
} else {
Cu = 1.0;
}
if (v == 0) {
Cv = 1.0 / sqrt(2.0);
} else {
Cv = (1.0); //mistake was here - the same is in dct()
}
greyLevel = dctMatrix[u][v];
// Multiply by Cv and Cu here!
idct = (greyLevel * Cu * Cv *
cos((2 * i + 1) * u * PI / 16.0) *
cos((2 * j + 1) * v * PI / 16.0));
sum += idct;
}
}
// not "* Cv * Cu" here!
idctMatrix[i][j] = 0.25 * sum;
fprintf(fp, "\n %f", idctMatrix[i][j]);
}
fprintf(fp, "\n");
}
}
この場合、出力値は255、0、255、0などに近いです。