Pixel Link with keras

Prerequisite

  • python-3.6
  • keras-2.1.5+
  • numpy-1.14.3
  • Pillow-5.5.0

Structure

All main source code is in the root directory of the project.

  • cfg.py -- this is configure file
  • data_generator.py
  • label.py
  • losses.py
  • model.py pixel link model
  • train.py
  • predict.py
  • util this is a folder with some util(log,image...)

Train

Retrain

Test

Noted