PDillis / stylegan3-fun

Modifications of the official PyTorch implementation of StyleGAN3. Let's easily generate images and videos with StyleGAN2/2-ADA/3!

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

convert_to_grayscale

pgtinsley opened this issue · comments

commented

I lost hours to this "trolling to Pillow" since I'm actually working with a grayscale SG network 😢
Is there a better way to name this or at least call attention to the fact this is for 3-channel models?
Thanks!

if convert_to_grayscale:
    image = image.convert('L').convert('RGB')  # We do a little trolling to Pillow (so we have a 3-channel image)

Ah! So sorry this was a waste of your time. This line was just another way of gearing all the channels to the same, as then the Discriminator will generate completely different images during its synthesis stage. Might I ask what you got with/without it? Since I don't have a grayscale SG network, I don't know how to help for now, but let me know!

Hello @pgtinsley. I am currently trying to use a grayscale SG network. Did you only set your dataset to use 1 channel via the -force-channels parameter, or were there any other changes that needed to be made?