Metalurgija, Vol. 63 No. 3-4, 2024.
Izvorni znanstveni članak
Surface defect detection of steel based on improved YOLOv7 model
W. Z. Teng
; School of Computer Science and Software Engineering, University of Science and Technology Liaoning, China
Y. J. Zhang
; School of Computer Science and Software Engineering, University of Science and Technology Liaoning, China
*
H. G. Zhang
; School of Computer Science and Software Engineering, University of Science and Technology Liaoning, China
D. X. Gao
; School of Computer Science and Software Engineering, University of Science and Technology Liaoning, China
* Autor za dopisivanje.
Sažetak
In response to the inevitable surface defects in the manufacturing process of hot-rolled steel, this paper proposes an improved steel surface defect detection model based on YOLOv7. In the Extended Efficient Large Aggregation Network (E-ELAN), the model replaces conventional convolution with Omni-Dimensional Dynamic Convolution (ODConv) to enhance the network’s sensitivity to feature extraction using a combination of various attention mechanisms. Additionally, the detection head in the head section is replaced with an Efficient Decoupled Detection Head, enhancing the model’s capability to classify and locate small defects. The proposed model is tested on the public dataset NEU-DET, achieving a high mAP of 76,5 %. This effectively enhances the model’s ability to detect surface defects in steel while maintaining a fast detection speed.
Ključne riječi
hot-rolled; steel; surface defect; YOLOv7 model; efficient decoupled detection head
Hrčak ID:
315685
URI
Datum izdavanja:
1.7.2024.
Posjeta: 214 *