» Articles » PMID: 37688077

EFC-YOLO: An Efficient Surface-Defect-Detection Algorithm for Steel Strips

Overview
Journal Sensors (Basel)
Publisher MDPI
Specialty Biotechnology
Date 2023 Sep 9
PMID 37688077
Authors
Affiliations
Soon will be listed here.
Abstract

The pursuit of higher recognition accuracy and speed with smaller model sizes has been a major research topic in the detection of surface defects in steel. In this paper, we propose an improved high-speed and high-precision Efficient Fusion Coordination network (EFC-YOLO) without increasing the model's size. Since modifications to enhance feature extraction in shallow networks tend to affect the speed of model inference, in order to simultaneously ensure the accuracy and speed of detection, we add the improved Fusion-Faster module to the backbone network of YOLOv7. Partial Convolution (PConv) serves as the basic operator of the module, which strengthens the feature-extraction ability of shallow networks while maintaining speed. Additionally, we incorporate the Shortcut Coordinate Attention (SCA) mechanism to better capture the location information dependency, considering both lightweight design and accuracy. The de-weighted Bi-directional Feature Pyramid Network (BiFPN) structure used in the neck part of the network improves the original Path Aggregation Network (PANet)-like structure by adding step branches and reducing computations, achieving better feature fusion. In the experiments conducted on the NEU-DET dataset, the final model achieved an 85.9% mAP and decreased the GFLOPs by 60%, effectively balancing the model's size with the accuracy and speed of detection.

Citing Articles

Lightweight strip steel defect detection algorithm based on improved YOLOv7.

Lu J, Yu M, Liu J Sci Rep. 2024; 14(1):13267.

PMID: 38858448 PMC: 11164880. DOI: 10.1038/s41598-024-64080-x.

References
1.
Ren S, He K, Girshick R, Sun J . Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks. IEEE Trans Pattern Anal Mach Intell. 2016; 39(6):1137-1149. DOI: 10.1109/TPAMI.2016.2577031. View

2.
Tulbure A, Tulbure A, Dulf E . A review on modern defect detection models using DCNNs - Deep convolutional neural networks. J Adv Res. 2022; 35:33-48. PMC: 8721352. DOI: 10.1016/j.jare.2021.03.015. View

3.
Guo Z, Wang C, Yang G, Huang Z, Li G . MSFT-YOLO: Improved YOLOv5 Based on Transformer for Detecting Defects of Steel Surface. Sensors (Basel). 2022; 22(9). PMC: 9100565. DOI: 10.3390/s22093467. View

4.
Hussain M, Al-Aqrabi H, Munawar M, Hill R, Alsboui T . Domain Feature Mapping with YOLOv7 for Automated Edge-Based Pallet Racking Inspections. Sensors (Basel). 2022; 22(18). PMC: 9501564. DOI: 10.3390/s22186927. View