search

mchong6 / JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization
MIT License
1.42k stars 206 forks source link

A question about how to preserve colors #24

Closed PanggNOTlovebean closed 2 years ago

PanggNOTlovebean commented 2 years ago

The following methods are used in the source code to preserve the color if preserve_color: id_swap = [9,11,15,16,17] else: id_swap = list(range(7, generator.n_latent))

I wanna know how to get the exact ID of these layers. Are there some papers that introduce such things? Or through a lot of attempts to know which layers control the color?

mchong6 commented 2 years ago

I tried manually to see which layers control colors the most.