encoder-decoder

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from utils import encoder, decoder
+from utils import encoder, decoder, decoder_noise
 
 title = "Playing with kNN for denoising"
 description = "The encoder takes any image and remove a fraction of the pixels producing a noisy image. The decoder takes a noisy image and achieves denoising using kNN where you can choose k."
@@ -16,21 +16,26 @@ with demo:
                 b1 = gr.Button("Encoder")
         with gr.Column():
             encoded_img = gr.Image()
+            with gr.Row():
+                bt = gr.Button("Decoder noisy image above")
             
 
             
     with gr.Row():
         with gr.Column():
+            
             noise_img = gr.Image(source="upload", type="filepath", label="init_img")
             k = gr.Slider(label='k', minimum = 1, maximum = 10, step = 1, value = 1)
             with gr.Row():
                 b2 = gr.Button("Decoder")
+                #bt = gr.Button("Decoder noisy image above")
         with gr.Column():
                 decoded_img = gr.Image()
             
 
     b1.click(encoder, inputs=[source_img, noise], outputs=encoded_img)
     b2.click(decoder, inputs=[noise_img,k], outputs=decoded_img)
+    bt.click(decoder_noise, inputs=[encoded_img,k], outputs=decoded_img)
     
     
 demo.launch()

utils.py

@@ -6,7 +6,8 @@ VALUE = 512
 
 def resize(value,img):
   img = Image.open(img)
-  img = img.resize((value,value), Image.Resampling.LANCZOS)
+  #img = img.resize((value,value), Image.Resampling.LANCZOS)
+  img.thumbnail((VALUE,VALUE), Image.Resampling.LANCZOS)
   return img
 
 def get_mask(img,p):
@@ -34,8 +35,7 @@ def encoder_cp(img,color_points):
     return img2
 
 def encoder(img,p):
-    #img = resize(VALUE,img)
-    img = Image.open(img)
+    img = resize(VALUE,img)
     mask = get_mask(img,p)
     c_p, n_p = generate_points(mask)
     return encoder_cp(img, c_p)
@@ -78,6 +78,12 @@ def restore(img, k, color_points, noise_points):
     return img
 
 def decoder(img,k):
-    img = Image.open(img)
+    img = resize(VALUE,img)
+    c_p, n_p = get_points(img)
+    return restore(img,int(k),c_p,n_p)
+
+def decoder_noise(img,k):
+    img = Image.fromarray(img)
+    #img = resize(VALUE,img)
     c_p, n_p = get_points(img)
     return restore(img,int(k),c_p,n_p)