特征压缩激活作用下U-Net网络的煤尘颗粒特征提取

Feature extraction of coal dust particles based on U-Net combined with squeeze and excitation module

  • 摘要: 为研究煤尘颗粒的图像特征内在机理,以选煤厂为项目背景区,包含6个工作面1 500个煤尘图像信息为依托,搭建特征压缩激活U-Net卷积神经网络并对煤尘颗粒图像进行语义分割。首先通过现场采样获取煤尘颗粒样本,建立图像数据集并输入到分割网络;其次通过网络左半部模型完成煤尘特征的批量归一化和压缩激活,获得输出特征传递到网络右半部模型进行上采样以恢复煤尘特征图像尺寸,完成煤尘颗粒信息的二分类;最后运用二值交叉熵及合页协同损失函数训练网络并缓解分割任务中的颗粒特征类别不平衡问题。通过搭建模型进行仿真试验:① 通过试验仿真二值交叉熵及合页协同损失函数对网络分割性能的影响;② 通过试验评估所提出的优化网络与常规颗粒图像提取算法FCN,SegNet,DeepLab,SENet,U-Net网络的分割性能,并验证所提出的优化网络的特征信息提取能力;③ 在试验①,② 结果基础上,采用八邻域特征算法实现煤尘颗粒的定位标注,提取颗粒占比特性参数。研究结果表明,二值交叉熵及合页协同损失函数可获得较优分割效果;所提出的Unet-SE改进网络模型对不同形状及粒径的颗粒类别具有较优的学习效果,其学习性能指标:准确率、召回率以及F1分数分别提高到0.873 2,0.843 4和0.858 0。与其他识别算法相比,改进算法可有效增强煤尘特征的学习能力,大幅缩短训练时间,并精确获取颗粒特征更多细节信息。

     

    Abstract: To study the inherent mechanism of image characteristics of coal dust particles,a convolutional neural network U-Net combined with squeeze and excitation is proposed for carrying out semantic segmen tation of particle images.The sample data information includes 1 500 coal dust images in six working faces captured from coal preparation plant.Firstly,the image data sets are built with the dust particles by spot sampling and as inputs transmitted to the segmentation network.Secondly,the left-half network is achieved with batch normalization,squeezing and exciting the coal dust features.The obtained features are transferred to the right-half network to up-sample for recovering the image size of particle feature,thus the binary classifi cation of particles is completed.Finally,the binary cross entropy based hinge collaborative loss function is applied to train the network,which can alleviate classification imbalance in segmentation task.The simulation experiments are carried out in three steps as below:① simulating through experiments to evaluate the performance of binary cross entropy based hinge collaborative loss function on network segmentation;② the performance comparison of the proposed network with other conventional image extraction approaches,such as FCN,SegNet,DeepLab,SENet,U-Net network,are evaluated and the feature information extraction ability of Unet SE network is verified;and ③ the eight neighbor algorithm is proposed to locate and label the particles and extract the particle proportion characteristic parameters.The research results show that the binary cross entropy based hinge collaborative loss function can get better segmentation effect,and the improved Unet SE network has a better learning effect on the particle classification with different shapes and sizes.The performance indexes of accuracy,recall rate and F1 scores increase to 0.873 2,0.843 4 and 0.858 0 respectively.It can achieve more accurate extraction,faster training time,and more detailed particles feature acquisition.

     

/

返回文章
返回
Baidu
map