scumechanics/Surface-Defect-Detection

Segmentation-Based Deep-Learning Approach for Surface-Defect Detection

Surface-Defect-Detection

PyTorch implementation of "Segmentation-Based Deep-Learning Approach for Surface-Defect Detection" .

Done

• add Depthwise+Pointwise
• add Inception
• add SE

Get started quickly run_demo.ipynb

Paper Model

Performance

Trained model：

segment_net_100.pth

decision_net_100.pth

DATASET：

KolektorSDD_Data

Fork https://github.com/seanXYZ/SegDecNet