CASDD: Automatic Surface Defect Detection Using a Complementary Adversarial Network

Sukun Tian, Pan Huang, Haifeng Ma, Jilai Wang, Xiaoli Zhou, Silu Zhang, Jinhua Zhou, Renkai Huang, Yangmin Li

Research output: Journal article publicationJournal articleAcademic researchpeer-review

16 Citations (Scopus)

Abstract

Surface defect detection (SDD) plays an extremely important role in the manufacturing stage of products. However, this is a fundamental yet challenging task, mainly because the intraclass defects have large differences in shape and distribution, and low contrast between the object regions and background, and it is difficult to adapt to other materials. To address this problem, we propose a complementary adversarial network-driven SDD (CASDD) framework to automatically and accurately identify various types of texture defects. Specifically, CASDD consists of an encoding–decoding segmentation module with a specially designed loss measurement and a novel complementary discriminator mechanism. In addition, to model the defect boundaries and enhance the feature representation, the dilated convolutional (DC) layers with different rates and edge detection (ED) blocks are also incorporated into CASDD. Moreover, a complementary discrimination strategy is proposed, which employs two independent yet complementary discriminator modules to optimize the segmentation module more effectively. One discriminator identifies contextual features of the object regions in the input defect images, while the other discriminator focuses on edge detail differences between the ground truth and the segmented image. To obtain more edge information during training, a new composite loss measurement containing edge information and structural features is designed. Experimental results show that CASDD can be suitable for defect detection on four real-world and one artificial defect database, and its detection accuracy is significantly better than the state-of-the-art deep learning methods.
Original languageEnglish
Pages (from-to)19583 - 19595
Number of pages13
JournalIEEE Sensors Journal
Volume22
Issue number20
DOIs
Publication statusPublished - 15 Oct 2022

Keywords

  • Image segmentation
  • Feature extraction
  • Steel
  • Image edge detection
  • Surface treatment
  • Databases
  • Generative adversarial networks

Fingerprint

Dive into the research topics of 'CASDD: Automatic Surface Defect Detection Using a Complementary Adversarial Network'. Together they form a unique fingerprint.

Cite this