基于EAIDK的智能煤矸分拣系统设计

王冠军,苏婷婷,刘文博,钱智平,李佳泽

(中国矿业大学 计算机科学与技术学院, 江苏 徐州 221116)

摘要现有基于图像识别的煤矸石分拣方法实时性较差且整体分拣准确率不高,而基于密度的分拣方法适用于井下初选,成本较高。针对上述问题,设计实现了一种基于EAIDK的智能煤矸分拣系统。采用嵌入式人工智能开发平台EAIDK构建矸石识别和分拣控制硬件平台,在嵌入式深度学习框架Tengine下利用深度学习算法搭建卷积神经网络,建立端到端可训练图像检测模型,并利用智能摄像机获取的图像数据训练模型;通过手眼标定获得摄像机坐标系与机械臂坐标系之间的关系,控制机械臂进行矸石追踪和分拣。实验结果表明,该系统矸石识别准确率稳定保持在95%以上,机械臂跟踪时间小于30 ms,执行误差为1 mm左右,可以满足煤矸分拣工艺要求。

关键词选煤; 自动煤矸分拣; 矸石识别; 卷积神经网络; 机械臂控制; EAIDK平台; 深度学习

0 引言

传统的煤矸石人工分选方法是指原煤被运出井口后,经带式输送机运到人工拣矸楼,由工人把矸石与杂物拣出来,再输送到选煤车间。对煤矸石自动分拣方法的研究主要集中在煤矸石图像识别技术上,例如小波分析与DSP[1-2]、粗糙集理论[3]、纹理分析[4-5]、核典型相关分析[6]、灰度自适应边缘检测[7]等。以上方法均可实现简单的矸石检测,存在的主要不足是实时性较差且整体分拣准确率不高。利用密度进行分选[8-9]的方法适用于井下初选,且成本较高。针对上述问题,本文设计了一种基于EAIDK的智能煤矸分拣系统。

1 系统总体设计

1.1 系统结构

基于EAIDK的智能煤矸分拣系统由EAIDK-610平台、视觉系统、机械臂等组成,如图1所示。EAIDK-610是一种面向边缘计算的嵌入式人工智能开发平台,硬件方面,使用高性能ARM SoC搭载OPEN AI LAB嵌入式AI开发平台AID,具备语音、视频等数据采集能力,有适用于多场景的运动控制接口;软件方面,支持视觉处理与分析、语音识别、语义分析、SLAM(即时定位与地图构建)等应用的基础平台和主流开源算法。视觉系统由PLC、智能摄像机、触摸屏等构成,实现图像采集与处理功能。机械臂安装于输送带一侧轨道上,可以水平运动。

图1 基于EAIDK的智能煤矸分拣系统结构

Fig.1 Structure of intelligent coal and gangue sorting system based on EAIDK

1.2 系统工作流程

原煤经过带式输送机输送到原煤仓时,通过带式输送机上方的摄像机获得当前原煤图像,识别煤和矸石,并测量矸石位置和朝向;通过手眼标定获得摄像机坐标系与机械臂坐标系之间的关系;机械臂以合适的姿态运动到矸石位置,抓取矸石并移动到矸石桶位置后放下矸石;最后机械臂归位,进入下一轮煤矸分拣。系统工作流程如图2所示。

图2 基于EAIDK的智能煤矸分拣系统工作流程

Fig.2 Working flow of intelligent coal and gangue sorting system based on EAIDK

2 基于EAIDK的矸石识别

在处理的图像中,煤块是黑色的,而矸石基本上是灰色的,煤块本身灰度比较低,只有当煤块反光时,反光部分出现亮点,煤块的这些部分灰度级才会较高,而矸石一般灰度级都比较高。以灰度分布的均值和方差作为特征值来区分煤块和矸石,根据模式识别的原理,将煤和矸石的标准特征值作为分拣标准。在识别时,将实际煤和矸石的灰度分布与之比较,理论上就可以将煤和矸石区分开来。

2.1 图像预处理

使用TEngine深度学习框架所支持的基于卷积神经网络的VGG16深度学习模型搭建检测模型,VGG16模型如图3所示。VGG16总共有16层,含13个卷积层和3个全连接层。图像经过64个卷积核的2次卷积,128个卷积核2次卷积,256个卷积核3次卷积,2次512个卷积核3次卷积后,分别进行一次池化处理,最后进行3次全连接。将矸石分拣图像输入VGG16模型,在VGG16模型中将图片的每个像素值减去在训练集上计算得到的平均RGB值,并将智能处理单元EAIDK-610接收到的图像信息调整为符合神经网络输入数据要求的格式。

2.2 图像识别标记

(1) 特征提取。将预处理后的图像输入特征提取网络得到特征图。VGG16模型经过fine-tune处理后作为特征提取网络,即移除VGG16模型中conv5最后一层的后续所有层,并使用剩余的权重作为模型下一次训练的起点。这种特征提取方式相比投影方法更为科学合理,且可通过改变卷积核大小、池化和最后输出的特征向量的大小控制整体模型的拟合能力,相比其他特征提取方法更为灵活。

图3 VGG16模型

Fig.3 VGG16 model

(2) 目标检测。对特征图进行滑动扫描,生成一组预定义的锚点框。用回归模型和分类模型对预定义的锚点框进行坐标预测和矸石概率预测,判定目标是否为矸石并定位锚点框顶点坐标。最后,将回归和分类预测的输出进行非极大值抑制筛选,得到最终矸石筛选结果。在conv5中加入3个新的卷积层搭建目标检测网络:第1部分是一个常规卷积层,用于对前一层提供的数据进行编码;第2部分卷积层用于回归和预测;第3部分负责预测目标概率。

(3) 图像标记。根据预定义区域矸石检测结果对矸石进行标记。

通过多媒体接口将识别标记出的矸石信息输出到机器人接口进行后续拣矸操作,并在显示设备上实时显示,EAIDK-610可提供显示触摸一体屏配件。

3 矸石分拣控制

3.1 手眼标定

为获得摄像机与机械臂坐标系的关系,需要进行手眼标定。对于固定视觉手眼系统的标定,两坐标系之间的齐次坐标变换关系可记为[10-11]

(1)

式中:xc,yc,zc为目标点在摄像机坐标系中的坐标;xr,yr,zr为目标点在机械臂坐标系中的坐标;R为机械臂坐标系相对摄像机坐标系旋转3×3的正交矩阵,描述了机械臂坐标系在摄像机坐标系中的位置和姿态;Y为机械臂坐标系的原点在摄像机坐标系中的坐标。

3.2 机械臂控制

为实现煤矸石拣取,需要控制机械臂运动,使机械臂在规定时间内以合理姿态到达规定位置。在保证控制性能和精度的情况下,首先建立两自由度机械臂简化模型,通过拉格朗日方程建立动力学方程[12]

(2)

式中:q为角度向量,为角速度与角加速度;T=[T1 T2],为驱动力矩向量;为正定惯性矩阵;为科里奥力和离心力的结合矩阵;G(q)为重力势能矩阵。

根据动力学方程对机械臂进行正逆运动学解算,利用FPGA技术构建专用的机器人主-从手臂运动学硬件模型,在毫秒级的时间内实现复杂的主机械臂模型正解和从机械臂模型逆解计算,最后结合轨迹规划完成机械臂运动规划。机械臂运动过程分为3个阶段:① 趋近阶段:机械臂运动到输送带末端并打开抓手到合适大小。② 拣取阶段:抓取煤矸石并投放到矸石通道中。③ 回位阶段:机械臂回位到初始位置,等待下一次拣取。

4 实验验证

通过基于EAIDK的智能煤矸分拣系统采集100组数据,每组包含100张图像,原始图像预处理后送入VGG16模型进行训练,模型平均训练时间为10 min,利用训练完的神经网络模型进行矸石识别和分拣。模型的训练在服务器端进行,使用AI高性能计算(Alibaba Cloud HPC)平台完成。由于模型只需训练一次,所付出的时间代价是可接受的。经平台处理后的矸石识别图像如图4所示。矸石识别准确率曲线如图5所示。可以看出,矸石识别准确率稳定保持在95%以上。

图4 经平台处理后的矸石识别图像

Fig.4 Gangue recognition image after platform processing

图5 矸石识别准确率曲线

Fig.5 Curve of gangue recognition accuracy

在河南神火集团薛湖选煤厂进行了系统测试,得出视觉系统的平均工作周期为110 ms,选煤厂自动机械臂响应跟踪时间小于30 ms,执行误差为1 mm左右,可以满足分拣工艺要求。

5 结语

设计了基于EAIDK的智能煤矸分拣系统,利用EAIDK-610平台作为控制中心,通过视觉系统实现图像采集与处理,并采用机械臂实现矸石分拣。该系统矸石识别准确率高,机械臂跟踪时间短,执行误差小。该系统存在的主要问题是当原煤含矸量较高时,系统响应受限,此外,由于自然光源和灯光等干扰,增加了图像处理难度,图像检测模型的训练时间也较长,下一步工作将集中于提高系统响应速度,进一步提高矸石识别率,并完善实时追踪系统。

参考文献(References):

[1] 胡倍倍.基于DSP的煤和煤矸石图像识别研究[D].开封:河南大学,2017.

HU Beibei.Coal and coal gangue image recognition research based on DSP[D].Kaifeng: Henan University, 2017.

[2] 陈立,杜文华,曾志强,等.基于小波变换的煤矸石自动分选方法[J].工矿自动化,2018,44(12):60-64.

CHEN Li,DU Wenhua,ZENG Zhiqiang,et al.Automatic separation method of coal and gangue based on wavelet transform[J].Industry and Mine Automation,2018,44(12):60-64.

[3] 梁澈.粗糙集理论在煤矸石图像识别技术中的应用[D].西安:西安科技大学,2010.

LIANG Che.Application of rough set theory in coal gangue image recognition[D].Xi'an:Xi'an University of Science and Technology,2010.

[4] 陈玲,沈红标,李咸伟,等.改进的图像纹理检索方法在矿石识别中的应用[J].中国图象图形学报,2006,11(11):1700-1703.

CHEN Ling,SHEN Hongbiao,LI Xianwei,et al. Application of image texture retrieval in mineral discrimination[J].Journal of Image and Graphics,2006,11(11):1700-1703.

[5] 何敏,王培培,蒋慧慧.基于SVM和纹理的煤和煤矸石自动识别[J].计算机工程与设计,2012,33(3):1117-1121.

HE Min,WANG Peipei,JIANG Huihui.Recognition of coal and stone based on SVM and texture[J].Computer Engineering and Design,2012,33(3):1117-1121.

[6] 翟永前,王浩,赵力.一种基于核典型相关分析的煤炭矸石鉴别方法[J].电子器件,2013,36(1):58-60.

ZHAI Yongqian, WANG Hao, ZHAO Li. Waste rock discrimination based on kernel canonical correlation analysis[J].Chinese Journal of Electron Devices,2013,36(1):58-60.

[7] 余乐,郑力新,杜永兆,等.采用部分灰度压缩扩阶共生矩阵的煤和煤矸石图像识别[J].华侨大学学报(自然科学版),2018,39(6):906-912.

YU Le,ZHENG Lixin,DU Yongzhao,et al.Image recognition method of coal and coal gangue based on partial grayscale compression extended coexistence matrix[J].Journal of Huaqiao University(Natural Science), 2018,39(6):906-912.

[8] 张宁波,鲁岩,刘长友,等.综放开采煤矸自动识别基础研究[J].采矿与安全工程学报,2014,31(4):532-536.

ZHANG Ningbo,LU Yan,LIU Changyou,et al.Basic study on automatic detection of coal and gangue in the fully mechanized top coal caving mining[J].Journal of Mining & Safety Engineering,2014,31(4):532-536.

[9] 程学珍,王伟,卫阿盈.煤与矸石在线分选系统设计[J].工矿自动化,2013,39(8):95-98.

CHENG Xuezhen,WANG Wei, WEI Aying.Design of online separation system for coal and gangue[J].Industry and Mine Automation,2013,39(8):95-98.

[10] 苏剑波.ADRC理论和技术在机器人无标定视觉伺服中的应用和发展[J].控制与决策,2015,30(1):1-8.

SU Jianbo.Robotic uncalibrated visual serving based on ADRC[J].Control and Decision,2015,30(1):1-8.

[11] 洪磊,嵇保健,王玉国,等.改进的单特征点机器人手眼自标定方法[J].北京邮电大学学报,2016,39(1):96-101.

HONG Lei,JI Baojian,WANG Yuguo,et al.Improved self-calibration method to hand-eye relation of robot based on a single feature point[J].Journal of Beijing University of Posts and Telecommunications,2016,39(1):96-101.

[12] 赵海滨,于清文,刘冲,等.机械臂控制系统仿真实验设计[J].实验室研究与探索,2018,37(11):100-104.

ZHAO Haibin,YU Qingwen,LIU Chong,et al.Simulation experiment design of manipulator control system[J]. Research and Exploration in Laboratory,2018,37(11):100-104.

Design of intelligent coal and gangue sorting system based on EAIDK

WANG Guanjun,SU Tingting,LIU Wenbo,QIAN Zhiping,LI Jiaze

(School of Computer Science and Technology, China University of Mining and Technology, Xuzhou 221116, China)

AbstractExisting coal and gangue sorting method based on image identification has poor real-time performance and low sorting accuracy, the density-based sorting method is suitable for underground preparation but has high cost. In view of above problems, an intelligent coal and gangue sorting system based on EAIDK was designed. Embedded artificial intelligence development platform EAIDK is used to build hardware platform for gangue recognition and sorting control, deep learning algorithm is used to build a convolutional neural network under embedded deep learning framework Tengine, and end-to-end trainable image detection model is established and trained by image data obtained by smart cameras.Relationship between the camera coordinate system and the robot arm coordinate system is obtained through hand-eye calibration, and the gangue is tracked and sorted by robot arm. The experimental results show that the system's gangue recognition accuracy remains stable above 95%, the tracking time of robot arm is less than 30 ms, and the execution error is about 1 mm, which can meet the requirements of coal gangue sorting process.

Key words:coal preparation; automatic coal and gangue sorting; gangue recognition; convolutional neural network; robotic arm control; EAIDK platform; deep learning

中图分类号:TD94

文献标志码:A

文章编号1671-251X(2020)01-0105-04

DOI:10.13272/j.issn.1671-251x.2019050019

收稿日期:2019-05-09;修回日期:2019-12-29;责任编辑:胡娴。

基金项目:国家自然科学基金项目(61772530,61402483,51104157);江苏省自然科学基金面上项目(BK20171192);教育部产学合作协同育人项目(ARM/NXP-2017;MICRODUINO 2018);江苏省大学生创新训练计划项目。

作者简介:王冠军(1981-),男,山东临沂人,讲师,博士,主要研究方向为EDA/机器人技术,E-mail:zywgj@cumt.edu.cn。

引用格式:王冠军,苏婷婷,刘文博,等.基于EAIDK的智能煤矸分拣系统设计[J].工矿自动化,2020,46(1):105-108.

WANG Guanjun,SU Tingting,LIU Wenbo,et al.Design of intelligent coal and gangue sorting system based on EAIDK[J].Industry and Mine Automation,2020,46(1):105-108.