图像分割之sauvola

最新推荐文章于 2023-03-24 16:59:12 发布

大海里捡宝贝

最新推荐文章于 2023-03-24 16:59:12 发布

阅读量446

点赞数

分类专栏：图像分割

本文链接：https://blog.csdn.net/FLY_CAFFE/article/details/84783038

版权

图像分割专栏收录该内容

3 篇文章 0 订阅

订阅专栏

该代码实现了一个sauvola算法的函数，用于图像分割。它首先创建积分图像和平方积分图像，然后计算窗口内的平均值和标准差，通过调整阈值进行二值化处理。算法涉及到了积分图像的计算和区域平均值与标准差的求解。

摘要由CSDN通过智能技术生成

void sauvola(unsigned char * grayImage, unsigned char * biImage, int w, int h, int k, int windowSize)
{
int whalf = windowSize >> 1;

   int i, j;
   int IMAGE_WIDTH = w;
   int IMAGE_HEIGHT = h;
   // create the integral image
   unsigned long * integralImg = (unsigned long*)malloc(IMAGE_WIDTH*IMAGE_HEIGHT*sizeof(unsigned long*));
   unsigned long * integralImgSqrt = (unsigned long*)malloc(IMAGE_WIDTH*IMAGE_HEIGHT*sizeof(unsigned long*));
   int sum = 0;
   int sqrtsum = 0;
   int index;
   for (i = 0; i<IMAGE_HEIGHT; i++)
   {
       // reset this column sum
       sum = 0;
       sqrtsum = 0;

       for (j = 0; j<IMAGE_WIDTH; j++)
       {
           index = i*IMAGE_WIDTH + j;

sum += grayImage[index];
sqrtsum += grayImage[index] * grayImage[index];

           if (i == 0)
           {
               integralImg[index] = sum;
               integralImgSqrt[index] = sqrtsum;
           }
           else
           {
               integralImgSqrt[index] = integralImgSqrt[(i - 1)*IMAGE_WIDTH + j] + sqrtsum;
               integralImg[index] = integralImg[(i - 1)*IMAGE_WIDTH + j] + sum;
           }
       }
   }

   //Calculate the mean and standard deviation using the integral image
   int xmin, ymin, xmax, ymax;
   double mean, std, threshold;
   double diagsum, idiagsum, diff, sqdiagsum, sqidiagsum, sqdiff, area;

   for (i = 0; i<IMAGE_WIDTH; i++){
       for (j = 0; j<IMAGE_HEIGHT; j++){
           xmin = std::max(0, i - whalf);
           ymin = std::max(0, j - whalf);
           xmax = std::min(IMAGE_WIDTH - 1, i + whalf);
           ymax = std::min(IMAGE_HEIGHT - 1, j + whalf);

           area = (xmax - xmin + 1) * (ymax - ymin + 1);
           if (area <= 0)
           {
               biImage[i * IMAGE_WIDTH + j] = 255;
               continue;
           }

           if (xmin == 0 && ymin == 0){
               diff = integralImg[ymax * IMAGE_WIDTH + xmax];
               sqdiff = integralImgSqrt[ymax * IMAGE_WIDTH + xmax];
           }
           else if (xmin > 0 && ymin == 0){
               diff = integralImg[ymax * IMAGE_WIDTH + xmax] - integralImg[ymax * IMAGE_WIDTH + xmin - 1];
               sqdiff = integralImgSqrt[ymax * IMAGE_WIDTH + xmax] - integralImgSqrt[ymax * IMAGE_WIDTH + xmin - 1];
           }
           else if (xmin == 0 && ymin > 0){
               diff = integralImg[ymax * IMAGE_WIDTH + xmax] - integralImg[(ymin - 1) * IMAGE_WIDTH + xmax];
               sqdiff = integralImgSqrt[ymax * IMAGE_WIDTH + xmax] - integralImgSqrt[(ymin - 1) * IMAGE_WIDTH + xmax];;
           }
           else{
               diagsum = integralImg[ymax * IMAGE_WIDTH + xmax] + integralImg[(ymin - 1) * IMAGE_WIDTH + xmin - 1];
               idiagsum = integralImg[(ymin - 1) * IMAGE_WIDTH + xmax] + integralImg[ymax * IMAGE_WIDTH + xmin - 1];
               diff = diagsum - idiagsum;

               sqdiagsum = integralImgSqrt[ymax * IMAGE_WIDTH + xmax] + integralImgSqrt[(ymin - 1) * IMAGE_WIDTH + xmin - 1];
               sqidiagsum = integralImgSqrt[(ymin - 1) * IMAGE_WIDTH + xmax] + integralImgSqrt[ymax * IMAGE_WIDTH + xmin - 1];
               sqdiff = sqdiagsum - sqidiagsum;
           }

           mean = diff / area;
           std = sqrt((sqdiff - diff*diff / area) / (area - 1));
           threshold = mean*(1 + k*((std / 128) - 1));
           if (grayImage[j*IMAGE_WIDTH + i] < threshold)
               biImage[j*IMAGE_WIDTH + i] = 0;
           else
               biImage[j*IMAGE_WIDTH + i] = 255;
       }
   }

free(integralImg);
free(integralImgSqrt);
}