发明内容
本发明是要解决现有的分割算法无法有效解决相互靠近的多目标分割的技术问题,从而提供了一种基于三值化图像聚类的快速多目标分割方法。
一种基于三值化图像聚类的快速多目标分割方法是按以下步骤实现:
一、计算彩色图像的梯度
(1)通过摄像头对目标摄像采样得到原始的r、g、b三通道图像A,然后对图像A缩小后得到彩色图像B,彩色图像B的大小w×h,w为图像宽度上的像素点个数,h为图像高度上的像素个数;
(2)计算步骤(1)中彩色图像B的r、g、b三通道图像上的方向梯度;
(3)结合步骤(2)中彩色图像B的r、g、b三通道图像上的方向梯度计算彩色图像B两个轴向上的方向梯度;
(4)结合步骤(3)中两个轴向上的方向梯度描述彩色图像B的方向梯度强度图像M;
(5)计算步骤(1)中彩色图像B的饱和度,得到饱和度图像Z;
二、结合步骤一中(4)方向梯度强度图像M和(5)饱和度图像Z的信息对彩色图像B进行阈值处理,将彩色图像逐个像素映射为“前景、轮廓、背景”三个值,通过下式,得到三值化图像C;
其中,100表示“轮廓”,255表示“前景”,0表示“背景”;Mthreshold为梯度阈值,大于等于该阈值代表该像素是轮廓点,小于该阈值代表该像素不是轮廓点;Sthreshold为饱和度阈值,大于该阈值表示像素点可能是具有鲜艳颜色的目标点;
三、对步骤二中三值化图像C进行基于形态学的聚类,得到聚类结果图像E;
四、结合步骤三中聚类结果图像E进行目标区域修复及目标分割。
本发明包括以下有益效果:
本发明组合地采用了包含轮廓信息的三值化方法、基于形态学的聚类方法、和基于扫描方法的边缘提取方法,使得本发明提出的多目标分割方法相比现有的多目标分割检测技术,具有快速性和准确性双重优点,并且能够在多目标靠近的情况下有效地分割出各个目标。
具体实施方式一:结合图1、图2和图3说明本实施方式,本实施方式基于三值化图像聚类的快速多目标分割方法是按以下步骤实现:
一、计算彩色图像的梯度
(1)通过摄像头对目标摄像采样得到原始的r、g、b三通道图像A,然后对图像A缩小后得到彩色图像B,彩色图像B的大小w×h,w为图像宽度上的像素点个数,h为图像高度上的像素个数;
(2)通过以下公式计算步骤(1)中彩色图像B的r、g、b三通道图像上的方向梯度;
其中,x表示像素点的横坐标,y表示像素点的纵坐标,n表示图像的通道标号,表示第n个通道图像的X轴的方向梯度,表示第n个通道图像的Y轴的方向梯度;
(3)通过以下公式,结合步骤(2)中彩色图像B的r、g、b三通道图像上的方向梯度计算彩色图像B两个轴向上的方向梯度;
其中,表示彩色图像B上X轴的方向梯度,表示彩色图像B上Y轴的方向梯度;
(4)通过以下公式,结合步骤(3)中两个轴向上的方向梯度描述彩色图像B的方向梯度强度图像M:
其中,M(x,y)表示(x,y)坐标处的方向梯度强度;
(5)通过以下公式计算步骤(1)中彩色图像B的饱和度,得到饱和度图像Z;
其中,Z(x,y)表示(x,y)坐标处的像素点的饱和度;
二、结合步骤一中(4)方向梯度强度图像M和(5)饱和度图像Z的信息对彩色图像B进行阈值处理,将彩色图像逐个像素映射为“前景、轮廓、背景”三个值,通过下式,得到三值化图像C;
其中,100表示“轮廓”,255表示“前景”,0表示“背景”;Mthreshold为梯度阈值,大于等于该阈值代表该像素是轮廓点,小于该阈值代表该像素不是轮廓点;Sthreshold为饱和度阈值,大于该阈值表示像素点可能是具有鲜艳颜色的目标点;
三、对步骤二中三值化图像C进行基于形态学的聚类,得到聚类结果图像E:在三值化图像的搜索过程中使用了“3乘3”的搜索方块,具体过程如下:
(1)为每个“前景”像素点建立一个是否被分类的标志位属性:“未被聚类”和“已被聚类”,并全部初始化为“未被分类”;用与图像C大小相等的二值化图像D来记录对应点的属性值,为每个“前景”点建立一个无符号整形的类别标志属性,初始时全部初始为“0”,并用与图像C等大小的图像E来记录该属性值;建立一个空的堆栈S,设置搜索的起始位置p0为图片C的左上角的第一个像素点,初始化类别标志号i为“0”;
(2)类别标志号i=i+1;在图像C上继续p0的位置按行顺序往后查找一个“未被聚类”的“前景”像素点p;如果已经搜索完了所有的点则跳到步骤(6),否则记录p0=p为下一次搜索的起始位置,然后进入步骤(3);
(3)将查找到的那个“前景”像素点p压入到堆栈S中;
(4)如果堆栈S是空的则返回第(2)步,否则从堆栈S中取栈顶元素q,在图像E中将其的类别标志位属性设置为i,并在图像D中标记q为“已被分类”;
(5)查询像素点q上、下、左、右、左上、左下、右上、右下的这八个邻域点,对于上点、下点、左点、右点,如果是“未被分类”且是“前景”点,则压入到堆栈S中;完成邻域点的压栈工作后跳回至步骤(4);
(6)得到初步聚类结果图像,类标志位属性相同的像素点就属于同一聚类;计算每个聚类的像素点的个数,将像素点总数过小聚类删除,即将其中的每一个像素点的类别标志属性设置为“0”,得到聚类结果图像E;
四、结合步骤三中聚类结果图像E进行目标区域修复及目标分割,是按以下步骤进行:
(1)进行目标区域修复:在图像E中对每个聚类k,搜索每一行的属于聚类k的最左边的像素点下标L表示最左边,搜索这行属于聚类k的最右边的像素点下标R代表最右边,然后将这行中存在与和中间的点,也归入到聚类k中;然后对每一列、每一条45°斜线、每一条-45°斜线都执行类似的操作,最终得到修复好的目标区域图像F;
(2)进行目标边缘检测:在图像F中对每个聚类k,搜索每一行最左端、最右端,每一列最上端、最下端的点,这些点的集合即为聚类k的轮廓,最后得到只包含轮廓信息的目标边缘图像G;
(3)按照目标边缘图像G所描绘的区域,在图像B中提取各个目标的彩色图像区域得到目标的分割效果图H,从而实现最后的多目标分割。
彩色图像梯度强度图像M如图5所示,从图5可以看出,通过本专利的彩色梯度计算方法能够清晰地表达彩色图像中的轮廓信息;
三值化图像C如图6所示,从图6可以看出,本专利的三值化处理不仅仅将目标区域显示了出来,同时还将图5中的轮廓信息也表达出来了;
聚类结果图像E如图7所示,从图7可以看出,通过聚类处理之后,连通在一起的像素点能够全部聚成一类,从而将不同餐具划分出来,虽然有两个颜色一致的餐具重叠在一起了,但是本专利方法还是能够有效地将两者分割;
修复好的目标区域图像F如图8所示,从图8可以看出经过目标区域修复及聚类之后,破损的目标区域得到了修复;
目标边缘图像G如图9所示,从图9可知,目标边缘检测方法能够有效地提取出边缘像素点;
分割效果图H如图10所示,从图10可知,盘中的餐具被很好的分割出来。