New website getting online, testing
    • 摘要: 为增强钢材表面检测中对小目标缺陷的检测能力,提出一种改进的YOLOv8-SOE模型。该模型通过设计FSCConv模块来处理P2层特征,通过压缩P2层特征并将其与P3层特征深度融合,有效增强模型对小目标特征的敏感性,同时避免引入额外检测层带来的计算负担。为进一步优化多尺度特征融合能力,采用CSP-OK (cross stage partial omni-kernel)模块优化多尺度特征融合,提高不同尺度特征的整合效率。通过引入SIoU损失函数优化边界框回归,进一步提升定位精度。实验结果表明,YOLOv8-SOE模型在NEU-DET数据集上的mAP达80.7%,较基准模型提升5.4%,且在VOC2012数据集上具有较强的泛化能力。该模型在提升小目标检测精度的同时,保持较高的计算效率,展现出良好的应用潜力。

       

      Abstract: In order to improve the detection capability of small target defects in steel surface inspection, an improved YOLOv8-SOE model is proposed. The model processes the P2 layer features by designing the FSCConv module. By compressing the P2 layer features and deeply fusing them with the P3 layer features, the model's sensitivity to small target features is effectively enhanced, while avoiding the computational burden caused by the introduction of additional detection layers. In order to further optimize the multi-scale feature fusion capability, cross stage partial omni-kernel (CSP-OK) module is used to optimize the multi-scale feature fusion, which improves the integration efficiency of features of different scales. The SIoU loss function is introduced to optimize the bounding box regression, which further improves the positioning accuracy. Experimental results show that the mAP of the YOLOv8-SOE model on the NEU-DET dataset achieves 80.7%, which is 5.4% higher than the baseline model, and has good generalization ability on the VOC2012 dataset. While improving the accuracy of small target detection, the model maintains a high computational efficiency and has good application prospects.