实验中需要用到区域联通的算法,就是类似于matlab中bwlabel的函数。网上找了找c++源码未果,bwlabel-python版用python描述了matlab中的实现方法,但是最后对标签的处理部分并未看明白,故自己用c++实现了一个。先直接看bwlabel函数代码:

cv::Mat bwlabel(const cv::Mat in, int * num, const int mode)
{
const int num_runs = number_of_runs(in);
int * sc = new int[num_runs];
int * ec = new int[num_runs];
int * r = new int[num_runs];
int * labels = new int[num_runs];
memset(labels, 0, sizeof(int)*num_runs);
fill_run_vectors(in, sc, ec, r);
first_pass(sc, ec, r, labels, num_runs, mode);
cv::Mat result = cv::Mat::zeros(in.size(), CV_8UC1); int number = 0;
for(int i = 0; i < num_runs; i++)
{
uchar * p_row = result.ptr<uchar>(r[i]);
for(int j = sc[i]; j <= ec[i]; j++)
p_row[j] = labels[i];
if(number < labels[i])
number = labels[i];
}
if(num != NULL)
*num = number;
delete [] sc;
delete [] ec;
delete [] r;
delete [] labels;
return result;
}

bwlabel中要用到三个辅助函数:number_of_runs,fill_run_vectors,first_pass。函数number_of_runs计算每一行中非零像素团的个数并累加起来。

1 1 0 0 0 1 1 1 0 0

比如,上面这一行就有2个非零像素团,我们称这样的像素团为Run,函数number_of_runs实现如下:

int number_of_runs(const cv::Mat in)
{
const int rows = in.rows;
const int cols = in.cols;
int result = 0;
for(int row = 0; row < rows; row++)
{
const uchar * p_row = in.ptr<uchar>(row);
if(p_row[0] != 0)
result++;
for(int col = 1; col < cols; col++)
{
if(p_row[col] != 0 && p_row[col-1] == 0)
result++;
}
}
return result;
}

这个函数算法思想是,扫描每一行,对每一行,如果当前元素非零并且前一元素为零则Run的个数加一。

函数fill_run_vectors的作用是填充三个数据结构:sc[],ec[],r[],它们分别表示开始列标、结束列标和行标,数组长度为由number_of_runs函数得到的Run的个数。函数fill_run_vectors实现如下:

void fill_run_vectors(const cv::Mat in, int sc[], int ec[], int r[])
{
const int rows = in.rows;
const int cols = in.cols;
int idx = 0;
for(int row = 0; row < rows; row++)
{
const uchar * p_row = in.ptr<uchar>(row);
int prev = 0;
for(int col = 0; col < cols; col++)
{
if(p_row[col] != prev)
{
if(prev == 0)
{
sc[idx] = col;
r[idx] = row;
prev = 1;
}
else
{
ec[idx++] = col - 1;
prev = 0;
}
}
if(col == cols-1 && prev == 1)
{
ec[idx++] = col;
}
}
}
}

算法思想还是遍历每一行,用变量prev保存一行中上一个团是0还是1,如果出现01跳变那么就要记录下新的Run的开始列标和行标,如果出现10跳变(或者这行结束并且prev=1)那么就记录下这个Run的结束列标。

函数first_pass顾名思义,字面上说第一次扫描。因为函数扫描每一个Run块,给它打标签。当出现如下情况时:

1 1 0 0 1 1 1 0
0 1 1 1 1 0 0 0

函数给第一行第一个Run打上标签1,第二个Run打上标签2,当遍历到第二行时,发现这一行的一个Run与第一行第一个Run相邻,故打上标签1,但当继续遍历时发现这个Run也与第一行第二个Run相邻,但函数并没有改变第一行第二个Run的标签,而是记录下这两个标签其实该一样。遍历完第二行结果为:

1 1 0 0 2 2 2 0
0 1 1 1 1 0 0 0

遍历完每一个Run过后就是处理刚才未处理的标签了。函数first_pass实现如下:

void first_pass(const int sc[], const int ec[], const int r[],int labels[], const int num_runs, const int mode)
{
int cur_row = 0;
int next_label = 1;
int first_run_on_prev_row = -1;
int last_run_on_prev_row = -1;
int first_run_on_this_row = 0;
int offset = 0;
int * equal_i = new int[num_runs];
int * equal_j = new int[num_runs];
int equal_idx = 0;
if(mode == 8)
offset = 1;
for(int k = 0; k < num_runs; k++)
{
if(r[k] == cur_row + 1)
{
cur_row += 1;
first_run_on_prev_row = first_run_on_this_row;
first_run_on_this_row = k;
last_run_on_prev_row = k - 1;
}
else if(r[k] > cur_row + 1)
{
first_run_on_prev_row = -1;
last_run_on_prev_row = -1;
first_run_on_this_row = k;
cur_row = r[k];
}
if(first_run_on_prev_row >= 0)
{
int p = first_run_on_prev_row;
while(p <= last_run_on_prev_row && sc[p] <= (ec[k] + offset))
{
if(sc[k] <= ec[p] + offset)
{
if(labels[k] == 0)
labels[k] = labels[p];
else if(labels[k] != labels[p])
{
//labels[p] = labels[k];
equal_i[equal_idx] = labels[k];
equal_j[equal_idx] = labels[p];
equal_idx += 1;
}
}
p += 1;
}
}
if(labels[k] == 0)
{
labels[k] = next_label++;
}
}
/////////////////////// process labels
for(int i = 0; i < equal_idx; i++)
{
int max_label = equal_i[i] > equal_j[i] ? equal_i[i] : equal_j[i];
int min_label = equal_i[i] < equal_j[i] ? equal_i[i] : equal_j[i];
for(int j = 0; j < num_runs; j++)
{
if(labels[j] == max_label)
labels[j] = min_label;
}
}
delete [] equal_i;
delete [] equal_j;
/////////////////////process ignore labels
int * hist = new int[next_label];
int * non_labels = new int[next_label];
memset(hist, 0, sizeof(int)*next_label);
int non_num = 0;
for(int i = 0; i < num_runs; i++)
{
hist[labels[i]]++;
}
for(int i = 1; i < next_label; i++)
{
if(hist[i] == 0)
non_labels[non_num++] = i;
}
for(int j = 0; j < num_runs; j++)
{
int k = labels[j];
for(int i = non_num-1; i >= 0; i--)
{
if(k > non_labels[i])
{
labels[j] -= (i+1);
break;
}
}
}
delete [] hist;
delete [] non_labels;
}

前面遍历每一个Run分两种情况,上一行有Run和上一行无Run:当上一行无Run时就分配一个新的标签,当上一行有Run时还要考虑是否与上一行Run相邻,若相邻则打上上一行的标签,当出现上面讲到的情况时就保存这两个标签到数组equal_i,equal_j中。

接下来就是处理equal_i和equal_j这两个数组了,要将它们当中相同族的不同标签合并到一起(注释process labels下面代码)。

这样过后还不能完事,有可能出现标签间断的现象(如1,2,4,6),就是还必须把标签(如1,2,4,6)映射到一个连续的空间(1,2,3,4)。参见注释process ignore labels以下代码。

这样过后就差不多了,最后一步是在bwlabel中给返回的Mat中元素打上对应的标签。

@waring

bwlabel函数的c++实现的更多相关文章

  1. Python实现MATLAB中的 bwlabel函数

    最近做验证码识别,原本用MATLAB已经实现的整个识别模型,不过代码要部署在Linux服务器上还是需要用另外的语言实现,于是决定用Python + OpenCV来实现. bwlabel函数的作用是检测 ...

  2. Matlab图像处理函数:regionprops

    本篇文章为转载,仅为方便学术讨论所用,不用于商业用途.由于时间较久,原作者以及原始链接暂时无法找到,如有侵权以及其他任何事宜欢迎跟我联系,如有侵扰,在此提前表示歉意.----------------- ...

  3. matlab函数大全

    Matlab 图像处理相关函数命令大全 一.通用函数: colorbar  显示彩色条 语法:colorbar \ colorbar('vert') \ colorbar('horiz') \ col ...

  4. matlab boundaries和fchcode函数无法执行的解决办法 未定义与 'double' 类型的输入参数相对应的函数 'boundaries'

    在测试代码时发现,自己的matlab无法执行Freeman链码函数: boundaries和fchcode函数都无法正常运行: 需要在自己的工作目录中添加如下函数: boundaries   fchc ...

  5. MATLAB中图像处理的函数

    表1 图像显示 函数名 功能说明 函数名 功能说明 colorbar 颜色条显示 montage 按矩形剪辑方式显示多帧图像 getimage 从坐标系中获取图像数据 immovie 从多帧索引图像中 ...

  6. Python 小而美的函数

    python提供了一些有趣且实用的函数,如any all zip,这些函数能够大幅简化我们得代码,可以更优雅的处理可迭代的对象,同时使用的时候也得注意一些情况   any any(iterable) ...

  7. 探究javascript对象和数组的异同,及函数变量缓存技巧

    javascript中最经典也最受非议的一句话就是:javascript中一切皆是对象.这篇重点要提到的,就是任何jser都不陌生的Object和Array. 有段时间曾经很诧异,到底两种数据类型用来 ...

  8. JavaScript权威指南 - 函数

    函数本身就是一段JavaScript代码,定义一次但可能被调用任意次.如果函数挂载在一个对象上,作为对象的一个属性,通常这种函数被称作对象的方法.用于初始化一个新创建的对象的函数被称作构造函数. 相对 ...

  9. C++对C的函数拓展

    一,内联函数 1.内联函数的概念 C++中的const常量可以用来代替宏常数的定义,例如:用const int a = 10来替换# define a 10.那么C++中是否有什么解决方案来替代宏代码 ...

随机推荐

  1. apache2 httpd 基于域名的虚拟主机配置 for centos6X 和debian-8

    全系统虚拟主机: for debian 系统的apache2 域名 虚拟主机

  2. mysql 存储过程:提供查询语句并返回查询执行影响的行数

    mysql 存储过程:提供查询语句并返回查询执行影响的行数DELIMITER $$ DROP PROCEDURE IF EXISTS `p_get_select_row_number`$$ CREAT ...

  3. HDU 2845 Beans (DP)

    Problem Description Bean-eating is an interesting game, everyone owns an M*N matrix, which is filled ...

  4. android layout属性介绍

    android:id 为控件指定对应的ID android:text 指定控件其中显示的文字,须要注意的是,这里尽量使用strings.xml文件其中的字符串 android:gravity 指定Vi ...

  5. IOS 6.0+ Autolayout — UITableViewCell 高度调整

    要实现的效果 要求: storyboard上的准备工作 建好cell自定义AutoCell 类,关联好控件,事先准备好一些数据源 实现tableview的委托方法 核心部分,HeightForRow方 ...

  6. How to center anything with css

    1. 绝对居中定位技术 我经常用margin:0 auto来实现水平居中,而一直认为margin:auto不能实现垂直居中……实际上,实现垂直居中仅需要声明元素高度和下面的CSS 优点:  缺点: 1 ...

  7. VCS仿真 Dump Memory

    VCS仿真 Dump Memory 两种方法 vcs联合verdi生成fsdb文件 vcs生成vpd文件 VCS联合verdi生成fsdb文件 1.testbench中加入如下语句: initial ...

  8. My way to Python - Day04 - 模块

    re模块 什么是正则表达式 正则表达式,英文叫做Regular Expression.简单说,正则表达式就是一组规则,用于实现字符串的查找,匹配,以实现关于字符串的相关操作,比如替换,删除等. 正则表 ...

  9. Web Api Session开启会话支持

        1.WebApi中默认是没有开启Session会话支持的.需要在Global中重写Init方法来指定会话需要支持的类型           //代码如下 public override voi ...

  10. C# Winform中DataGridView的DataGridViewCheckBoxColumn CheckBox选中判断

    1.DataGridViewCheckBoxColumn CheckBox是否选中 在判断DataGridView中CheckBox选中列的时候,用DataGridViewRow.Cells[0].F ...