Question about loss decline about rvgan HOT 2 CLOSED

The d_loss will eventually diminish to 0.001. So the strategy was to re-train the model again after 100 epochs by loading the coarse_generator and fine_generator's weight. But we did not load the discriminator's weights while retraining. Unlike classification and segmentation architectures, GAN based generators wont show a pattern of diminishing loss (from high to low).

You can read more about how to train a GAN or other GAN hacks by Soumith Chintala (creator of PyTorch and DC-GAN) here:
Link : https://github.com/soumith/ganhacks

So a good way to see if your generators are predicting good segmentation output is to visualize the local_plot.png and global_plot.png generated after each epoch in our code. This will show you vessel segmentation output of random images after each epoch.

If the output images are good, it means the generators are learning to translate between the two modality for that specific weight.

For further validation, we loop over all the saved generator weights and print the associated metrics ( sensitivity, specificity, AUC, SSIM etc. ) to see which weight pair (both coarse and fine) gives us the best segmentation output for the test images.

Hope this helps, thanks !

from rvgan.

thanks a lot!

from rvgan.