Aluminum Surface Defect Detection Method Based on Lightweight YOLOv7-tiny

Ling-fei XI , ·Yaermaimaiti YILIHAMU

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (27) : 11786 -11794.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (27) : 11786-11794 DOI: 10.12404/j.issn.1671-1815.2400559
research-article
Aluminum Surface Defect Detection Method Based on Lightweight YOLOv7-tiny
Author information +
History +
PDF

Abstract

In order to improve the efficiency of small target detection of aluminum surface defects, an aluminum surface defect detection algorithm based on improved YOLOv7-tiny was proposed. Firstly, the SimAM (similarity-based attention module) attention mechanism was added to the network to make the network pay more attention to the useful information and suppress the interference of invalid samples in the image. Secondly, the sppcspc structure in the backbone network was improved to Ghostsppcspc, which reduced the parameter redundancy during model training, and GSconv was used in the detection layer instead of the ordinary convolution, which lightened the network and strengthened the feature fusion to improve the network detection accuracy, and finally, the NWD(normalized wasserstein distance) was introduced to replace the original CIOU(complete intersection over union) loss function, which improves the network's accuracy in detecting small targets. The improved algorithm was applied to the Tianchi aluminium dataset for validation, and the experimental results show that the model was able to identify different kinds of defects on the surface of aluminium profiles effectively, with a 10.1% increase in mAP, a 6.4% decrease in the number of parameters, and a 12.2% decrease in the amount of computation compared with the original YOLOv7-tiny algorithm. The proposed method achieves a lightweight network model and can meet the current requirements of defect detection in the production site of aluminium profile factories.

Keywords

defect detection / YOLOv7-tiny / attention mechanism / NWD / GSconv / Ghostsppcspc

Cite this article

Download citation ▾
Ling-fei XI, ·Yaermaimaiti YILIHAMU. Aluminum Surface Defect Detection Method Based on Lightweight YOLOv7-tiny. Science Technology and Engineering, 2024, 24(27): 11786-11794 DOI:10.12404/j.issn.1671-1815.2400559

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (6653KB)

19

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉