FDTransUnet: An aluminum surface defect segmentation model based on feature differentiation

Mingzhu Tang and Wencheng Wang

PLOS ONE, 2025, vol. 20, issue 3, 1-16

Abstract: Aiming at the current problems in the field of industrial defect segmentation, such as difficulty of obtaining a large number of defect samples, low recognition accuracy and lack of segmentation accuracy, a surface defect segmentation model for aluminum based on feature differentiation is proposed: FDTransUnet. First, the limited defective samples are effectively expanded by the feature differentiation data augmentation strategy to alleviate the overfitting problem caused by the insufficient sample. Second, the Transformer architecture is added by improving the U-net network, and the improved network combines the global self-attention mechanism of the Transformer and the hierarchical structure of the U-net, which can effectively extract the local and global information in the defect sample. Finally, a composite loss function is constructed to address the problem of unbalanced foreground and background sizes of defective samples and to improve segmentation accuracy. The experimental results show that FDTransUnet achieves 94.5% MPA and 89.7% Dice coefficient on the aluminum surface defect dataset. In the final generalization experiment, FDTransUnet is validated with other mainstream segmentation models on the steel surface defect dataset, and the experiment proves that the segmentation model has good generalization performance and robustness, and can be applied to different scenarios of industrial inspection.

Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0320060 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 20060&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0320060

DOI: 10.1371/journal.pone.0320060

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().