I get noise only

[mykeehu]

I installed this app under windows, setup everything, but I get this noise for any model. What's happened?

I have RTX 3060, is this compatible?

[k-l-lambda]

Try run python ./test.py, then check images/example.png. If it is also noise, that maybe due to your model weight files' problem.

[mykeehu]

This is very interesting, because the sample picture is perfect. So what is the problem?

[k-l-lambda]

Try to replace this line in file http_server.py:

	image.save(fp, PIL.Image.registered_extensions()['.png'])

with:

	image.save('./test.png')

And see if the output is normal.

[mykeehu]

Unfortunately it didn't help, now a white image is saved, although the console says there is data in it.

generation cost: 0.3530001640319824
127.0.0.1 - - [10/Apr/2024 07:11:46] "GET /generate?psi=0.5&randomize_noise=1&latents=Y%2FYpPQcE3Dy2EuC8Q%2F2WPfEB6Tqv1xQ9DA6QPXvmBj5pu4i7wK4mvS660TulpoS9JTk5vXp6Hb2MErs9TOaqvbISDTrRiFi8m9rpPCnwij2ibo67JSRRPQKO2TzGVum9Ch6WvK2CVjy0zhE9kQBGPC%2FVDT15lQG9e+meu41QkT1LCrS8GdjUvDbtkL0XIZS9UTTGPAIMN71pQj07tlTNPNHZij3LO6Y8tc0qPRhp1rsaWdc78pPwvNjrsTx4YgS9PtiIvVHpF73nCCo9IXqzuvp%2F77yBv148kuH8ugGXKj3Xhp67dtGwu4Yj1LxNwVO8I7abvHPAYDyHd0K8Ls%2FqPIdb3zscjvQ8khLGvLdQVD2Xj6c8XYmCvW+merw5eJs9Gop2PXFRZ7z1f0Q9XWq2PA1xDD0dtES93PGAvAIPDj3BhPm8Q+BGPBZbEbt2aVq9jv8su9kM6LzWRvK8juvWvdLoL700zva8rXSIu03EBr448FU72vcVvclBQT0jP6M8C+0mPfTrdb2EXS47folhPFp8aLxCQQO95ksOPRiALD076TI8grILvTWt1Dshkbo9cvtIvXsqHz33uKQ8XE7RPYKnbLyD+Sc9mmWAvRL3dr0DUns9SfnVPFq5l7x+XAO9oTPivCz+QzydVO08%2FJ%2FVPET2tTw7L+09D6zMPKl8gr3+CWw8JjuqvHVPrD3jWDg9EhLiPZ2NO7zF18K7fJ04PXUb1zwK3zE9Z4oOvRtlQj2Eo+w8vdaBPZxaBj1LjOW9qOLYvO1S4DzHP4W7bOW1vFAANzxVZxg+xFALvZ8mcbyGibw8A7edvbaYZrsLfwo9LCBEPPaEGr1KKk084xyKOjcWMT3kQ%2FA8oVpzvAFlYjs8HGQ99ckfPcE9vLzokeo84EeNPMkMsjy3PH09LFGZPMLBMr3a%2FUm9PbkDPX1xjTzK2RQ7f%2FQOPSN9nr3jGig9R0i8vGQEjD1P7xO7CFR2vQP0nrvMGqM86w1gPU9VRr2GDIc8Qgs8PcbI4T2Iwh+95vGEO9zquLx%2FLNC8VgR6PIIYVL1N%2FXE81Q2pO6uvnTxz5cq9T2EePb4qCz3xNyM85Q1NPeRtxjv15xk9R64TPWRKvTwc4mG9EhqmO%2Ffphrxkl6o6TEZ6PdQKnjzXt7k9LxK2PE9SpTyBpQk9R1OLvMwSUj3zT3A94j6UvWVo0b3fihY7VmMSvQ+YHLxCelY9MqU1PdvU7Lzsd9Q8tjcvvWdQVT1w0am8CBHsvCyzMzwkU7o8LZzNO35RjL0Q2BU8f0%2FvPIn4ML1NO648y5gQPTiy77zQZyq8tz5APOIaiz2Nvx+9MNBrPPS437vLohi9digkvNatOj0hAP473ilfPEKYjDwjmow8sI3YPHPU4zwW8Sc9eDpgPVnGDD2iTxg9JpQ7OzwoQz0b%2Fik96QhFvUqKyD3NzkU6VlUivXpPnbxMZra8i4Q2vRBGuDs0Xi89cHG7PKqUEjyhVBo9U4ALvXHjn7wFYzC9rYuKPSWRhD0XJfM8wvJLPevyubz0CQc8Xm66vR+Rq7ssp+I88uqqvCVuEjwfQXa9WxTUvBbcMj2JZRa9zt4dPcA3Ir3o6Kq7f55hPBXOOz3VvRO9GdMrOzsgi7z%2FJmm8+u+JPUehljyb+C49IV0kPT67YLws5dI9kJrPOhTSZzwhz9c5TJnNPB0FC703Iwq9A8wQvY0mlj3nIdu8NqiIPATB4zws+a08bUpuPeuOCL2rbay88UyxvOEMjD1Eyfk73PKHPZ+FW71CLoK6j4ChPalaYz05YyW836a1vBqpD726yks6yRxuvdIbFb02wTk9JdVFPfhUgjyLYLc8ehs5O7n+XTyQJeu9bhB1vQDpHz00IK+9%2FGpqvZBR1bxxssy8D6nVPOshVj1kOvc8KzTevXmHjjxpJYs9IE5yvX2R6TyZesY8mgw7PQ11RzzKI9S9SgSlPQDXcz3ABNK7FPcnvfZ3ij2Dwja9sHfbPUfOOL0+tIa7f4AvPVRJE7wign49UcWXvfJ1kL2b85G9F9gGO2lRwjz6+gU7GCMBOiDcuLwuPFu95eGFvGKetzhH1To9594bvc8hUjpLu4E8zQJcvflwwDx5qfO771GZvfk7T7034pm8BBd2PA%2FsT7u8l+29QKT2PIbdSTx%2F9x+7emXIvMUV17woEO07vcRQvRyPm7xE3868e7UxvI+GoTxVRh68CdUVvPGUCLyc88I85FVDPSuUc7wPIrI7TITCvGedNrzRjqo8PIZGvbJrk7y3bJi8rizNuwsoZL28uAw9b9dkvYhR2Dwj2LM80xCVvTnwpTrwFXU93S9BvcCF0DyfxRE8p5mWPd4GEr0P9He9Oklzva7Vbrs%2FpQq9vdbBvJvHOzub2aK9l2BiPJ8BfLx85YA9mrmAvE2p1D3E2xa9RWkLPPPyqju5mf27skLtPSmblTkG4U68cok7vdUITLxf3Z89V44tPWhDWLwiOhK9Peo2PTlVWD0aSiA9IZJJvGhoh7s02b08lfllvYcqdz1AXDE9neouPR1ML71SqKQ6kT4FvU4m9bwxKge87IqcvZRIaT1B8JM9a%2FkVva0wqb1vNpA99BItPciS2ztLeWk9ckiCPHbGIL3CLTi9xGyqvSc+gTxYmJG80aBUvfiynLz4i7Y8km2rPBotubxzWK88QgECPG3DOD1DphU8dHa%2FPbL+1L0l8t88oJD6vGAOkT2XSni7qYUEOwtbG70%3D HTTP/1.1" 200 -

[k-l-lambda]

So, did you see the image of test.png?

[mykeehu]

Yes, png file has generated, but white only.

I use this in .env.local

MODEL_NAME=ffhq
MODEL_PATH_ffhq=./models/stylegan2-ffhq-config-a.pkl

This is the default generated image I get after loading the model:

127.0.0.1 - - [10/Apr/2024 07:20:42] "GET /generate?psi=0.5&randomize_noise=1&latents=AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA%3D HTTP/1.1" 200 -

[mykeehu]

I reinstalled venv with Flask 1.1.1 (it was version 2), but it won't start after that:

Traceback (most recent call last):
  File "I:\stylegan-web\http_server.py", line 7, in <module>
    import flask
  File "I:\stylegan-web\venv\lib\site-packages\flask\__init__.py", line 14, in <module>
    from jinja2 import escape
ImportError: cannot import name 'escape' from 'jinja2' (I:\stylegan-web\venv\lib\site-packages\jinja2\__init__.py)

[k-l-lambda]

stylegan2-ffhq-config-a.pkl seems not the final weight file of stylegan2. And zero latent should generate an average face for ffhq, this is not right.

[mykeehu]

I use this requirements.txt for perfect install:

tensorflow-gpu==1.14.0
python-dotenv>=0.10.1
Flask
numpy>=1.8.0
Pillow>=6.0.0
requests>=2.21.0
protobuf<3.20

And models from here (from gdrive), this has not changed in four years.:
https://github.com/NVlabs/stylegan2/blob/master/pretrained_networks.py

[mykeehu]

What I don't understand is why, if the test runs correctly (I have replaced the model), the generator doesn't take a good picture.

Update1: Progress: the base image came in after some waiting, but when I click on the Randomize button, colourful things appear.

[k-l-lambda]

What I don't understand is why, if the test runs correctly (I have replaced the model), the generator doesn't take a good picture.

Update1: Progress: the base image came in after some waiting, but when I click on the Randomize button, colourful things appear.

test.py use the weight file of stylegan2-ffhq-config-f.pkl, try that.

[mykeehu]

We are making progress. The sample image is already displayed, but whenever I change any of the parameters, I get colored noise:

[k-l-lambda]

We are making progress. The sample image is already displayed, but whenever I change any of the parameters, I get colored noise:

Right click the bad image, open in a new tab. Then restart python server, and refresh image page. Check if it change.

[mykeehu]

@k-l-lambda this trick works, i.e. if I refresh the image based on URL, the image is displayed instead of the noise.

[k-l-lambda]

It seems your hardware's problem. RTX3060's graph memory may be not enough. However you can try his 256x256 models, such as cat and horse.