《Efficient Graph-Based Image Segmentation》c++代码解读

最新推荐文章于 2021-08-29 18:23:21 发布

KirutoCode

最新推荐文章于 2021-08-29 18:23:21 发布

阅读量561

点赞数 3

分类专栏： Detection 文章标签：计算机视觉经典论文

本文链接：https://blog.csdn.net/McEason/article/details/99544497

版权

本文重点解析了基于图的图像分割算法的关键C++代码，涉及参数设置、图的构建、组件合并及颜色填充等步骤。通过高斯平滑处理图像，计算边权重，构建最小生成树，并对小于最小尺寸的组件进行合并，最终实现图像分割并进行颜色填充。若发现错误，欢迎指正。

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

文章目录

本文并没有将全部的代码放上来，只放了些觉得关键的部分。
论文PDF和c++代码下载处

1.需要输入的参数：sigma、阈值k、min(component的最小size)、输入、输出(存放的变量)。

2.生成G=(V，E)

读取图片的信息，RGB三个通道信息分别读取,然后对三个通道进行高斯平滑，去除伪阴影(这块代码还没搞懂，感觉怪怪的，显示一个没见过的正则化，然后高斯过滤的操作也很奇特)

//用了重载函数
static image<float> *smooth(image<float> *src, float sigma) {
   
  std::vector<float> mask = make_fgauss(sigma);
  normalize(mask);
  //mask是滤波器
  image<float> *tmp = new image<float>(src->height(), src->width(), false);
  image<float> *dst = new image<float>(src->width(), src->height(), false);
  //卷积两次，卷积后的结果放在第二个参数中
  convolve_even(src, tmp, mask);
  convolve_even(tmp, dst, mask);


  delete tmp;
  return dst;
}
//mask的len，不在这个位置，但我觉得挺重要，就先放在这里看着
//int len = (int)ceil(sigma * WIDTH) + 1; 

static void convolve_even(image<float> *src, image<float> *dst,
           std::vector<float> &mask) {
   
  int width = src->width();
  int height = src->height();
  int len = mask.size();

  //这个卷积卷的我头皮发麻，为什么一直在一行上操作？？？
  for (int y = 0; y < height; y++) {
   
    for (int x = 0; x < width; x++) {
   
      float sum = mask[0] * imRef(src, x, y);
      for (int i = 1; i < len; i++) {
   
   sum += mask[i] *
     (imRef(src, std::max(x-i,0), y) +
      imRef(src, std::min(x+i, width-1), y));
      }
      imRef(dst, y, x) = sum;
    }