投审稿入口

基于YOLO-SNBP的密集小目标废钢料型精准识别

Accurate identification of dense small target scrap steel material type based on YOLO-SNBP

  • 摘要: 废钢作为一种可循环利用的绿色资源,是钢铁工业重要低碳属性原料,对促进钢铁行业可持续发展发挥着关键作用。废钢的品级关系到生产成本及质量,其精准分类和评级对废钢回收至关重要。现有废钢分类的研究普遍存在小目标废钢检测效果不佳及背景干扰等问题,影响废钢分类的精准度。为此,提出一种基于YOLOv5改进的密集小目标废钢识别算法YOLO-SNBP。首先,通过增加小目标检测层以提升模型在验收场景下密集小目标废钢的识别效果;其次,引入BiFormer注意力机制来加强模型在复杂背景中对小目标特征的提取能力;最后,采用Soft-NMS(Soft Non-Maximum Suppression)算法替代传统NMS(Non-Maximum Suppression)以减少因重叠导致废钢漏检问题。在自建废钢数据集上对YOLO-SNBP模型进行训练和验证,并不同检测算法进行对比分析。试验结果表明,相比基础模型,YOLO-SNBP模型P(Precision)、R(Recall)和PmA(mean Average Precision)值分别提高2.8%、7.2%和7.4%,与前人算法相比PmA值分别提升29.0%、18.2%和18.8%,在废钢料型精准识别中展现出显著优势,为验收时的高效评级提供了有效支撑。

     

    Abstract: As a recyclable green resource, scrap steel is an important alternative raw material for iron ore and plays a key role in promoting the sustainable development of the steel industry. The grade of scrap is related to the production cost and quality, and its accurate classification and rating are very important for scrap recycling. The existing research on scrap classification generally has problems such as poor detection effect of small target scrap and background interference, which affect the accuracy of scrap classification. Therefore, an improved scrap recognition algorithm YOLO-SNBP for dense small targets based on YOLOv5 is proposed. Firstly, a small target detection layer is added to improve the recognition effect of the model on dense small target scrap in the acceptance scene. Secondly, the BiFormer attention mechanism is introduced to enhance the model's ability to extract small target features in complex backgrounds. Finally, the Soft-NMS algorithm is used to replace the traditional NMS to reduce the problem of missed detection of scrap due to overlap. The YOLO-SNBP model is trained and verified on the self-built scrap data set, and different detection algorithms are compared and analyzed. The experimental results show that compared with the basic model, the P(Precision), R(Recall) and PmA (mean Average Precision) values of the YOLO-SNBP model are increased by 2.8%, 7.2% and 7.4%, respectively. Compared with the previous algorithms, the PmA values are increased by 29.0%, 18.2% and 18.8%, respectively. It shows significant advantages in the accurate identification of scrap steel, and provides effective support for the efficient rating during acceptance.

     

/

返回文章
返回