Skip to content

Pytorch Implement of the paper "Segmentation-Based Deep-Learning Approach for Surface Defect Detection"

Notifications You must be signed in to change notification settings

seanXYZ/SegDecNet

Repository files navigation

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

About

Pytorch Implement of the paper "Segmentation-Based Deep-Learning Approach for Surface Defect Detection"

https://arxiv.org/abs/1903.08536

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages