1580158
/
stable-diffusion-webui-pixelization
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui-pixelization.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' into patch-1

pull/8/head
AUTOMATIC1111 2023-12-30 16:02:01 +03:00 committed by GitHub
parent b714d566ae 585ead922c
commit 280c78e1cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -16,8 +16,6 @@ Download all three models from the table and place them into the `checkpoints` d
| https://drive.google.com/file/d/17f2rKnZOpnO9ATwRXgqLz5u5AZsyDvq_/view?usp=sharing | alias_net.pth |
| https://drive.google.com/file/d/1i_8xL3stbLWNF4kdQJ50ZhnRFhSDh3Az/view?usp=sharing | 160_net_G_A.pth |
License of Pixelization seems to prevent me from reuploading models anywhere and google drive makes it impossible to download them automatically.
# Credits
* Original repo: https://github.com/WuZongWei6/Pixelization

3
install.py
View File

@ -10,3 +10,6 @@ try:
os.remove(os.path.join(path, "pixelization", "models", "__init__.py"))
except OSError as e:
pass
if not launch.is_installed("gdown"):
launch.run_pip("install gdown", "requirements for stable-diffusion-webui-pixelization")

28
scripts/postprocessing_pixelization.py
View File

@ -1,6 +1,6 @@
import os
from modules import scripts_postprocessing, devices, scripts
from modules import scripts_postprocessing, devices, scripts, ui_components
import gradio as gr
from modules.ui_components import FormRow
@ -12,6 +12,7 @@ import numpy as np
from pixelization.models.networks import define_G
import pixelization.models.c2pGen
import gdown
pixelize_code = [
233356.8125, -27387.5918, -32866.8008, 126575.0312, -181590.0156,
@ -104,21 +105,14 @@ class Model(torch.nn.Module):
def load(self):
os.makedirs(path_checkpoints, exist_ok=True)
missing = False
if not os.path.exists(path_pixelart_vgg19):
print(f"Missing {path_pixelart_vgg19} - download it from https://drive.google.com/uc?id=1VRYKQOsNlE1w1LXje3yTRU5THN2MGdMM")
missing = True
gdown.download("https://drive.google.com/uc?id=1VRYKQOsNlE1w1LXje3yTRU5THN2MGdMM", path_pixelart_vgg19)
if not os.path.exists(path_160_net_G_A):
print(f"Missing {path_160_net_G_A} - download it from https://drive.google.com/uc?id=1i_8xL3stbLWNF4kdQJ50ZhnRFhSDh3Az")
missing = True
gdown.download("https://drive.google.com/uc?id=1i_8xL3stbLWNF4kdQJ50ZhnRFhSDh3Az", path_160_net_G_A)
if not os.path.exists(path_alias_net):
print(f"Missing {path_alias_net} - download it from https://drive.google.com/uc?id=17f2rKnZOpnO9ATwRXgqLz5u5AZsyDvq_")
missing = True
assert not missing, 'Missing checkpoints for pixelization - see console for download links.'
gdown.download("https://drive.google.com/uc?id=17f2rKnZOpnO9ATwRXgqLz5u5AZsyDvq_", path_alias_net)
with torch.no_grad():
self.G_A_net = define_G(3, 3, 64, "c2pGen", "instance", False, "normal", 0.02, [0])
@ -171,14 +165,10 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
model = None
def ui(self):
with FormRow():
with gr.Column():
with FormRow():
enable = gr.Checkbox(False, label="Enable pixelization")
upscale_after = gr.Checkbox(False, label="Keep resolution")
with gr.Column():
pixel_size = gr.Slider(minimum=1, maximum=16, step=1, label="Pixel size", value=1, elem_id="pixelization_pixel_size")
with ui_components.InputAccordion(False, label="Pixelize") as enable:
with gr.Row():
upscale_after = gr.Checkbox(False, label="Keep resolution")
pixel_size = gr.Slider(minimum=1, maximum=16, step=1, label="Pixel size", value=4, elem_id="pixelization_pixel_size")
return {
"enable": enable,