Surface defect detection of steel based on improved YOLOv7 model
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-Di...
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Croatian Metallurgical Society
2024-01-01
|
| Series: | Metalurgija |
| Subjects: | |
| Online Access: | https://hrcak.srce.hr/file/456149 |
| Summary: | 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. |
|---|---|
| ISSN: | 0543-5846 1334-2576 |