search

HalbertCH / IEContraAST

This is the official PyTorch implementation of our paper: "Artistic Style Transfer with Internal-external Learning and Contrastive Learning".
MIT License
78 stars 7 forks source link

preserve colors #5

Open surfingnirvana opened 1 year ago

surfingnirvana commented 1 year ago

Is there any way to modify the code to preserve colors partly or fully? Thank you.

HalbertCH commented 1 year ago

Is there any way to modify the code to preserve colors partly or fully? Thank you.

Hi @surfingnirvana, thanks for your interest. Do you mean to enforce the color of the stylized image to be more consistent with that of the style image? I think you can try to increase the weights of Ls and decrease the weights of L{adv}.