AlamiMejjati / Unsupervised-Attention-guided-Image-to-Image-Translation

Unsupervised Attention-Guided Image to Image Translation
MIT License
329 stars 47 forks source link

The second stage #22

Open ghost opened 5 years ago

ghost commented 5 years ago

When I read your paper, it says:

Our algorithm is characterized by two training stages. In the first stage we train FS, FT , AS, AT and both discriminators DS and DT . In addition, the discriminators are trained with the holistic images as input. In the second stage we interrupt the training of attention networks AS and AT and train the discriminators using the foregrounds only.

But in your code I didn't find out where it sets the training of discriminators using the foregrounds only. Could you point it out? Thank you in advance.