unet-tensorflow-keras icon indicating copy to clipboard operation
unet-tensorflow-keras copied to clipboard

A concise code for training and evaluating Unet using tensorflow+keras

Results 9 unet-tensorflow-keras issues
Sort by recently updated
recently updated
newest added

Hello Mr Zhang, sorry to interrupt you i am a university student from China Could you please tell me if there is a dataset that can be used to train...

Hi, thank you for sharing code. Could you give me the dataset link?

With the default value of 256 for --imSize, the eval.py produces 768x256 images that comprises of the validation img image, the validation gt image, and the output from the uNet...

* plus some other minor formatting proposals

In train directory train/ img/ 0/ gt/ 0/ in 0/ need I to put one image or batch? In my case, I have 6 classes. How to look groundtruth mask...

at line136 in train.py, i add: dice = dice_coef_loss(y_batch[0], pred_map) but it seems don't work, could you please give me some information about how to use this metric, thx!

``` x_batch, y_batch = next(train_generator) feed_dict = { img: x_batch, label: y_batch } ``` This way is very slowly.