vipermu / StyleCLIP

Using CLIP and StyleGAN to generate faces from prompts.
MIT License
129 stars 28 forks source link

What's up with the colors? #1

Closed johnpaulbin closed 3 years ago

johnpaulbin commented 3 years ago

Putting anything will give a pretty good output, but in some situations the colors will become all mixed.

80

Is there any reason behind this?

johnpaulbin commented 3 years ago

Also how could I fix if possible ^

DuncanRowland commented 3 years ago

In clip_generate.py line 111: img = img[0].permute(1, 2, 0).detach().cpu().numpy() 256 should be img = img[0].permute(1, 2, 0).detach().cpu().numpy() 255

vipermu commented 3 years ago

Nice catch @DuncanRowland . That's how it should be, will fix it. Thanks!