search

orobix / retina-unet

Retina blood vessel segmentation with a convolutional neural network
1.26k stars 468 forks source link

Update keras code #46

Open pyaf opened 6 years ago

pyaf commented 6 years ago

Hi, I'm going through your code, I see some code which has been deprecated in due course of time. like here visualize_util has been changed to vis_utils and plot to plot_model.

I would like to fix such bugs by raising a PR, let me know your consent.

dcorti commented 6 years ago

Hi Yes, keras keeps updating and therefore many parts of the become obsolete. Please go ahead with the PR, thank you

ZexuanTHU commented 6 years ago

@pyaf I clone this repo and try to work through the training process, there do have several warnings and even rejects under the latest Keras environment. About the plot_model issue, following is a solution. I think just simply remove the import and references of plot_model from /src/retinaNN_training.py will work.