Skip to content

changGitHubJ/U-net_keras

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages