读书人

基于图像分块的图像瓜分

发布时间: 2013-02-24 17:58:56 作者: rapoo

基于图像分块的图像分割
阈值处理直观、实现简单且计算速度快,因此图像阈值处理在图像分割中处于核心地位。阈值处理可分为全局阈值处理和局部阈值处理(可变阈值处理)。本人在http://blog.csdn.net/wuhaibing_cver/article/details/8473498一文中介绍了基于OTSU的全局阈值处理,由于噪声和非均匀光照对阈值处理算法的性能起着重要作用,所以有时候利用全局阈值处理进行图像分割往往会导致错误的分割结果。

基于图像分块的局部图像分割。图像分块属于可变阈值处理中简单的方法,这种方法可补偿光照的不均匀性。选择的矩形要足够小,以便每个矩形的光照都是近视均匀的。需要强调的是,不能把图像分的过细,因为细分可能只包含物体或背景像素。

下面给出该方法的源代码,使用的工具为VS2008+OpenCV2.0.


  1. #include <cv.h>
  2. #include <highgui.h>
  3. #include <iostream>
  4. using namespace std;
  5. int otsu(const IplImage *image)
  6. {
  7. assert(NULL != image);
  8. int width = image->width;
  9. int height = image->height;
  10. int x=0,y=0;
  11. int pixelCount[256];
  12. float pixelPro[256];
  13. int i, j, pixelSum = width * height, threshold = 0;
  14. uchar* data = (uchar*)image->imageData;
  15. //初始化
  16. for(i = 0; i < 256; i++)
  17. {
  18. pixelCount[i] = 0;
  19. pixelPro[i] = 0;
  20. }
  21. //统计灰度级中每个像素在整幅图像中的个数
  22. for(i = y; i < height; i++)
  23. {
  24. for(j = x;j <width;j++)
  25. {
  26. pixelCount[data[i * image->widthStep + j]]++;
  27. }
  28. }
  29. //计算每个像素在整幅图像中的比例
  30. for(i = 0; i < 256; i++)
  31. {
  32. pixelPro[i] = (float)(pixelCount[i]) / (float)(pixelSum);
  33. }
  34. //经典ostu算法,得到前景和背景的分割
  35. //遍历灰度级[0,255],计算出方差最大的灰度值,为最佳阈值
  36. float w0, w1, u0tmp, u1tmp, u0, u1, u,deltaTmp, deltaMax = 0;
  37. for(i = 0; i < 256; i++)
  38. {
  39. w0 = w1 = u0tmp = u1tmp = u0 = u1 = u = deltaTmp = 0;
  40. for(j = 0; j < 256; j++)
  41. {
  42. if(j <= i) //背景部分
  43. {
  44. //以i为阈值分类,第一类总的概率
  45. w0 += pixelPro[j];
  46. u0tmp += j * pixelPro[j];
  47. }
  48. else //前景部分
  49. {
  50. //以i为阈值分类,第二类总的概率
  51. w1 += pixelPro[j];
  52. u1tmp += j * pixelPro[j];
  53. }
  54. }
  55. u0 = u0tmp / w0; //第一类的平均灰度
  56. u1 = u1tmp / w1; //第二类的平均灰度
  57. u = u0tmp + u1tmp; //整幅图像的平均灰度
  58. //计算类间方差
  59. deltaTmp = w0 * (u0 - u)*(u0 - u) + w1 * (u1 - u)*(u1 - u);
  60. //找出最大类间方差以及对应的阈值
  61. if(deltaTmp > deltaMax)
  62. {
  63. deltaMax = deltaTmp;
  64. threshold = i;
  65. }
  66. }
  67. //返回最佳阈值;
  68. return threshold;
  69. }
  70. int main()
  71. {
  72. char* filename = "D:\\technology\\CV\\Databases\\image\\septagon_noisy_shaded.tif";
  73. IplImage* src_image = cvLoadImage(filename,0);
  74. if(!src_image)
  75. return -1;
  76. cvNamedWindow("src",0);
  77. cvShowImage("src",src_image);
  78. CvSize size = cvGetSize(src_image);
  79. int height = size.height, width = size.width;
  80. IplImage* bi_image = cvCreateImage(size,8,1);
  81. #if 0
  82. int threshold = otsu(src_image);
  83. cout<<"threshold: "<<threshold<<endl;
  84. cvThreshold(src_image,bi_image,threshold,255,CV_THRESH_BINARY);
  85. #else
  86. int rows = 2, cols = 3;
  87. //第i行,第j列的左上角topleft,右下角bottomright
  88. CvPoint topleft,bottomright;
  89. CvRect rect;
  90. IplImage* roi_image = 0;
  91. IplImage* bi_roi = 0;
  92. for(int i=0;i<rows;i++)
  93. {
  94. for(int j=0;j<cols;j++)
  95. {
  96. topleft.x = cvRound(j*width/(float)cols);
  97. topleft.y = cvRound(i*height/(float)rows);
  98. bottomright.x = cvRound((j+1)*width/(float)cols);
  99. bottomright.y = cvRound((i+1)*height/(float)rows);
  100. int w = bottomright.x-topleft.x;
  101. int h = bottomright.y-topleft.y;
  102. rect = cvRect(topleft.x,topleft.y,w,h);
  103. cvSetImageROI(src_image,rect);
  104. cvSetImageROI(bi_image,rect);
  105. roi_image = cvCreateImage(cvSize(w,h),8,1);
  106. bi_roi = cvCreateImage(cvSize(w,h),8,1);
  107. cvCopy(src_image,roi_image);
  108. cvResetImageROI(src_image);
  109. int threshold = otsu(roi_image);
  110. cvThreshold(roi_image,bi_roi,threshold,255,CV_THRESH_BINARY);
  111. cvCopy(bi_roi,bi_image);
  112. cvResetImageROI(bi_image);
  113. cvReleaseImage(&bi_roi);
  114. cvReleaseImage(&roi_image);
  115. }
  116. }
  117. #endif
  118. cvNamedWindow("dst",0);
  119. cvShowImage("dst",bi_image);
  120. cvWaitKey(0);
  121. cvReleaseImage(&src_image);
  122. cvReleaseImage(&bi_image);
  123. cvDestroyWindow("src");
  124. cvDestroyWindow("dst");
  125. return 0;
  126. }

源图像如下,可以看出该图像光照不均匀。

基于图像分块的图像瓜分

当把图像分块成2行3列,即rows = 2, cols = 3时,分割结果如下,可以看到该分割结果符合我们的要求。

基于图像分块的图像瓜分

当rows = 3, cols = 4时,由于图像分的过细,某些局部只包含了物体或背景,导致分割失败,如下图所示:

基于图像分块的图像瓜分

读书人网 >图形图像

热点推荐