Generating Dice Coefficient When Predicting Images

[DXVSantoso]

Hi all,

I was wondering if there was a way to calculate the dice score when predicting an image.
Or rather calculating the dice score with two images, the original and predicted image.

I was wondering maybe a modification of the predict.py code to calculate the dice and have it store in a text file as an output would be nice, but I do not know now where to begin. Implementing the dice_score.py into the predict.py would be my first guess. Am I on the right track?

To add to this is there any way to log the dice score using a single model on the whole trained dataset to be plotted in a bar plot?
Looking forward to hearing some of the answers!