cyclomon / UNSB

Official Repository of "Unpaired Image-to-Image Translation via Neural Schrödinger Bridge" (ICLR 2024)
MIT License
169 stars 8 forks source link

Whether this model surpports controllable generation? #21

Closed kinredon closed 3 months ago

kinredon commented 4 months ago

For example, I can provide a trade-off parameter to control the degree of style transfer to the target domain style.

cyclomon commented 3 months ago

Thank you for your interest on our work! Since our model is targeting on domain-to-domain mapping, there is no explicit component for controlling the degree of stylization. However, as we have already shown in our manuscript, we can obtain weakly-stylized output when using lower NFE (e.g. NFE=1), and fully stylized (fully translated) output when using higher NFE.