C++图像识别转灰度

  1 #include <stdafx.h>
  2 #include <stdio.h>
  3 #include <string.h>
  4 #include <math.h>
  5 #include <windows.h>
  6 using namespace std;
  7 
  8 
  9 //将位图转换为256色灰度图
 10 void ToGray(const string& srcFile,const string& desFile)
 11 {
 12     BITMAPFILEHEADER bmfHeader;
 13     BITMAPINFOHEADER bmiHeader;
 14     
 15     FILE *pFile;
 16     if ((pFile = fopen(srcFile.c_str(),"rb")) == NULL)
 17     {
 18         printf("open bmp file error.");
 19         exit(-1);
 20     }
 21     //读取文件和Bitmap头信息
 22     fseek(pFile,0,SEEK_SET);
 23     fread(&bmfHeader,sizeof(BITMAPFILEHEADER),1,pFile);
 24     fread(&bmiHeader,sizeof(BITMAPINFOHEADER),1,pFile);
 25     //先不支持16位位图
 26     int bitCount = bmiHeader.biBitCount;
 27     if (bitCount == 16)
 28     {
 29         exit(-1);
 30     }
 31     double byteCount = (double)bitCount / 8;
 32     int nClr = 0;
 33     if (bitCount < 16)
 34     {        
 35         nClr = bmiHeader.biClrUsed ? bmiHeader.biClrUsed : 1 << bitCount;
 36         if (nClr > 256)
 37             nClr = 0;        
 38     }
 39     //读取调色板
 40     RGBQUAD *quad = NULL;
 41     if (nClr > 0)
 42     {
 43         quad = new RGBQUAD[nClr];
 44         fread(quad,sizeof(RGBQUAD) * nClr,1,pFile);
 45     }
 46     
 47     int srcW = bmiHeader.biWidth;
 48     int srcH = bmiHeader.biHeight;
 49     //原图像每一行去除偏移量的字节数
 50     int lineSize = bitCount * srcW >> 3;
 51     //偏移量，windows系统要求每个扫描行按四字节对齐
 52     int alignBytes = (((bmiHeader.biWidth * bitCount + 31) & ~31) >> 3)
 53         - ((bmiHeader.biWidth * bitCount) >> 3);
 54     //原图像缓存    
 55     int srcBufSize = lineSize * srcH;
 56     BYTE* srcBuf = new BYTE[srcBufSize];
 57     int i,j;
 58     //读取文件中数据
 59     for (i = 0; i < srcH; i++)
 60     {        
 61         fread(&srcBuf[lineSize * i],lineSize,1,pFile);
 62         fseek(pFile,alignBytes,SEEK_CUR);
 63     }
 64     //256色位图调色板
 65     RGBQUAD *quadDes = NULL;
 66     quadDes = new RGBQUAD[256];
 67     for (i = 0; i < 256; i++)
 68     {
 69         //灰度图的RGB值相等
 70         quadDes[i].rgbBlue = quadDes[i].rgbGreen = quadDes[i].rgbRed = i;
 71     }
 72     //灰度图每个像素采用8位表示
 73     int desBufSize = (((srcW * 8 + 31) & ~31) >> 3) * srcH;
 74     BYTE *desBuf = new BYTE[desBufSize];
 75     //每个扫描行占用字节数
 76     int desLineSize = ((srcW * 8 + 31) >> 5) * 4;
 77 
 78     for (i = 0; i < srcH; i++)
 79     {
 80         for (j = 0; j < srcW; j++)
 81         {
 82             //从调色板中读取RGB值
 83             if (nClr > 0)
 84             {
 85                 unsigned int pos = srcBuf[i * lineSize + int(j * byteCount)];
 86                 desBuf[i * desLineSize + j] = 0.3 * quad[pos].rgbRed
 87                     + 0.59 * quad[pos].rgbGreen
 88                     + 0.11 * quad[pos].rgbBlue;
 89             }
 90             else
 91                 desBuf[i * desLineSize + j] = 0.3 * srcBuf[i * lineSize + int(j * byteCount)] 
 92                 + 0.59 * srcBuf[i * lineSize + int(j * byteCount) + 1] 
 93                 +  0.11 * srcBuf[i * lineSize + int(j * byteCount) + 2];
 94         }
 95     }
 96 
 97     //创建目标文件
 98     HFILE hfile = _lcreat(desFile.c_str(),0);    
 99     //文件头信息
100     BITMAPFILEHEADER nbmfHeader;    
101     nbmfHeader.bfType = 0x4D42;
102     nbmfHeader.bfSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)
103         + 256 * sizeof(RGBQUAD) + srcW * srcH;
104     nbmfHeader.bfReserved1 = 0;
105     nbmfHeader.bfReserved2 = 0;
106     nbmfHeader.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD);
107     //Bitmap头信息
108     BITMAPINFOHEADER   bmi; 
109     bmi.biSize=sizeof(BITMAPINFOHEADER); 
110     bmi.biWidth=srcW; 
111     bmi.biHeight=srcH; 
112     bmi.biPlanes=1; 
113     bmi.biBitCount=8; 
114     bmi.biCompression=BI_RGB; 
115     bmi.biSizeImage=0; 
116     bmi.biXPelsPerMeter=0; 
117     bmi.biYPelsPerMeter=0; 
118     bmi.biClrUsed= 256; 
119     bmi.biClrImportant=0; 
120     
121     //写入文件头信息
122     _lwrite(hfile,(LPCSTR)&nbmfHeader,sizeof(BITMAPFILEHEADER));
123     //写入Bitmap头信息
124     _lwrite(hfile,(LPCSTR)&bmi,sizeof(BITMAPINFOHEADER));
125     if (quadDes)
126     {
127         _lwrite(hfile,(LPCSTR)quadDes,sizeof(RGBQUAD) * 256);
128     }
129     //写入图像数据
130     _lwrite(hfile,(LPCSTR)desBuf,desBufSize);
131     _lclose(hfile);
132     if (quad)
133     {
134         delete[] quad;
135         quad = NULL;
136     }
137     if (quadDes)
138     {
139         delete[] quadDes;
140         quadDes = NULL;
141     }
142 }
143 
144 int main(int argc, char* argv[])
145 {
146     string srcFile("e://t.bmp");
147     string desFile("e://gray.bmp");
148     ToGray(srcFile,desFile);
149     system("pause");

150     return 0;
151 }

转载于:https://my.oschina.net/RobinChinaSoft/blog/425544