Abstract:Precise locating and correct classification of each defect in the image is the primary goal of surface defect detection (SDD). This is very similar to object detection, which seeks to predict a set of bounding boxes (Bboxes) and category labels for each object of interest. However, due to the fact that the category dependencies among defects are obviously different from those among general objects, the methods developed for object detection cannot be directly utilized for detecting defects. To address this issue, we proposed a category-related non-maximum suppression (CR-NMS) method. Different from most NMS methods, CR-NMS use Cover Percent (CoP), instead of Intersection over Union (IoU), to guide the Bbox regression. Moreover, a two-stage Bbox regression algorithm is proposed to remove the duplicate Bboxes. This algorithm works in a course-to-fine manner and takes the correlation between Bboxes into account. By this means, the CR-NMS can remove the duplicate Bboxes more effectively and is easy to embed in existing neural networks. A wafer surface defect dataset including 6,000 images and 11 defect categories was set up for training and evaluating our method. Experiments demonstrated that in the detectors, which were built on Faster R-CNN and RetinaNet with backbone ResNet-101, and YOLOX with backbone CSPDarknet, compared with STD NMS or Soft NMS, CR-NMS increased by 3.4%, 9.3%, and 5.5% on mean average accuracy (mAA), and their values were as high as 92.3%, 91.8%, and 90.8%, respectively. The same tests were also performed on dataset NEU-DET. The results showed that compared with STD-NMS and Soft NMS, the CR-NMS can also obviously improve the performance of all the detectors, and their mAA was improved by at least 1.2%.

High-Speed and Accurate Cascade Detection Method for Chip Surface Defects

A Wafer Surface Defect Detection Method Built on Generic Object Detection Network

A Simple Confidence-Supervised Model for High-Resolution Defect Recognition

Attention-based deep learning for chip-surface-defect detection

Coarse-to-Fine Few-Shot Defect Recognition with Dynamic Weighting and Joint Metric

Attention-Based Multiscale Feature Fusion for Efficient Surface Defect Detection

Particleboard Surface Defect Inspection Based on Data Augmentation and Attention Mechanisms

A Deep Learning-Based Surface Defect Inspection System Using Multiscale and Channel-Compressed Features

Saliency-Based Defect Detection in Industrial Images by Using Phase Spectrum

Integrated Circuit Packaging Defect Analysis and Deep Learning Detection Method

A lightweight and efficient model for surface tiny defect detection

Joining Spatial Deformable Convolution and a Dense Feature Pyramid for Surface Defect Detection

A real-time anchor-free defect detector with global and local feature enhancement for surface defect detection

A small sample nonstandard gear surface defect detection method

TL-SDD: A Transfer Learning-Based Method for Surface Defect Detection with Few Samples

Metal Surface Defect Detection Method Based on Improved Cascade R-CNN

Small object detection method with shallow feature fusion network for chip surface defect detection

A Method of Defect Detection for Focal Hard Samples PCB Based on Extended FPN Model

An Efficient Network for Surface Defect Detection

Deep Learning-Based Integrated Circuit Surface Defect Detection: Addressing Information Density Imbalance for Industrial Application

A cascaded combination method for defect detection of metal gear end-face