Overview
- Deep learning for defection
- U-Net algorithm
- DAGM dataset
- keras + tensorflow
- python 3.6.9
- tensorflow-gpu 1.14.0
- keras 2.3.1
-
Configurate image and lable
place class1_def of DAGM dataset to ./data/
start configurate_data.py -
Training with tensorflow
start train.pys -
Training with keras+tensorflow
start train_keras.py -
Compare results
start test.py