基于小波变换与Otsu阈值去噪的脐橙识别方法

2020-08-05 11:22:42余长庚
华南农业大学学报 2020年5期
关键词:于小波脐橙识别率

余长庚,刘 凯

(1 桂林电子科技大学 信息与通信学院,广西 桂林 541004;2 贺州学院 信息与通信工程学院,广西 贺州 542899)

人工采摘脐橙存在劳动强度大、效率低、成本高等问题,建立自动化、智能化、现代化的采摘作业体系对保证农产品适时采收、降低采摘作业费、实现农业增收具有非常重要的意义。

机器人实现采摘功能,主要在于其能够在复杂的农业环境中准确识别到水果。基于视觉的水果识别主要通过对颜色模型或者灰度直方图等特征的处理来实现[1]。陈科尹等[2]基于视觉显著性和模糊聚类算法提出了水果图像识别方法。Peng等[3]基于机器视觉和SVM提出了多类水果识别研究方法,利用Otsu分割算法对水果图像进行分割并利用SVM分类器对目标进行分类识别。有研究基于K-means算法对原图像进行聚类分割,实现目标识别[4-5]。另外包含 SSD[6-8]、Faster-RCNN[9-10]和 AlexNet[11]的深度 学习算法在水果识别中也有广泛应用,但深度学习算法需要较高的硬件配置,不适用于性能受限的移动机器人。针对农业环境特别是光照、遮挡和表皮颜色等多种情况,多数算法存在对脐橙识别率不佳、背景噪声干扰严重的问题。

针对以上不足,本文统计分析农业环境下脐橙图像的特征信息,提出一种基于小波变换与Otsu阈值去噪的脐橙识别方法,通过YCbCr颜色模型分析与处理,采用基于小波变换的图像分割算法,并设计基于Otsu阈值的去噪算法针对已分割区域进行去噪处理,最后确定脐橙在图像中的位置并通过Matlab软件验证算法的有效性。

1 材料与方法

1.1 脐橙图像YCbCr颜色模型的建立

对颜色的感知,是基于颜色的饱和度、亮度、显著性、对比度和色调等基本属性的。为更有效地处理图像所包含的信息,需要选择合适的颜色模型。颜色模型通常包括RGB模型、HSV模型[12]、LAB模型[13]和YCbCr模型[14]等。相对其他模型,YCbCr模型有更好的对比度,当采集的脐橙图像亮度发生变化时,YCbCr模型色度的变化范围很小,更有利于农业环境中脐橙图像的后续处理。YCbCr模型中Y指色彩亮度分量,Cb指蓝色分量与亮度值之间的差异,Cr指红色分量与亮度值之间的差异。根据公式(1)将RGB模型转化为YCbCr模型。

式中,将RGB格式的原始脐橙图像转化为YCbCr模型,并提取各颜色模型的分量。通过对比分析,Cr分量的直方图有明显的波峰和波谷形态,有利于图像分割。图1为基于原始图像的YCbCr模型的分析图。

图1 原始图像的YCbCr模型分析图Fig.1 Analysis of the YCbCr model of the original image

1.2 基于小波变换的图像分割算法设计

本文利用小波变换提取脐橙的主要特征,参考李月娥等[15]的方法。首先对脐橙原始图像的Cr分量图像进行小波变换得到一系列分辨率逐次降低的子图像,每一级子图像均包含低频子图像和高频子图像,选取最低级的低频子图像并利用模糊C均值聚类算法 (Fuzzy C-means algorithm,FCM)进行分割,将低一级的分割结果扩展后应用于上一级图像,一直到最高级图像为止,得到图像的分割结果。

1.2.1 基于小波变换的脐橙特征提取方法 小波变换在时域和频域同时拥有多尺度分辨性的分析特性,有利于图像分割[16]。长(M)宽(N)尺寸的脐橙图像f(x,y)的离散小波变换原理如公式(2)和(3)所示:

式中,i代表水平(H)、垂直(V)和对角线(D)方向;m为水平平移;n为垂直平移;WL(j,m,n)为分解尺度j下脐橙图像的低频系数;WH(i)(j,m,n)为分解尺度j下脐橙图像在3个方向上的高频系数;Lj,m,n(x,y)表示对缩放函数进行缩放、平移变换;H(i)j,m,n(x,y)表示对3个方向上的小波函数进行缩放、平移变换。

利用小波变换将低通滤波器和高通滤波器同时作用于脐橙图像水平和垂直2个方向,产生频带子图像。低频子图像WL包含脐橙原始图像主要特征,而其他3个高频子图像包含脐橙图像次要特征,当继续对WL重复上述过程时可得到4个次一级的子图像,如图2所示。

图2 小波变换结构示意图Fig.2 Schematic diagram of wavelet transform structure

经过小波变换后,选择包含脐橙图像主要特征的低频子图像WL并利用FCM算法进行图像分割。

1.2.2 基于FCM算法的图像分割方法 FCM算法会计算每个像素点对所有类的隶属度,有效避免背景对目标脐橙的干扰,适合处理存在不确定性特点的图像。试验利用FCM算法将脐橙的低频子图像WL中n个像素点xi(i=1,2,···,n)分为c个模糊类,当各个像素点与聚类中心的误差平方和J(U,V)满足设定的阈值并计算得到像素点对聚类中心最小隶属度uik时 ,得到脐橙图像的聚类中心vi及分类特性。误差平方和J(U,V)的表达式如公式(4)所示。

式中,c为图像的聚类数,c=2;uik表示xi属于第k类区域的隶属函数;m为权重指数;U=[uik]是模糊归类矩阵;V={v1,v2,L,vc}是聚类中心矩阵;dik(xk,vi)为第i类的样本xk到第i类聚类中心vi的距离。dik(xk,vi)定义如公式(5)所示。

通过拉格朗日乘子法分别计算,并修正聚类中心值vi及各个像素点对聚类中心的隶属度uik,完成脐橙图像的分割,如公式(6)和(7)所示。

1.3 基于Otsu阈值的去噪算法设计

为分析脐橙与噪声干扰在图像中的信息,本文将原始图像与小波变换区域叠加进行目标区域颜色恢复,并针对小波变换区域设计Otsu阈值算法进行去噪。由于已分割部分以脐橙为主,使得区域中背景干扰与目标更加清晰分明。

提取目标区域的N个像素点并记录其灰度值,构成灰度值一维数组S(s0、s1、···、sn),将其位置记录在对应的数组L[(r1,c1)、(r2,c2)、···、(rn,cn)]中。灰度值为s的像素点在图像中占比P(s)为:

式中,n(s)代 表灰度值为s(0

S1和S2占脐橙图像像素点总数的比例p1(t)和p2(t)如公式(11)、(12)所示。

S1和S2所有像素点的灰度值总和I1(t)和I2(t)如公式(13)、(14)所示。

S1和S2的均值 µ1(t)和 µ2(t)如公式(15)、(16)所示。

同理,整幅图像的均值(µ)如公式(17)所示。

阈值为t时,背景与脐橙的类间方差如公式(18)所示。

在 0 ~S−1范围内选取拥有最大类间方差值的t作为Otsu算法的最终阈值(To),计算公式(19)为:

由于每幅图像目标脐橙的颜色均不同,且存在泛青色的情况,不能够确定单一不变的图像阈值(T)。针对这一情况,经过多次试验得出T=0.6t时,有较好的背景去噪效果。

由于背景噪声与识别目标颜色相近,处理后仍无法完全消除噪声,针对这一现象,拟采用质心补圆法进一步减少背景噪声并确定脐橙所在圆的圆心和半径。

1.4 基于质心补圆法的目标定位算法设计

为实现图像中目标脐橙的定位,并解决去噪问题,本文在Otsu阈值去噪算法基础上提出一种质心补圆法 ( Circle filling method based on the center of mass,CFCM),以去除分割区域遗留的噪声并确定脐橙在图像中的位置。

质心补圆法依据质心确定方程在分割范围内确定脐橙质心位置,并以质心为检测圆的圆心逐步扩大检测半径,最终得到符合像素点占比要求的半径(r)。设定初始检测半径r=2并以2为步进单位进行迭代,当检测圆内脐橙像素点占比达到分割后区域像素点总数的75%时,输出质心坐标以及半径r,以便为实际采摘过程预留出安全作业空间。

确定脐橙质心后计算检测圆半径,设α为检测圆半径内脐橙像素点数百分比,m为二值图像像素点灰度值为255的点的数目,n为二值图像中灰度值为255且与质心距离不大于半径r的像素点数目,α计算如公式(22)所示。

最后将检测圆圆心与半径绘制到脐橙原始图像中,确定脐橙位置。

脐橙实际形状近似为圆形,但不是标准圆形,所以设定要求算法所确定的圆心在目标中央位置,并且检测圆圈定范围包含95%以上的目标;检测圆的圈定半径不能够超出目标半径的一半。在误差允许范围内,可通过控制操作对目标进行预处理,去除遮挡,以便对脐橙进行更加精确地定位和采摘。

2 结果与分析

为验证算法性能,于广西正丰集团水果出口加工中心采集包含脐橙的原始图像共416幅,按照光照条件、遮挡情况和表皮颜色3部分进行数据统计。其中光照情况分为顺光和逆光,分别为284与132幅;遮挡情况分为遮挡和非遮挡,分别为196与220幅;脐橙表皮颜色情况分为泛青色和橙色,分别为124与292幅。所有采集的原始图像像素为 4 032×3 024,为符合实际处理要求,统一采用双线性内插法将图像压缩至640×480。试验设备为装有Intel Core i7处理器的联想台式电脑,采用Windows 10系统并搭载Matlab2018b软件。

利用Matlab结合上述算法对不同光照条件、不同遮挡情况和不同表皮颜色状况下的脐橙图像进行目标识别,并选择3种典型情况的脐橙图像进行识别算法分析,分别将脐橙原始图像转化为YCbCr颜色模型,并提取其Cr分量图像进行基于小波变换的分割处理,针对已分割区域采用Otsu阈值算法进行去噪处理,而后确定脐橙位置,具体识别情况见图3。

图3 不同情况下基于小波变换与Otsu阈值去噪的脐橙识别方法分析Fig.3 Analysis of navel orange recognition method based on wavelet transform and Otsu threshold denoising under different conditions

YCbCr颜色模型下基于小波变换的分割方法的效果分别如图3a2、3b2和3c2所示,在小波分割过程中农业环境的复杂性导致运行基于小波变换的FCM算法平均消耗时间最长,约0.5 s。然后根据原始图像将其进行图像颜色恢复,由于农业环境下脐橙与背景的灰度值相互混叠,算法无法找到合适的阈值对脐橙进行准确分割而存在大量背景噪声干扰,如图3a3、3b3与3c3所示。针对已分割区域设计Otsu阈值去噪方法以减少背景噪声干扰,并依据经验取图像阈值T=0.6t,图像去噪处理如图3a4、3b4与3c4所示,其中去噪过程中运行平均时间约为0.013 s。最后设计质心补圆法在去噪处理的基础上确定脐橙在二维图像中的位置信息,如图3a5、3b5与3c5所示,其中定位过程运行时间约为0.324 s。基于本文算法的脐橙图像识别、去噪和定位处理平均消耗总时间约为0.840 s。

在保证算法运行速度的情况下,不同遮挡情况、光照条件和表皮颜色条件下的脐橙识别率统计数据如表1所示。

由表1可知,算法对顺光照射下的脐橙有较好的识别效果,而在逆光照射情况下脐橙所处环境较暗,而且可能会在图像中形成高亮度光斑导致脐橙识别率相对顺光条件下较低(90.15%),低于顺光情况下的脐橙识别率(92.96%)。脐橙与背景颜色的差异导致表皮颜色不同的脐橙识别结果不同。对于正常橙色,在顺光和逆光条件下脐橙识别率分别达到95.81%和91.09%,不考虑光照影响的情况下脐橙识别率为94.18%;而对于泛青色脐橙,目标脐橙与背景枝叶的颜色相似导致识别率相对较低,在2种光照条件下均为87.10%。仅考虑遮挡和非遮挡情况时,脐橙识别率分别为90.82%和93.18%。试验数据表明,YCbCr颜色模型下的小波变换与Otsu阈值去噪方法总体识别率为92.07%,能有效地去除识别过程中产生的噪声干扰。

表1 脐橙识别率统计结果Table 1 Statistics of navel orange recognition rate %

3 结论

本文针对农业环境中存在的不同遮挡情况、光照条件和表皮颜色条件下的脐橙识别精度不佳的问题,提出一种基于小波变换和Otsu阈值去噪的识别算法,通过小波变换算法对复杂农业环境下的图像进行粗略分割,然后针对已分割区域进行Otsu阈值去噪处理,最后针对图像分割区域设计一种质心补圆法确定脐橙位置,同时消除算法产生的噪声干扰。试验结果表明,基于小波变换与Otsu阈值去噪的脐橙识别算法能有效解决复杂农业环境下的噪声干扰和脐橙识别精度不佳的问题。

猜你喜欢
于小波脐橙识别率
脐橙连上物联网 扫码便知“前世今生”
今日农业(2022年3期)2022-11-16 13:13:50
基于类图像处理与向量化的大数据脚本攻击智能检测
计算机工程(2020年3期)2020-03-19 12:24:50
基于真耳分析的助听器配戴者言语可懂度指数与言语识别率的关系
基于小波去噪的称重雨量数据分析
一种新的基于小波基的时变信道估计
铁道学报(2018年5期)2018-06-21 06:21:08
提升高速公路MTC二次抓拍车牌识别率方案研究
基于小波和Hu 矩的飑线雷达回波识别
赣南早脐橙在几种中间砧木上高接换种的表现
现代园艺(2017年21期)2018-01-03 06:41:45
奉节脐橙
天津诗人(2017年2期)2017-03-16 03:09:39
高速公路机电日常维护中车牌识别率分析系统的应用