DPCM压缩系统的实现和分析
DPCM是差分预测编码调制的缩写,是比较典型的预测编码系统。在DPCM系统中,需要注意的是预测器的输入是已经解码以后的样本。之所以不用原始样本来做预测,是因为在解码端无法得到原始样本,只能得到存在误差的样本。因此,在DPCM编码器中实际内嵌了一个解码器,如编码器中虚线框中所示。
分别进行线性预测器和量化器的优化设计。
【实验要求】
首先读取一个
256级的灰度图像,采用自己设定的预测方法计算预测误差,并对预测误差进行8比特均匀量化(基本要求) 。还可对预测误差进行1比特、2比特和4比特的量化设计。
【代码】
【读入文件】
FILE *renFile = NULL;
if ((renFile = fopen("D:\\数据压缩\\DPCM\\ren.yuv", "rb")) == NULL)
{
printf("open Linnafile fail!");
}
FILE *PredictedFile = NULL;
if ((PredictedFile = fopen("D:\\数据压缩\\DPCM\\PredictedFile.yuv", "wb")) == NULL)
{
printf("open ForecastFile fail!");
}
FILE *RebuildFile = NULL;
if ((RebuildFile = fopen("D:\\数据压缩\\DPCM\\RebuildFile.yuv", "wb")) == NULL)
{
printf("open RebuildFile fail!");
}
int w = 256;
int h = 256;
int N = 8;
unsigned char *ybuffer, *ubuffer, *vbuffer, *yPre, *yRe;
ybuffer = (unsigned char *)malloc(sizeof(char)*(w*h));
yPre = (unsigned char *)malloc(sizeof(char)*(w*h));
yRe = (unsigned char *)malloc(sizeof(char)*(w*h));
ubuffer = (unsigned char *)malloc(sizeof(char)*(w*h) / 4);
vbuffer = (unsigned char *)malloc(sizeof(char)*(w*h) / 4);
fread(ybuffer, 1, w*h, renFile);
fread(ubuffer, 1, w*h / 4, renFile);
fread(vbuffer, 1, w*h / 4, renFile);
下面这一部分是因为发现在写代码的过程中,图片颜色不对。
for (int k = 0; k < (w*h / 4); k++)
{
ubuffer[k] = 128;
vbuffer[k] = 128;
}
是这个样子的图
(图片的正反有问题,后面会解释)
【DPCM】
void DPCM(unsigned char *ybuffer, unsigned char *yPre, unsigned char *yRe, int N, int w, int h)
{
char *error;
error = (char *)malloc(sizeof(char)*(w*h));//量化前误差
int *Qerror;
Qerror = (int *)malloc(sizeof(int)*(w*h));//量化后误差
double Nbit = pow(2.0, N);//量化bite数
double step = 512 / Nbit;//量化间隔
for (int i = 0; i < h; i++)
{
//第一列像素
error[i*w + 0] = (char)(ybuffer[i*w + 0] - 128);
Qerror[i*w + 0] = int((error[i*w + 0] + 255) / step + 0.5);//第一列像素预测误差值等于0
yPre[i*w + 0] = (unsigned char)Qerror[i*w + 0];
if (yPre[i*w + 0] > Nbit - 1)
yPre[i*w + 0] = (unsigned char)(Nbit - 1);
if (yPre[i*w + 0] < 0)
yPre[i*w + 0] = 0;
yRe[i*w + 0] = (yPre[i*w + 0] * step - 255) + 128;
}
for (int i = 0; i < h; i++)
{
for (int j = 1; j < w; j++)
{
error[i*w + j] = (char)(ybuffer[i*w + j] - yRe[i*w + j - 1]);
Qerror[i*w + j] = int((error[i*w + j] + 255) / step + 0.5);//量化
yPre[i*w + j] = (unsigned char)Qerror[i*w + j];
if (yPre[i*w + j] > Nbit - 1)
yPre[i*w + j] = (unsigned char)(Nbit - 1);
if (yPre[i*w + j] < 0)
yPre[i*w + j] = 0;
yRe[i*w + j] = (yPre[i*w + j] * step - 255) + yRe[i*w + j - 1];//反量化
if (yRe[i*w + j] > 255)
yRe[i*w + j] = 255;
if (yRe[i*w + j] < 0)
yRe[i*w + j] = 0;
}
}
}
//DPCM
DPCM(ybuffer, yPre, yRe, N, w, h);
//计算频率
//输出原始图像频率分布
//FILE *renFile = NULL;
if ((renFile = fopen("D:\\数据压缩\\DPCM\\ren.txt", "wb")) == NULL)
{
printf("open ren fail!");
}
double fre1[256] = { 0 };
for (int i = 0; i < 256; i++)
{
for (int j = 0; j < w*h; j++)
{
if (ybuffer[j] == i)
fre1[i]++;
}
fre1[i] = fre1[i] / (w*h);
fprintf(renFile, "%lf\n", fre1[i]);
}
//输出重建图像频率分布
FILE *RebuildFre = NULL;
if ((RebuildFre = fopen("D:\\数据压缩\\DPCM\\RebuildFre.txt", "wb")) == NULL)
{
printf("open RebuildFre fail!");
}
double fre2[256] = { 0 };
for (int i = 0; i < pow(2.0, N); i++)
{
for (int j = 0; j < w*h; j++)
{
if (yRe[j] == i)
fre2[i]++;
}
fre2[i] = fre2[i] / (w*h);
fprintf(RebuildFre, "%lf\n", fre2[i]);
}
【PSNR】
//计算MSE和PSNR
double MSE = 0;
double PSNR = 0;
for (int i = 0; i < w*h; i++)
{
MSE = MSE + pow((ybuffer[i] - yRe[i]), 2.0);
}
MSE = MSE / (w*h);
PSNR = 10 * log10((255 * 255) / MSE);
printf("N: %d\n", N);
//printf("MSE is %lf\n", MSE);
printf("PSNR is %lf\n", PSNR);
【写入】
//写入预测误差图像
fwrite(yPre, 1, w*h, PredictedFile);
fwrite(ubuffer, 1, w*h / 4, PredictedFile);
fwrite(vbuffer, 1, w*h / 4, PredictedFile);
//写入预测图像
fwrite(yRe, 1, w*h, RebuildFile);
fwrite(ubuffer, 1, w*h / 4, RebuildFile);
fwrite(vbuffer, 1, w*h / 4, RebuildFile);
fclose(renFile);
fclose(PredictedFile);
fclose(RebuildFile);
fclose(renFile);
fclose(RebuildFre);
free(ybuffer);
free(ubuffer);
free(vbuffer);
free(yPre);
free(yRe);
return 0;
【得到的文件】
【频率分布图】
遇到的问题:
(1)图片的颜色不对,是uv出了问题,于是加了一个判断
for (int k = 0; k < (w*h / 4); k++)
{
ubuffer[k] = 128;
vbuffer[k] = 128;
}
(2)图片的正反不对,猜测是因为直接通过改名更改文件文件格式使图片出现了问题。
后面用现成的yuv文件就成功了。PS出问题,没能转换成功,用别的图片完成本实验