Add pixelzation

app.py

@@ -586,6 +586,15 @@ print(os.listdir(os.path.join("pretrained")))
 
 # plot_images(images, 5)
 
+def pixelate_rgb(img, window):
+    n, m, _ = img.shape
+    n, m = n - n % window, m - m % window
+    img1 = np.zeros((n, m, 3))
+    for x in range(0, n, window):
+        for y in range(0, m, window):
+            img1[x:x+window,y:y+window] = img[x:x+window,y:y+window].mean(axis=(0,1))
+    return img1
+
 class InferenceWrapper:
     def __init__(self, model):
         self.model = model
@@ -604,7 +613,7 @@ class InferenceWrapper:
             self.noise = self.style_gan.generate_noise(batch_size=batch_size)
         else:
             print(f"Model '{self.model}' already loaded, reusing it.")
-        return self.style_gan({"style_code": self.w, "noise": self.noise, "alpha": 1.0})[0]
+        return pixelate_rgb(self.style_gan({"style_code": self.w, "noise": self.noise, "alpha": 1.0})[0],64)
 
 
 wrapper = InferenceWrapper('celeba')