What can I do to use the dice metric in this code?

zizhaozhang / unet-tensorflow-keras

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

MIT License

186 stars 72 forks source link

What can I do to use the dice metric in this code? #14

Open li-pengcheng opened 5 years ago

li-pengcheng commented 5 years ago

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!