Pytorch Implement of the paper "Segmentation Based Deep-Learning Approach for Surface DefectDetection" https://arxiv.org/abs/1903.08536
Modified little things without changing the accuracy
step1: train segment net
python train_segment.py
step2: train decision net
python train_decision.py
step3: test
python test.py