1 #include "iostream"
2 #include "opencv2/opencv.hpp"
3 #include "vector"
4
5 using namespace std;
6 using namespace cv;
7
8 /*计算真彩色图像的直方图*/
9 void splitRgbImgPro(const Mat img, vector<Mat>& imgRGB);
10 void calImgHistPro(vector<Mat> imgrgb, vector<vector<int>>& histData);
11 void showImgHistPro(vector<vector<int>>& histD);
12 void rotateImgPro(Mat& img, float angle);
13 int main()
14 {
15 Mat imgSrc = imread("E:\\VideoPlateRecog\\ETC\\1008\\SRC\\144.bmp");
16 /*色彩空间分离*/
17 vector<Mat> imgRGB;
18 splitRgbImgPro(imgSrc, imgRGB);
19 /*计算三通道直方图*/
20 vector<vector<int>> histData;
21 calImgHistPro(imgRGB, histData);
22 /*绘制三通道直方图*/
23 showImgHistPro(histData);
24
25 system("pause");
26 return 0;
27 }
28
29 void calImgHistPro(vector<Mat> imgrgb,vector<vector<int>>& histData)
30 {
31 vector<Mat> imgrgbSrc = imgrgb;
32 int width = imgrgb[0].cols;
33 int height = imgrgb[0].rows;
34 for (int n = 0; n < imgrgb.size(); n++)
35 {
36 Mat imgSrc = imgrgbSrc[n].clone();
37 int img_sum = -1;
38 vector<int> p(256);
39 for (int m = 0; m <p.size() ; m++)
40 {
41 p[m] = 0;
42 }
43 for (int i = 0; i < height; i++)
44 {
45 uchar* ptr = imgSrc.ptr<uchar>(i);
46 for (int j = 0; j < width; j++)
47 {
48 int k = ptr[j];
49 p[k]++;
50 }
51 }
52 /*find max element of vector*/
53 for (auto it = p.begin(); it != p.end(); it++)
54 {
55 if (*it > img_sum)
56 {
57 img_sum = *it;
58 }
59 }
60 for (int m = 0; m < 256; m++)
61 {
62 p[m] = p[m]*255 / img_sum;
63 }
64 histData.push_back(p);
65 }
66 }
67
68 void showImgHistPro(vector<vector<int>>& histD)
69 {
70 vector<vector<int>> histSrc=histD;
71 int cols = histD[0].size();
72 int rows = -1;
73 Mat histRGB(cols , cols * 3, CV_8UC1, Scalar(0));
74 for (int n = 0; n < histD.size(); n++)
75 {
76 vector<int> hist = histD[n];
77 Mat histImg(cols, cols, CV_8UC1, Scalar(0));
78 for (int i = 0; i < cols; i++)
79 {
80 uchar* ptr = histImg.ptr(i);
81 for (int j = 0; j < hist[i]; j++)
82 {
83 ptr[j] = 255;
84 }
85 }
86 /*旋转90°*/
87 Mat histImg1(cols, cols, CV_8UC1, Scalar(0));
88 for (int l = 0; l < cols; l++)
89 {
90 for (int k = 0; k < cols; k++)
91 {
92 histImg1.at<uchar>(l, k) = histImg.at<uchar>(k, cols-l-1);
93 }
94 }
95 histImg1.copyTo(histRGB(Rect(cols*n,0, cols, cols)));
96 }
97 imshow("Img_R_G_B", histRGB);
98 waitKey(0);
99 }
100
101 void rotateImgPro(Mat& img,float angle)
102 {
103 Mat retMat(img.cols, img.rows, CV_8UC1, Scalar(0));
104 float anglePI = (float)(angle*CV_PI / 180);
105 int xSm, ySm;
106 for (int i = 0; i < retMat.rows; i++)
107 {
108 for (int j = 0; j < retMat.cols; j++)
109 {
110 xSm = (int)((i - retMat.rows / 2)*cos(anglePI) -
111 (j - retMat.cols / 2)*sin(anglePI) + 0.5);
112 ySm = (int)((i - retMat.rows / 2)*sin(anglePI) +
113 (j - retMat.cols / 2)*cos(anglePI) + 0.5);
114 retMat.at<uchar>(i, j) = img.at<uchar>(xSm, ySm);
115
116 }
117 }
118 }
119
120
121
122
123 void splitRgbImgPro(const Mat img, vector<Mat>& imgRGB)
124 {
125 Mat imgSrc = img.clone();
126 int width = imgSrc.rows;
127 int height = imgSrc.cols;
128 Mat imgR(width, height, CV_8UC1);
129 Mat imgG(width, height, CV_8UC1);
130 Mat imgB(width, height, CV_8UC1);
131 for (int i = 0; i < width; i++)
132 {
133 Vec3b* imgPtr = imgSrc.ptr<Vec3b>(i);
134 uchar* imgSPtr = imgSrc.ptr<uchar>(i);
135 uchar* imgRPtr = imgR.ptr<uchar>(i);
136 uchar* imgGPtr = imgG.ptr<uchar>(i);
137 uchar* imgBPtr = imgB.ptr<uchar>(i);
138 for (int j = 0; j < height; j++)
139 {
140 imgRPtr[j] = imgSPtr[0];
141 imgGPtr[j] = imgSPtr[1];
142 imgBPtr[j] = imgSPtr[2];
143 imgSPtr += 3;
144 }
145 }
146 imgRGB.push_back(imgR);
147 imgRGB.push_back(imgG);
148 imgRGB.push_back(imgB);
149 }
