How to generate images using model.predict just as the validation ones?

tdeboissiere / DeepLearningImplementations

Implementation of recent Deep Learning papers

MIT License

1.81k stars 652 forks source link

How to generate images using model.predict just as the validation ones? #94

Closed marlon-shiftone closed 2 years ago

marlon-shiftone commented 2 years ago

I'm using your code and I'm loading the weights and making predictions. However, the images are getting out in color.

How can I get the same output format as model validation plot? I am feeding the NN with cancer images and getting out segmented black and while images.

Thanks