nikhilroxtomar / Polyp-Segmentation-using-UNET-in-TensorFlow-2.0

Implementing polyp segmentation using the U-Net and CVC-612 dataset.
https://idiotdeveloper.com/polyp-segmentation-using-unet-in-tensorflow-2/
MIT License
66 stars 31 forks source link

Use of predict script for data other than test train and validation. #2

Closed saqibshakeel035 closed 4 years ago

saqibshakeel035 commented 4 years ago

I have followed your article to train a model based on my custom data set successfully. Now what I want is to use it for the segmentation of 1 picture per iteration. Any kind of help in this regard would be appreciated.