taesungp / contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)
https://taesung.me/ContrastiveUnpairedTranslation/
Other
2.23k stars 417 forks source link

3D CUT for medical images #170

Open CasperDL2805 opened 1 year ago

CasperDL2805 commented 1 year ago

Hi!

I have altered your code to work with 3D images, however, I do not see a gradual decrease in NCE loss (see image). I could not find any images of your loss, and I know it is highly dependent on the data etc - but do you see a gradual decrease? In that case, it may indicate that I have done something wrong... I have trained the CycleGAN with the same data, and here I see a gradual decrease in both Cycle loss and Identity loss.

Thank you in advance!

image