Duplicate from prusnak/troublinggan

Co-authored-by: Pavol Rusnak <prusnak@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "stylegan2"]
+	path = stylegan2
+	url = https://github.com/NVlabs/stylegan2-ada-pytorch.git

README.md

@@ -0,0 +1,13 @@
+---
+title: Troublinggan
+emoji: 📊
+colorFrom: pink
+colorTo: purple
+sdk: gradio
+sdk_version: 3.29.0
+app_file: app.py
+pinned: false
+duplicated_from: prusnak/troublinggan
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,63 @@
+import os
+import sys
+
+import gradio as gr
+import numpy as np
+import torch
+from PIL import Image
+
+sys.path.append(os.path.join(os.path.dirname(__file__), "stylegan2"))
+
+import dnnlib
+import legacy
+
+DEVICE = "cpu"  # or "cuda"
+
+device = torch.device(DEVICE)
+with dnnlib.util.open_url("troublinggan-square.pkl") as f:
+    G_square = legacy.load_network_pkl(f)["G_ema"].to(device)
+with dnnlib.util.open_url("troublinggan-rectangle.pkl") as f:
+    G_rectangle = legacy.load_network_pkl(f)["G_ema"].to(device)
+
+
+def generate(sigma, model):
+    G = {"v1": G_square, "v2": G_rectangle}[model]
+    label = torch.zeros([1, G.c_dim], device=device)
+    z = torch.from_numpy(np.random.RandomState().randn(1, G.z_dim) * sigma).to(device)
+    img = G(
+        z,
+        label,
+        truncation_psi=1,
+        noise_mode="random",
+        force_fp32=(DEVICE == "cpu"),
+    )
+    img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)
+    img = Image.fromarray(img[0].cpu().numpy(), "RGB")
+    if model == "v2":
+        img = img.crop((0, 172, 1024, 1024 - 172))
+    return img
+
+
+iface = gr.Interface(
+    fn=generate,
+    inputs=[
+        gr.Slider(
+            value=1.0,
+            minimum=0.5,
+            maximum=2.0,
+            step=0.25,
+            info="perceived randomness of the image",
+        ),
+        gr.Radio(
+            ["v1", "v2"],
+            value="v1",
+            info="v1: model containing photos from 2020 (resolution 512x512); v2: model containing photos from 2020-2021 (resolution 1024x680)",
+        ),
+    ],
+    outputs=["image"],
+    title="TroublingGAN",
+    description="Generate a random image from the TroublingGAN model",
+    article="Read more about TroublingGAN at [http://hamosova.com/TroublingGAN](http://hamosova.com/TroublingGAN)",
+    allow_flagging="never",
+)
+iface.launch()

pyproject.toml

@@ -0,0 +1,22 @@
+[tool.poetry]
+name = "troublinggan"
+version = "0.1.0"
+description = ""
+authors = []
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = ">=3.8,<3.12"
+gradio = "^3.32.0"
+torch = "^2.0.1"
+numpy = "^1.24.3"
+pillow = "^9.5.0"
+scipy = "^1.10.1"
+
+[tool.poetry.group.dev.dependencies]
+black = "^23.3.0"
+isort = "^5.12.0"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

requirements.txt

@@ -0,0 +1,73 @@
+aiofiles==23.1.0 ; python_version >= "3.8" and python_version < "3.12"
+aiohttp==3.8.4 ; python_version >= "3.8" and python_version < "3.12"
+aiosignal==1.3.1 ; python_version >= "3.8" and python_version < "3.12"
+altair==5.0.0 ; python_version >= "3.8" and python_version < "3.12"
+anyio==3.6.2 ; python_version >= "3.8" and python_version < "3.12"
+async-timeout==4.0.2 ; python_version >= "3.8" and python_version < "3.12"
+attrs==23.1.0 ; python_version >= "3.8" and python_version < "3.12"
+certifi==2023.5.7 ; python_version >= "3.8" and python_version < "3.12"
+charset-normalizer==3.1.0 ; python_version >= "3.8" and python_version < "3.12"
+click==8.1.3 ; python_version >= "3.8" and python_version < "3.12"
+colorama==0.4.6 ; python_version >= "3.8" and python_version < "3.12" and platform_system == "Windows"
+contourpy==1.0.7 ; python_version >= "3.8" and python_version < "3.12"
+cycler==0.11.0 ; python_version >= "3.8" and python_version < "3.12"
+fastapi==0.95.2 ; python_version >= "3.8" and python_version < "3.12"
+ffmpy==0.3.0 ; python_version >= "3.8" and python_version < "3.12"
+filelock==3.12.0 ; python_version >= "3.8" and python_version < "3.12"
+fonttools==4.39.4 ; python_version >= "3.8" and python_version < "3.12"
+frozenlist==1.3.3 ; python_version >= "3.8" and python_version < "3.12"
+fsspec==2023.5.0 ; python_version >= "3.8" and python_version < "3.12"
+gradio-client==0.2.5 ; python_version >= "3.8" and python_version < "3.12"
+gradio==3.32.0 ; python_version >= "3.8" and python_version < "3.12"
+h11==0.14.0 ; python_version >= "3.8" and python_version < "3.12"
+httpcore==0.17.1 ; python_version >= "3.8" and python_version < "3.12"
+httpx==0.24.1 ; python_version >= "3.8" and python_version < "3.12"
+huggingface-hub==0.14.1 ; python_version >= "3.8" and python_version < "3.12"
+idna==3.4 ; python_version >= "3.8" and python_version < "3.12"
+importlib-resources==5.12.0 ; python_version >= "3.8" and python_version < "3.10"
+jinja2==3.1.2 ; python_version >= "3.8" and python_version < "3.12"
+jsonschema==4.17.3 ; python_version >= "3.8" and python_version < "3.12"
+kiwisolver==1.4.4 ; python_version >= "3.8" and python_version < "3.12"
+linkify-it-py==2.0.2 ; python_version >= "3.8" and python_version < "3.12"
+markdown-it-py==2.2.0 ; python_version >= "3.8" and python_version < "3.12"
+markdown-it-py[linkify]==2.2.0 ; python_version >= "3.8" and python_version < "3.12"
+markupsafe==2.1.2 ; python_version >= "3.8" and python_version < "3.12"
+matplotlib==3.7.1 ; python_version >= "3.8" and python_version < "3.12"
+mdit-py-plugins==0.3.3 ; python_version >= "3.8" and python_version < "3.12"
+mdurl==0.1.2 ; python_version >= "3.8" and python_version < "3.12"
+mpmath==1.3.0 ; python_version >= "3.8" and python_version < "3.12"
+multidict==6.0.4 ; python_version >= "3.8" and python_version < "3.12"
+networkx==3.1 ; python_version >= "3.8" and python_version < "3.12"
+numpy==1.24.3 ; python_version >= "3.8" and python_version < "3.12"
+orjson==3.8.12 ; python_version >= "3.8" and python_version < "3.12"
+packaging==23.1 ; python_version >= "3.8" and python_version < "3.12"
+pandas==2.0.1 ; python_version >= "3.8" and python_version < "3.12"
+pillow==9.5.0 ; python_version >= "3.8" and python_version < "3.12"
+pkgutil-resolve-name==1.3.10 ; python_version >= "3.8" and python_version < "3.9"
+pydantic==1.10.7 ; python_version >= "3.8" and python_version < "3.12"
+pydub==0.25.1 ; python_version >= "3.8" and python_version < "3.12"
+pygments==2.15.1 ; python_version >= "3.8" and python_version < "3.12"
+pyparsing==3.0.9 ; python_version >= "3.8" and python_version < "3.12"
+pyrsistent==0.19.3 ; python_version >= "3.8" and python_version < "3.12"
+python-dateutil==2.8.2 ; python_version >= "3.8" and python_version < "3.12"
+python-multipart==0.0.6 ; python_version >= "3.8" and python_version < "3.12"
+pytz==2023.3 ; python_version >= "3.8" and python_version < "3.12"
+pyyaml==6.0 ; python_version >= "3.8" and python_version < "3.12"
+requests==2.31.0 ; python_version >= "3.8" and python_version < "3.12"
+scipy==1.10.1 ; python_version >= "3.8" and python_version < "3.12"
+semantic-version==2.10.0 ; python_version >= "3.8" and python_version < "3.12"
+six==1.16.0 ; python_version >= "3.8" and python_version < "3.12"
+sniffio==1.3.0 ; python_version >= "3.8" and python_version < "3.12"
+starlette==0.27.0 ; python_version >= "3.8" and python_version < "3.12"
+sympy==1.12 ; python_version >= "3.8" and python_version < "3.12"
+toolz==0.12.0 ; python_version >= "3.8" and python_version < "3.12"
+torch==2.0.1 ; python_version >= "3.8" and python_version < "3.12"
+tqdm==4.65.0 ; python_version >= "3.8" and python_version < "3.12"
+typing-extensions==4.5.0 ; python_version >= "3.8" and python_version < "3.12"
+tzdata==2023.3 ; python_version >= "3.8" and python_version < "3.12"
+uc-micro-py==1.0.2 ; python_version >= "3.8" and python_version < "3.12"
+urllib3==2.0.2 ; python_version >= "3.8" and python_version < "3.12"
+uvicorn==0.22.0 ; python_version >= "3.8" and python_version < "3.12"
+websockets==11.0.3 ; python_version >= "3.8" and python_version < "3.12"
+yarl==1.9.2 ; python_version >= "3.8" and python_version < "3.12"
+zipp==3.15.0 ; python_version >= "3.8" and python_version < "3.10"

stylegan2

@@ -0,0 +1 @@
+Subproject commit 6f160b3d22b8b178ebe533a50d4d5e63aedba21d

troublinggan-rectangle.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1240e05c061ba2edd076f7c9877f69b643890ced81d24390cc92a0018f926f3
+size 369008389

troublinggan-square.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:859760239bd12bb43750e988069332ee706f1f5ab8d664af88dfba20a844e3ec
+size 351315276