This is the reference implementation of the models and code for the U-Net which was proposed by O Ronneberger,P Fischer and T Brox. You can set parameters like file directory, image size, number of classes etc. in configuration.txt, then run U-Net_Training.py to train the model, run U-Net_Predict.py to segment images.
joannapang / u-net-segmentation Goto Github PK
View Code? Open in Web Editor NEWThis project forked from fosaken/u-net-segmentation