wangpeng000 / visualinspection

The code for "Self-supervised Context Learning for Visual Inspection of Industrial Defects"

SCL-VI: Self-supervised Context Learning for Visual Inspection of Industrial Defects

We address the challenge of detecting object defects through the self-supervised learning approach of solving the jigsaw puzzle problem.

Results

Dependencies

Since I did this project a long time ago, there may be some potential issues with environmental dependencies.

• Tested with Python 3.8
• Pytorch v1.6.0

Dateset

• Dataset : MvTec AD

Run Training

• python train.py --obj=cable --lambda_value=1 --D=64 --epoches=400 --lr=1e-4 --gpu=0

Run Affinity Testing

• python test.py --obj=cable --gpu=0
• enc.load(obj, N) N is the serial number of the obtained training weight file

Anomaly maps

• python heat_map.py --obj=cable
• enc.load(obj, N) N is the serial number of the obtained training weight file

Details:

• The input of the network should be 256x256
• data.npy contains the relative positions and their reference numbers.

visualinspection's Issues

训练时长

这代码训练需要多长时间？我看了代码好像主要时间都在

这个上面，请问这个需要花费多长时间？

where is the code?

When to open source？