The U net seems that it is not being trained (which is in accordance to the paper). However, you do not load any weights (pretrained model) into your U Net model as far as I was able to see. This means that unet weights are as initialized.
Is this a correct observation? Was the purpose of the code just showcasing the implementation of Probabilistic UNet, because I assume it will be difficult to train without a trained segmentation network.
The U net seems that it is not being trained (which is in accordance to the paper). However, you do not load any weights (pretrained model) into your U Net model as far as I was able to see. This means that unet weights are as initialized.
Is this a correct observation? Was the purpose of the code just showcasing the implementation of Probabilistic UNet, because I assume it will be difficult to train without a trained segmentation network.