投审稿入口

加热炉出钢过程钢坯定位的NDS-yolov8目标检测算法

NDS-yolov8 object detection algorithm for billet positioning during transfer from the heating furnace

  • 摘要: 在热轧过程中,钢坯位置定位至关重要,可以实现自动化的钢坯位置检测,从而提高生产效率和质量控制。但实际生产过程中,由于钢坯生产环境恶劣,造成钢板定位检测困难。针对上述问题,提出一种稳定的Yolov8钢坯检测算法:提出混合注意力机制网络,解决了金字塔网络中浅层特征丢失的问题,增加网络学习局部特征的能力,轻量化模型,提升检测精度。实施方法为:首先在网络中引入注意力模块,提高对图像细节特征信息的保留能力,提升目标的整体检测精度;其次,设计NPANet特征融合结构,加强对图像的多尺度特征融合能力;然后,改进卷积模块,轻量化网络模型;最后,改进损失函数,提高了算法的回归性能,降低稳定框生成误差。实验结果表明:改进的NDS-yolov8模型相比于初始网络结构,其权重文件大小由6.2 MB变为4.6 MB、浮点运算性能由8.1 GFLOPS变为6.4 GFLOPS、不同IoU值时的平均精度(PmA@0.5∶0.95)提升0.5%。与实际场景中的真实值相比,NDS-yolov8网络模型在精度上相较于原始Yolov8网络模型误差显著降低,能够更准确地估计钢坯的实时位置,从而有效提升钢坯检测与定位的性能。

     

    Abstract: During the hot rolling process, the precise positioning of steel billets is of paramount importance, as it enables automated billet inspection, which in turn enhances both production efficiency and quality control. However, the actual production process is arduous due to the harsh conditions of the billet production environment, which presents challenges for plate inspection. To address the problems mentioned above, a robust Yolov8 algorithm for billet detection is proposed. A hybrid attention mechanism network is proposed to solve the problem of shallow feature loss in pyramid networks, enhancing the model's ability to learn local features and improving detection accuracy while maintaining model lightweightness. The implementation method involves the introduction of attention modules into the network with the objective of enhancing the preservation of detailed image feature information, thereby improving the overall detection accuracy of the targets. Subsequently, the NPANet feature fusion structure is designed to enhance the network's ability to fuse multi-scale features of images. This is achieved by refining the convolution modules to make the network model lighter. Finally, the loss function is improved to enhance the algorithm's regression performance and reduce the error in stable box generation. The experimental results demonstrate that the improved NDS-yolov8 model, in comparison to the initial network structure, exhibits a reduction in the weight file size from 6.2 MB to 4.6 MB, a decrease infloating-point performance from 8.1 GFLOPS to 6.4 GFLOPS, and an increase in the mean average precision (PmA@0.5:0.95by 0.5% at different IoU values. Compared to actual values in real-world scenarios, the NDS-yolov8 network model demonstrates significantly reduced error margins relative to the original Yolov8 network model. It achieves more accurate estimation of the billet's real-time position, thereby effectively enhancing the performance of billet detection and localization.

     

/

返回文章
返回