U-Net

Split medical images with U-Net

Dependency

Keras and Numpy

How to use

1. Put your image dataset in data directory.
2. Run data.py.
3. Run unet.py.
4. Get the results in results directory.

How to train for more epochs

Just run unet.py again. It will automatically load previously trained model and continue training it. If you want to train a brand-new model, just delete unet.hdf5, which is the checkpoint file.