基于YOLOv8的轻量化煤屑颗粒群实例分割方法

Lightweight coal particle group instance segmentation method based on YOLOv8

  • 摘要: 在矿井瓦斯防治的基础研究领域,大量颗粒煤被广泛用于各类实验。由于实验煤样的粒径分布对实验结果有显著影响,因此准确快速地测定煤屑颗粒群的粒径分布具有重要的研究意义。精确的颗粒实例分割是图像法检测颗粒粒径分布的关键前提。煤屑颗粒具有体积微小、形状各异、数量众多以及颜色单一等特点,使其在图像中表现为特征微小、边缘不规则以及颗粒之间存在黏连的特征。这些因素使得深度模型难以准确捕捉和描述颗粒的边缘形状,而且对于密集小目标的分割,需要更高的计算复杂度,但为保持算法的工程实用性,还必须考虑网络的轻量性。以上因素均为分割模型的设计增加了难度。另外,深度学习是由数据驱动的,数据集的缺乏也是限制深度学习算法使用的一大挑战。为了应对以上挑战,提出了基于YOLOv8的轻量化煤屑颗粒群实例分割方法。首先,研发了基于柔性振动盘分离颗粒的图像采集系统,用于快速批量获取煤屑颗粒图片,提出基于SAM模型预标注的高效颗粒图像标注方法,提高了数据标注效率,从而构建了一个样本多样性丰富的煤屑颗粒群数据集,解决了煤屑颗粒群数据集缺乏的问题。其次,基于YOLOv8n-seg主干网络,在Backbone部分引入跨阶段局部动态蛇形卷积增强特征提取能力,从而提升模型捕捉不规则小目标边缘信息的能力,解决了煤屑颗粒这类密集不规则小目标难以特征提取的问题。再次,在特征融合网络层中设计了基于自适应权重的高性能特征融合路径,增强了模型的表征能力,解决了煤屑颗粒边缘不规则以及颗粒之间存在黏连而难以识别边缘的问题。最后,通过裁剪负责大尺寸目标的特征提取层,降低了模型复杂度和参数量,解决了分割模型计算复杂度而应用性不足的问题。实验结果显示,相较于YOLOv8n-seg,提出的DP-YOLOv8在精度方面mAP@0.5:0.95提升了0.8%,同时参数量Params从3.264×106降至1.476×106,运算量FLOPs由12.1×109降至11.4×109。这表明基于YOLOv8的轻量化煤屑颗粒群实例分割方法在模型分割精度和轻量化之间实现了更好的权衡。

     

    Abstract: In the foundational research field of mine gas control, a large quantity of coal particles is widely employed in various experiments. Due to the significant impact of particle size distribution on experimental results, the precise and rapid determination of the particle size distribution of coal samples is of paramount research significance. Accurate particle instance segmentation is the key premise of image method to detect particle size distribution. Coal particles possess characteristics such as small volume, diverse shapes, numerous quantities, and a single color, which make them appear in images with small features, irregular edges, and adhesion between particles. These factors make it challenging for deep models to accurately capture and describe the edge shapes of particles. Moreover, the segmentation of dense small targets requires higher computational complexity, and to maintain the algorithm's engineering practicality, the lightweight of the network must be considered. All these factors contribute to the increased difficulty in the design of segmentation models. Additionally, since deep learning is driven by data, the lack of data sets is also a challenge that limits the use of deep learning algorithms. To address the challenges mentioned above, a lightweight coal particle group instance segmentation method based on YOLOv8 is proposed. Firstly, a fast image acquisition system based on flexible vibrating tray to separate particles is independently developed to quickly obtain coal particle pictures in batches. Then, an efficient particle image annotation method based on the SAM model is proposed, which pre-annotates the images and enhances data annotation efficiency. Consequently, a coal particle group dataset with abundant sample diversity is constructed, thus mitigating the scarcity issue in existing datasets for coal particle group. Secondly, based on the YOLOv8n-seg Backbone network, cross-stage local dynamic snake convolution is introduced in the backbone part to enhance the feature extraction capability, thereby improving the ability of the model to capture edge information of small irregular targets, and solving the problem that it is difficult to extract features from dense irregular small targets such as coal particles. Additionally, a high-performance feature fusion path based on adaptive weights is designed in the feature fusion network layer, enhancing the model's representational power. This addresses the challenge of identifying edges caused by irregular edges of coal particles and the adhesion between particles. Finally, by cropping the feature extraction layer responsible for large-size targets, the complexity and parameter count of the model are reduced, thus addressing the issue of computational complexity of segmentation models and their limited applicability. Experimental results show that compared with YOLOv8n-seg, the proposed DP-YOLOv8 improves achieved a 0.8% improvement in mAP@0.5:0.95, while the Params decreases from 3.264×106 to 1.476×106, and the FLOPs decreases from 12.1×109 to 11.4×109. The results show that the lightweight coal particle group instance segmentation method based on YOLOv8 achieves a better trade-off between model segmentation accuracy and lightweight.

     

/

返回文章
返回