College of Information Engineering, Dalian Ocean University, Dalian, 116021, China.
Sci Rep. 2023 Jul 8;13(1):11077. doi: 10.1038/s41598-023-38085-x.
Deep learning is currently being used to automate surface defect detection in aluminum. The common target detection models based on neural networks often have a large number of parameters and a slow detection speed, which is not conducive to real-time detection. Therefore, this paper proposes a lightweight aluminum surface defect detection model, M2-BL-YOLOv4, based on the YOLOv4 algorithm. First, in the YOLOv4 model, the complex CSPDarkNet53 backbone network was modified into an inverted residual structure, which greatly reduced the number of parameters in the model and increased the detection speed. Second, a new feature fusion network, BiFPN-Lite, is designed to improve the fusion ability of the network and further improve its detection accuracy. The final results show that the mean average precision of the improved lightweight YOLOv4 algorithm in the aluminum surface defect test set reaches 93.5%, the number of model parameters is reduced to 60% of the original, and the number of frames per second (FPS) detected is 52.99, which increases the detection speed by 30%. The efficient detection of aluminum surface defects is realized.
深度学习目前正被用于实现铝表面缺陷的自动化检测。常见的基于神经网络的目标检测模型通常具有大量的参数和较慢的检测速度,不利于实时检测。因此,本文提出了一种基于 YOLOv4 算法的轻量化铝表面缺陷检测模型 M2-BL-YOLOv4。首先,在 YOLOv4 模型中,将复杂的 CSPDarkNet53 骨干网络修改为倒置残差结构,大大减少了模型的参数量,并提高了检测速度。其次,设计了一种新的特征融合网络 BiFPN-Lite,以提高网络的融合能力,进一步提高其检测精度。最终的实验结果表明,改进后的轻量化 YOLOv4 算法在铝表面缺陷测试集上的平均准确率达到 93.5%,模型参数数量减少到原始模型的 60%,检测帧率(FPS)达到 52.99,检测速度提高了 30%。实现了对铝表面缺陷的高效检测。
Front Plant Sci. 2022-11-11
Entropy (Basel). 2022-11-2
Sensors (Basel). 2023-1-21
Entropy (Basel). 2021-10-30
Sensors (Basel). 2022-11-4
Sensors (Basel). 2022-10-13
Front Plant Sci. 2022-3-11
Materials (Basel). 2019-5-23