stylegan2-pytorch icon indicating copy to clipboard operation
stylegan2-pytorch copied to clipboard

Are there generation options?

Open vortextemporum opened this issue 4 years ago • 4 comments

I am on half way training my transparent image dataset with this repo (first of all, thank you for this cool library).

However, I can't see options to generate images with seeds (or interpolations) like I can do in StyleGan2 tensorflow library. What are my options to generate thousands of images?

Thank you

vortextemporum avatar Feb 08 '21 20:02 vortextemporum

since it generates a .pkl file, you can train the model from this then do all sorts of things with that .pkl file in the tensorflow version

aertist avatar Feb 08 '21 20:02 aertist

@vortextemporum there's a section dedicated to that! https://github.com/lucidrains/stylegan2-pytorch#advanced-use it's as easy as one command for each of what you want to do

lucidrains avatar Feb 08 '21 21:02 lucidrains

Thank you both for your responds!

since it generates a .pkl file, you can train the model from this then do all sorts of things with that .pkl file in the tensorflow version

My trained model files are .pt files. So does it mean that I should convert them back .pkl? Also, original Stylegan2 didn't allow for transparent images. If I just convert the model directly to pkl would it mess up?

@vortextemporum there's a section dedicated to that! https://github.com/lucidrains/stylegan2-pytorch#advanced-use it's as easy as one command for each of what you want to do

I checked this section and using the commands in both here and lightweight-gan, but I don't see an option to set a seed range (like 10-8000) on generation, nor for interpolation.

vortextemporum avatar Feb 10 '21 20:02 vortextemporum

Thank you both for your responds!

since it generates a .pkl file, you can train the model from this then do all sorts of things with that .pkl file in the tensorflow version

My trained model files are .pt files. So does it mean that I should convert them back .pkl? Also, original Stylegan2 didn't allow for transparent images. If I just convert the model directly to pkl would it mess up?

@vortextemporum there's a section dedicated to that! https://github.com/lucidrains/stylegan2-pytorch#advanced-use it's as easy as one command for each of what you want to do

I checked this section and using the commands in both here and lightweight-gan, but I don't see an option to set a seed range (like 10-8000) on generation, nor for interpolation.

Check out this repo and his stiff on YT. It's really helpful in the context of this problem

aertist avatar Feb 11 '21 00:02 aertist