search

eladrich / pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework
https://eladrich.github.io/pixel2style2pixel/
MIT License
3.19k stars 570 forks source link

Using pre-trained weights for LSUN Horses from e4e. #286

Closed arturandre closed 2 years ago

arturandre commented 2 years ago

Hi, I'm working with the LSUN Horses dataset and I noticed that in the ReStyle repo there are pre-trained weights for the e4e encoder. Is there a straightforward way to use the e4e encoder from ReStyle repository to perform sketch-to-image translation in this repository? Thank you.

yuval-alaluf commented 2 years ago

Sure. What I would do is add support for the sketch-to-image dataset from this repo into the ReStyle repo. Other than changing the dataset, there shouldn't be any other changes needed (other than maybe adjusting the loss weights). You can take a look in this repo of how to define the sketch-to-image dataset and add that to the ReStyle code.

arturandre commented 2 years ago

Thank you very much! I'll look into that.