一种基于机器视觉的矿石识别方法
阅读说明：本技术 一种基于机器视觉的矿石识别方法 (Ore identification method based on machine vision ) 是由 吴佳 李明宸 唐文妍 莫海红 郭宇豪 孟江南 于 2020-11-10 设计创作，主要内容包括：本发明公开了一种基于机器视觉的矿石识别方法,包括以下步骤：获取矿石图像并制作成数据集,将数据集中的图像进行标注,分为合格与不合格两类(合格标注为1,不合格标注为0)；将数据集进行预处理,包括去噪、裁剪等操作；对预处理后的图像进行颜色特征与纹理特征的提取,将两种特征进行组合形成新的特征集；采用特征约简算法对组合后的特征集进行特征选择；最后使用机器学习分类算法进行分类识别。本发明形成了一套完整的、基于机器视觉的矿石识别方法,算法简单高效,识别精度高。(The invention discloses an ore identification method based on machine vision, which comprises the following steps: acquiring an ore image, manufacturing the ore image into a data set, and labeling the image in the data set into a qualified type and an unqualified type (the qualified label is 1, and the unqualified label is 0); preprocessing a data set, including operations such as denoising and cutting; extracting color features and texture features of the preprocessed image, and combining the two features to form a new feature set; performing feature selection on the combined feature set by adopting a feature reduction algorithm; and finally, performing classification and identification by using a machine learning classification algorithm. The invention forms a set of complete ore identification method based on machine vision, and the algorithm is simple and efficient, and the identification precision is high.)
技术领域
本发明涉及一种图像识别方法，特别涉及一种基于机器视觉的矿石识别方法。
背景技术
矿石筛选是矿石工艺生产中的重要环节，杂质含量较高的矿石会影响整体矿石的质量，并且不适用于生产建筑用料。传统的矿石分拣方法为人工分选，而人工分选方法存在着操作环境差、生产效率低等缺点。近些年随着计算机视觉技术的发展，基于图像处理的识别技术得到了广泛应用，在图像识别中最关键的一环是图像特征的提取，图像特征主要包括颜色、纹理和形状三种，特征的选用对识别结果有重要的影响。由于矿石具有一些特殊的表面特征，如表面粗糙斑驳，粉尘与矿石颗粒之间的粘连、堆叠等，单个纹理、颜色特征不足以对复杂图像进行有效的识别。
在应用机器学习方法对矿石进行分类时，图像特征的选择对分类识别性能产生至关重要的影响，冗余的特征不仅导致识别速度变慢，而且会使得分类准确度降低，现有的特征提取方法只是针对某一方面进行特征的提取，并没有对特征进行选择，而是将所有特征应用于分类器的识别中，这个将无法很好地满足对于矿石图像的分类。
发明内容
为了解决上述技术问题，本发明提供一种算法简单高效、识别精度高的基于机器视觉的矿石识别方法。
本发明解决上述问题的技术方案是：一种基于机器视觉的矿石识别方法，包括以下步骤：
步骤一：获取矿石图像，将采集的图像进行标注，合格标注为1、不合格标注为0；
步骤二：对数据集中的图像进行预处理，主要包括去噪等操作；
步骤三：提取图像的颜色(RGB)特征与灰度共生矩阵(GLCM)特征，将两种进行特征组合，形成一组新的特征集；
步骤四：将组合后的特征集使用特征约简算法，精简特征集中的特征数；
步骤五：将关键特征集与对应的类型标签一起输入到分类器进行训练。
上述基于机器视觉的矿石识别方法，所述步骤一中，在矿石加工厂现场收集多种矿石图像样本集；对样本集中的矿石图像进行分割并标注，合格标为1，不合格标为0。
上述基于机器视觉的矿石识别方法，所述步骤二中，对样本集中的图像进行预处理，去除成像过程中造成的图像质量损坏，预处理方法包括中值滤波、裁剪等操作。
上述基于机器视觉的矿石识别方法，所述步骤三中，矿石图像的识别包括以下步骤：
Ⅰ)提取样本集中每块矿石图像的颜色(RGB)特征，得到每块矿石图像的R、G、B三个颜色分量的均值；
Ⅱ)提取样本集中每块矿石图像的灰度共生矩阵(GLCM)特征，得到每块矿石图像的对比度、相关性、能量和逆差矩4种纹理特征，通过计算在0°、45°、90°、135°四个方向上对应的值，得到一个16维特征集；
式中，f1、f2、f3、f4分别为对比度、相关性、能量和逆差矩；L为灰度级数；i，j分别表示一幅图像中不同位置的灰度值；d为距离，通常取值为1；θ为方向，取值为0°、45°、90°、135°；u1为变量i的均值，u2为变量j的均值，σ1为变量i的方差，σ2为变量j的方差，
Ⅲ)将提取的颜色特征与灰度共生矩阵特征进行组合，组成一个具有19维的特征集。
上述基于机器视觉的矿石识别方法，所述步骤四中，使用基于邻域粒化及不一致性度量的属性约简算法对组合后的特征进行约简，选择合适的评价函数约简得到能识别矿石图像的关键特征集。
上述基于机器视觉的矿石识别方法，所述步骤五中，将关键特征集与对应的类型标签作为分类器中的训练样本，通过训练获得基于机器视觉的矿石分类模型，并进行实验验证。
本发明的有益效果在于：本发明首先通过在矿石加工厂现场采集矿石图像，将收集到的矿石图像制作成样本集，对样本集中的矿石图像分割后进行标注，分为合格与不合格两类；然后对图像进行预处理，消除获取图片时受到的来自外部环境和成像设备等一定程度上噪音的影响，以获取高质量清晰的图像样本；再对预处理后的图像进行特征提取，提取每块矿石图像的颜色(RGB)特征和纹理特征中的灰度共生矩阵(GLCM)特征，并且将两种特征进行组合，形成一个新的特征集；采用基于邻域粒化及不一致度量的快速属性约简算法对于组合后的特征进行约简，去除冗余的特征，保留对识别最重要的特征，形成新的特征集；最后将关键特征集与对应的类型标签作为分类器的训练样本，通过训练得到分类模型。整个方法描述了使用基于机器视觉的矿石识别方法，通过提取不同特征后进行组合，再使用特征选择算法对特征进行选择得到关键特征集，实现了矿石识别过程中的特征数的简化，提高分类器的识别率，算法简单，识别精度高。
附图说明
图1为本发明的流程图。
图2为本发明的特征选择流程图。
图3为本发明的属性约简与分类精度结果图。
具体实施方式
下面结合附图对本发明作进一步的说明。
如图1所示，一种基于机器视觉的矿石识别方法，包括以下步骤：
步骤一：根据在矿石加工厂现场中得到的矿石，采集矿石图像并创建数据集，并对数据集中的图像先进行分割再标记。具体分为两类，合格矿石图像标注为1，不合格矿石图像标注为0。
步骤二：对数据集中的图像进行预处理。
图像在获取过程中，通常会受到外部环境和成像设备的影响，一定程度上带来了噪声干扰，使得图像的质量下降。在对矿石图像进行特征提取前有必要进行图像的预处理，以此来获得高质量清晰的图像，根据矿石开采和成像的实际场景带来的噪声，可以采用中值滤波进行降噪，在过滤噪声的同时还可以保证图像边缘信息不丢失。
步骤三：提取每块矿石图像的颜色特征与纹理特征，将两种特征进行组合。包括以下步骤：
Ⅰ)提取样本集中每块矿石图像的颜色(RGB)特征，得到每块矿石图像的R、G、B三分量的均值，作为一个3维特征集；
Ⅱ)提取每块矿石图像的对比度、相关性、能量和逆差矩4种纹理特征，通过计算在0°、45°、90°、135°四个方向上对应的值，得到一个16维特征集；
式中，f1、f2、f3、f4分别为对比度、相关性、能量和逆差矩；L为灰度级数；i，j分别表示一幅图像中不同位置的灰度值；d为距离，通常取值为1；θ为方向，取值为0°、45°、90°、135°；u1为变量i的均值，u2为变量j的均值，σ1为变量i的方差，σ2为变量j的方差，
Ⅲ)将提取的颜色特征与灰度共生矩阵特征进行组合，组成一个具有19维的特征集。
步骤四：如图2所示，对于步骤三所得到的特征，使用基于邻域粒化及不一致度量的快速属性约简算法进行特征约简，如表1所示，通过约简算法后得到了能识别矿石图像的关键特征集，进一步降低特征维数，并保留原有的大量特征信息。
表1关键特征集及分类精度结果
步骤五：对于步骤四得到的关键特征集，将其与对应的类型标签一起作为分类器中的训练样本。图3为属性约简与分类精度结果图，其中在邻域半径σ＝0.28，约简属性数目为11时，达到最高分类精度90.78％。表2为邻域半径对约简特征数的影响、分类精度随约简后特征数变化的部分数据。
表2部分特征集数目与分类精度结果对比
- 上一篇：一种医用注射器针头装配设备
- 下一篇：图像处理方法、装置、计算机设备和可读存储介质