pixel2style2pixel icon indicating copy to clipboard operation
pixel2style2pixel copied to clipboard

Do you ever retrain the stylegan network?

Open Zhentao-Liu opened this issue 2 years ago • 3 comments

pSp is a really great job for image to image translation. But there is a question that do you ever retrain the stylegan network for your applications (inversion, impainting, super-resolution)? I find that the stylegan model file you provide in your project is different from the NVIDIA official implementation. So I wander if you used the fixed stylegan generator (the same as NVIDIA provided) or retrain it.

Zhentao-Liu avatar Sep 17 '22 13:09 Zhentao-Liu

The StyleGAN generator we use here is the official generator that was converted from tensorflow to pytorch. We do not train our own generator.

yuval-alaluf avatar Sep 17 '22 16:09 yuval-alaluf

Thanks for your reply! I find that I do my research on stylegan2-ada-pytorch, that's maybe different from your stylegan implementation. So you mean that you just copy the weights of the original stylegan for your decoder part without retraining it?

---Original--- From: @.> Date: Sun, Sep 18, 2022 00:02 AM To: @.>; Cc: "Zhentao @.@.>; Subject: Re: [eladrich/pixel2style2pixel] Do you ever retrain the stylegannetwork? (Issue #287)

The StyleGAN generator we use here is the official generator that was converted from tensorflow to pytorch. We do not train our own generator.

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

Zhentao-Liu avatar Sep 17 '22 16:09 Zhentao-Liu

Correct, we don't retrain it. We used rosinality's implementation of StyleGAN2 rather than StyleGAN2-ada-pytorch, but you can convert the models by following the instructions here.

yuval-alaluf avatar Sep 23 '22 06:09 yuval-alaluf