摘要:目的 针对现有钢材表面缺陷检测算法检测精度不足和模型复杂度高的问题,提出一种基于 YOLOv8 的改进 钢材表面缺陷检测算法,命名为 YOLOv8-RDP。 方法 首先,引入 RepNCSPELAN4 模块替换 YOLOv8n 模型中的 C2f 模块,通过并行处理不同尺度的特征,并在最终的卷积层中融合这些特征来优化模型的特征提取和融合能力;其 次,在骨干网络部位集成 DA(Deformable Attention)注意力机制,通过自适应调整卷积核采样点,增强模型对不同形 状和大小物体的特征捕捉能力,从而提高对关键信息的捕捉效率;最后,为减少模型所需的计算资源,结合 PConv (Partial Convolution)模块改进基线模型检测头,利用特征图中的冗余性,根据数据是否缺失动态调整卷积核的作 用区域,以减少计算量。 结果 在 NEU-DET 数据集上的实验结果表明,YOLOv8-RDP 的 mAP 达到了 78. 8%,较基 线模型提升了 1. 8%;参数量减少至 1. 87 M,GFLOPs 降至 3. 5 G,分别比基线模型降低了 37. 9%和 57. 0%。 改进后 的模型在保持高精度的同时,大幅度减少了计算资源的需求。 结论 YOLOv8-RDP 算法在钢材表面缺陷检测中表 现出较高的检测精度和较低的模型复杂度,对钢材表面缺陷检测具有一定的应用价值。