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