mean_final.zip_mean

#include <stdio.h> #include <stdlib.h> #include <math.h> #include <windows.h> unsigned char *pBmpBuf;//读入图像数据的指针 unsigned char *pBmpBuf2;//读入图像数据的指针 int bmpWidth;//图像的宽 int bmpHeight;//图像的高 RGBQUAD *pColorTable;//颜色表指针 int biBitCount;//图像类型，每像素位数 bool readBmp(char *bmpName) { FILE *fp=fopen(bmpName,"rb");//二进制读方式打开指定的图像文件 if(fp==0) return 0; fseek(fp, sizeof(BITMAPFILEHEADER),0);//跳过位图文件头结构BITMAPFILEHEADER BITMAPINFOHEADER head; //定义位图信息头结构变量，读取位图信息头进内存，存放在变量head中 fread(&head, sizeof(BITMAPINFOHEADER), 1,fp); //获取图像宽、高、每像素所占位数等信息 bmpWidth = head.biWidth; bmpHeight = head.biHeight; biBitCount = head.biBitCount; int lineByte=(bmpWidth * biBitCount/8+3)/4*4; //定义变量，计算图像每行像素所占的字节数（必须是4的倍数） //灰度图像有颜色表，且颜色表表项为256 if(biBitCount==8){ //申请颜色表所需要的空间，读颜色表进内存 pColorTable=new RGBQUAD[256]; fread(pColorTable,sizeof(RGBQUAD),256,fp); } //申请位图数据所需要的空间，读位图数据进内存 pBmpBuf=new unsigned char[lineByte * bmpHeight]; fread(pBmpBuf,1,lineByte * bmpHeight,fp); fclose(fp); return 1; } bool readorgBmp(char *bmpName) { FILE *fp=fopen(bmpName,"rb");//二进制读方式打开指定的图像文件 if(fp==0) return 0; fseek(fp, sizeof(BITMAPFILEHEADER),0);//跳过位图文件头结构BITMAPFILEHEADER BITMAPINFOHEADER head; //定义位图信息头结构变量，读取位图信息头进内存，存放在变量head中 fread(&head, sizeof(BITMAPINFOHEADER), 1,fp); //获取图像宽、高、每像素所占位数等信息 bmpWidth = head.biWidth; bmpHeight = head.biHeight; biBitCount = head.biBitCount; int lineByte=(bmpWidth * biBitCount/8+3)/4*4; //定义变量，计算图像每行像素所占的字节数（必须是4的倍数） //灰度图像有颜色表，且颜色表表项为256 if(biBitCount==8){ //申请颜色表所需要的空间，读颜色表进内存 pColorTable=new RGBQUAD[256]; fread(pColorTable,sizeof(RGBQUAD),256,fp); } //申请位图数据所需要的空间，读位图数据进内存 pBmpBuf2=new unsigned char[lineByte * bmpHeight]; fread(pBmpBuf,1,lineByte * bmpHeight,fp); fclose(fp); return 1; } bool saveBmp(char *bmpName, unsigned char *imgBuf, int width, int height, int biBitCount, RGBQUAD *pColorTable) { //如果位图数据指针为0，则没有数据传入，函数返回 //if(!imgBuf) //return 0; //颜色表大小，以字节为单位，灰度图像颜色表为1024字节，彩色图像颜色表大小为0 int colorTablesize=0; if(biBitCount==8) colorTablesize=1024; //待存储图像数据每行字节数为4的倍数 int lineByte=(width * biBitCount/8+3)/4*4; //以二进制写的方式打开文件 FILE *fp=fopen(bmpName,"wb"); //if(fp==0) return 0; //申请位图文件头结构变量，填写文件头信息 BITMAPFILEHEADER fileHead; fileHead.bfType = 0x4D42;//bmp类型 //bfSize是图像文件4个组成部分之和 fileHead.bfSize= sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + colorTablesize + lineByte*height; fileHead.bfReserved1 = 0; fileHead.bfReserved2 = 0; //bfOffBits是图像文件前3个部分所需空间之和 fileHead.bfOffBits=54+colorTablesize; //写文件头进文件 fwrite(&fileHead, sizeof(BITMAPFILEHEADER),1, fp); //申请位图信息头结构变量，填写信息头信息 BITMAPINFOHEADER head; head.biBitCount=biBitCount; head.biClrImportant=0; head.biClrUsed=0; head.biCompression=0; head.biHeight=height; head.biPlanes=1; head.biSize=40; head.biSizeImage=lineByte*height; head.biWidth=width; head.biXPelsPerMeter=0; head.biYPelsPerMeter=0; //写位图信息头进内存 fwrite(&head, sizeof(BITMAPINFOHEADER),1, fp); //如果灰度图像，有颜色表，写入文件 if(biBitCount==8) fwrite(pColorTable, sizeof(RGBQUAD),256, fp); //写位图数据进文件 fwrite(imgBuf, height*lineByte, 1, fp); fclose(fp); return 1; } int mean_filter(int D[9]) { unsigned int sum=0; int i; for(i=0;i<9;i++) sum+=D[i]; return sum/9; } //开方函数 int square(int x) { if (x<0) exit(0);//开方的值需大于0 int i,t,a; t = 0; i = 2; if (x>3) { while(1) { a = i*i; t = x - a; if (t<0) { i--; break; } i++; } } else i=x; return i; } //用sobel算子边缘滤波 int sobel(int D[9]) { int i; int a[9],b[9]; int sum=0,sum1=0,sum2=0; a[0]=-1; a[1]=-2; a[2]=-1; a[3]= 0; a[4]= 0; a[5]= 0; a[6]= 1; a[7]= 2; a[8]= 1; //y方向算子 b[0]=-1; b[1]= 0; b[2]= 1; b[3]=-2; b[4]= 0; b[5]= 2; b[6]=-1; b[7]= 0; b[8]= 1; //x方向算子 for (i=0;i<9;i++) { sum1 += D[i]*a[i]; //D矩阵乘以y方向算子 } for (i=0;i<9;i++) { sum2 += D[i]*b[i]; //D矩阵乘以x方向算子 } sum = square((sum1*sum1) + (sum2*sum2)); //开方函数求最后返回值 if (sum>255) { sum=255; //若大于255置为255 } return sum; } void main() { char readPath[]="pic1.bmp"; //加噪后BMP文件地址 char orgPath[]="pic3.bmp"; //加噪前BMP文件地址 readBmp(readPath); //读入加噪后BMP文件地址 printf("width=%d,height=%d,biBitCount=%d\n", bmpWidth,bmpHeight,biBitCount); //输出图像的信息 readorgBmp(orgPath);//读入加噪前BMP文件地址 int i,j; int **readData;//加噪后BMP文件数据 int **writeData;//加噪后BMP文件滤波后的数据 int **orgData;//加噪前BMP文件数据 int **sobelData;//sobel处理后数据 //根据得到的位宽和位高数据申请二维动态数组 readData=(int **)malloc(sizeof(int *)*bmpHeight); writeData=(int **)malloc(sizeof(int *)*bmpHeight); orgData=(int **)malloc(sizeof(int *)*bmpHeight); sobelData=(int **)malloc(sizeof(int *)*bmpHeight); for (i=0;i<bmpHeight;i++) { readData[i]=(int *)malloc(sizeof(int)*bmpWidth); writeData[i]=(int *)malloc(sizeof(int)*bmpWidth); orgData[i]=(int *)malloc(sizeof(int)*bmpWidth); sobelData[i]=(int *)malloc(sizeof(int)*bmpWidth); } int k=0; int readbuf; for(i=0;i<bmpHeight;i++) { for(j=0;j<bmpWidth;j++) { readbuf=*(pBmpBuf+k); readData[i][j]=readbuf; sobelData[i][j]=readbuf; writeData[i][j]=readbuf; orgData[i][j]=*(pBmpBuf2+k);//由得到的文件指针输出各个二维数组 k++; } } int D[9]; //选取框 for(i=1;i<bmpHeight-1;i++) for(j=1;j<bmpWidth-1;j++) { D[0]=readData[i-1][j+1]; D[1]=readData[i][j+1]; D[2]=readData[i+1][j+1]; D[3]=readData[i-1][j]; D[4]=readData[i][j]; D[5]=readData[i+1][j]; D[6]=readData[i-1][j-1]; D[7]=readData[i][j-1]; D[8]=readData[i+1][j-1];//选取框赋值 writeData[i][j]=mean_filter(D);//滤波 } k=0; for(i=0;i<bmpHeight;i++) { for(j=0;j<bmpWidth;j++) { *(pBmpBuf+k)=writeData[i][j];//滤波后的值送入*(pBmpBuf+k) k++; } } char writePath[]="pic2.bmp";//滤波后的图片输出地址 saveBmp(writePath, pBmpBuf, bmpWidth, bmpHeight, biBitCount, pColorTable);//保存滤波后的图片 //computer_snr double noise1, noise2, signal, temp; noise1=0.0; noise2=0.0; signal=0.0; for(i=0;i<bmpHeight;i++) for(j=0;j<bmpWidth;j++) { signal+=(double)(orgData[i][j]*orgData[i][j]); temp=(double)(orgData[i][j]-readData[i][j]); temp=(double)(orgData[i][j]-writeData[i][j]); noise2+=temp*temp; } printf("\n the input image's snr = %lfdB \n", 10.0*log10(signal/noise1));//输入图像SNR printf("\n the input image's PSNR = %lfdB \n", 10.0*log10(bmpHeight*bmpWidth*255.0*255.0/noise1));//输入图像PSNR printf("\n the improved snr = %lfdB \n", 10.0*log10(noise1/noise2)); printf("\n the output image's snr = %lfdB \n", 10.0*log10(signal/noise2));//输出图像SNR printf("\n the output image's PSNR = %lfdB \n", 10.0*log10(bmpHeight*bmpWidth*255.0*255.0/noise2));//输出图像PSNR char sobelPath[]="pic4.bmp";//sobel算子边缘检测后的输出地址 for(i=1;i<bmpHeight-1;i++) for(j=1;j<bmpWidth-1;j++) { D[0]=writeData[i-1][j+1]; D[1]=writeData[i][j+1]; D[2]=writeData[i+1][j+1]; D[3]=writeData[i-1][j]; D[4]=writeData[i][j]; D[5]=writeData[i+1][j]; D[6]=writeData[i-1][j-1]; D[7]=writeData[i][j-1]; D[8]=writeData[i+1][j-1]; sobelData[i][j]=sobel(D);//sobel算子边缘检测 } k=0; for(i=0;i