README.md

Defection using U-Net written in keras

Overview

• Deep learning for defection
• U-Net algorithm
• DAGM dataset
• keras + tensorflow

Requirement

• python 3.6.9
• tensorflow-gpu 1.14.0
• keras 2.3.1

Usage

1. Configurate image and lable
   place class1_def of DAGM dataset to ./data/
   start configurate_data.py

2. Training with tensorflow
   start train.pys

3. Training with keras+tensorflow
   start train_keras.py

4. Compare results
   start test.py

Licence

MIT