關于彩色圖像高斯反向投影基于OpenCV的C++代碼

彩色圖像高斯反向投影

一：介紹

圖像反向投影的最終目的是獲取ROI然后實現(xiàn)對ROI區(qū)域的標注、識別、測量等圖像處理與分析，是計算機視覺與人工智能的常見方法之一。圖像反向投影通常是彩色圖像投影效果會比灰度圖像效果要好，原因在于彩色圖像帶有更多對象細節(jié)信息，在反向投影的時候更加容易判斷、而轉(zhuǎn)為灰度圖像會導致這些細節(jié)信息丟失、從而導致分割失敗。最常見的是基于圖像直方圖特征的反向投影。我們這里介紹一種跟直方圖反向投影不一樣的彩色圖像反向投影方法，通過基于高斯的概率分布公式（PDF）估算，反向投影得到對象區(qū)域，該方法也可以看做最簡單的圖像分割方法。缺點是對象顏色光照改變和尺度改變不具備不變性特征。所以需要在光照度穩(wěn)定情況下成像采集圖像數(shù)據(jù)。 在這種情況下使用的高斯概率密度公式為：

1. 輸入模型M，對M的每個像素點（R,G,B）計算I=R+G+B r=R/I, g=G/I, b=B/I

2. 根據(jù)得到權重比例值，計算得到對應的均值 與標準方差

3. 對輸入圖像的每個像素點計算根據(jù)高斯公式計算P(r)與P(g)的乘積

4. 歸一化之后輸出結果，即為最終基于高斯PDF的反向投影圖像

二：算法步驟與代碼實現(xiàn)

1. 首先加載模型圖像與測試圖像

2. 根據(jù)模型圖像計算得到每個通道對應的均值與標準方差參數(shù)

3. 根據(jù)參數(shù)方差計算每個像素點的PDF值

4. 歸一化概率分布圖像-即為反向投影圖像，顯示

5. 根據(jù)Mask得到最終顏色模型對象分割

完整的基于OpenCV的C++代碼如下：

1. #include

2. #include

3. #include

5. using namespace cv;

6. using namespace std;

8. int main(int argc, char** argv) {

9. // 加載模型圖像與測試圖像

10. Mat src = imread("D:/gloomyfish/gc_test.png");

11. Mat model = imread("D:/gloomyfish/gm.png");

12. if (src.empty() || model.empty()) {

13. printf("could not load image... ");

14. return -1;

15. }

16. imshow("input image", src);

18. // 對每個通道 計算高斯PDF的參數(shù)

19. // 有一個通道不計算，是因為它可以通過1-r-g得到

20. // 無需再計算

21. Mat R = Mat::zeros(model.size(), CV_32FC1);

22. Mat G = Mat::zeros(model.size(), CV_32FC1);

23. int r = 0, g = 0, b = 0;

24. float sum = 0;

25. for (int row = 0; row < model.rows; row++) {

26. uchar* current = model.ptr(row);

27. for (int col = 0; col < model.cols; col++) {

28. b = *current++;

29. g = *current++;

30. r = *current++;

31. sum = b + g + r;

32. R.at(row, col) = r / sum;

33. G.at(row, col) = g / sum;

34. }

35. }

37. // 計算均值與標準方差

38. Mat mean, stddev;

39. double mr, devr;

40. double mg, devg;

41. meanStdDev(R, mean, stddev);

42. mr = mean.at(0, 0);

43. devr = mean.at(0, 0);

45. meanStdDev(G, mean, stddev);

46. mg = mean.at(0, 0);

47. devg = mean.at(0, 0);

49. int width = src.cols;

50. int height = src.rows;

52. // 反向投影

53. float pr = 0, pg = 0;

54. Mat result = Mat::zeros(src.size(), CV_32FC1);

55. for (int row = 0; row < height; row++) {

56. uchar* currentRow = src.ptr(row);

57. for (int col = 0; col < width; col++) {

58. b = *currentRow++;

59. g = *currentRow++;

60. r = *currentRow++;

61. sum = b + g + r;

62. float red = r / sum;

63. float green = g / sum;

64. pr = (1 / (devr*sqrt(2 * CV_PI)))*exp(-(pow((red - mr), 2)) / (2 * pow(devr, 2)));

65. pg = (1 / (devg*sqrt(2 * CV_PI)))*exp(-(pow((green - mg),2)) / (2 * pow(devg, 2)));

66. sum = pr*pg;

67. result.at(row, col) = sum;

68. }

69. }

71. // 歸一化顯示高斯反向投影

72. Mat img(src.size(), CV_8UC1);

73. normalize(result, result, 0, 255, NORM_MINMAX);

74. result.convertTo(img, CV_8U);

75. Mat segmentation;

76. src.copyTo(segmentation, img);

78. // 顯示

79. imshow("backprojection demo", img);

80. imshow("segmentation demo", segmentation);

82. waitKey(0);

83. return 0;

84. }

三：測試圖像與效果演示

藍色矩形框為模型，整個圖像為測試圖像

反向投影結果

分割提取結果

四：總結

大家看了這個例子總是有點怪怪的，總會想起點什么，如果你能想起點什么的話就是GMM，高斯混合模型，高斯混合模型正是在此基礎上進一步演化而來。

審核編輯 ：李倩