Thank you for your code.
It seems that there is a small issue with the inputs of the Mumfordshah_loss.
loss_mumfordshah = mumfordshah_loss(volume_batch, outputs_soft) should be modified to loss_mumfordshah = mumfordshah_loss(outputs_soft, volume_batch).
Thank you for your code. It seems that there is a small issue with the inputs of the Mumfordshah_loss.
loss_mumfordshah = mumfordshah_loss(volume_batch, outputs_soft)
should be modified toloss_mumfordshah = mumfordshah_loss(outputs_soft, volume_batch)
.