Steel surface defect detection algorithm based on MCB-FAH-YOLOv8
To address the problems of misdetection,omission,and low detection accuracy in existing deep learning-based algorithms for detecting defects on steel surfaces,a YOLOv8 steel surface defect detection algorithm was proposed based on a modified CBAM(MCB)and replaceable four-head ASFF prediction head(FAH),abbreviated as MCB-FAH-YOLOv8.By integrating the modified convolutional attention mechanism module(CBAM),the algorithm could achieve better determination of the densely populated targets.By changing the FPN structure to BiFPN,it could extract context information more efficiently.It also incorporated adaptive feature fusion(ASFF)for the automatic identification of the most suitable fusion features.The algorithm also boosted its precision by replacing the SPPF module with the SimCSPSPPF module.Meanwhile,for tiny object detection,a four-head ASFF prediction head was proposed,designed to be replaceable based on the dataset characteristics.The experimental results demonstrated that the MCB-FAH-YOLOv8 algorithm could achieve a detection accuracy(mAP)of 88.8％on the VOC2007 dataset and 81.8％on the NEU-DET steel defect detection dataset,outperforming the benchmark model by 5.1％and 3.4％,respectively.This new algorithm achieved a higher detection accuracy with less loss of detection speed,thus ensuring a good balance between accuracy and speed.
NETL
NSTL
万方数据
