exemples

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
 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."
+title = "**Playing with kNN for denoising**"
+description = """The encoder takes any image (on the top left) and remove a fraction of the pixels producing a noisy image (on the top right).  
+The decoder takes a noisy image (on the bottom left) and achieves denoising using kNN where you can choose the number of neighbors k.  
+You can directly use the decoder on the encoded image (iy you used the encoder)."""
 
 demo = gr.Blocks()
 
 with demo:
+    gr.Markdown(title)
+    gr.Markdown(description)
     with gr.Row():
         with gr.Column():
             source_img = gr.Image(source="upload", type="filepath", label="init_img")
@@ -14,10 +18,12 @@ with demo:
             noise = gr.Slider(label='noise', minimum = 0.5, maximum = 1, step = .005, value = .95)
             with gr.Row():
                 b1 = gr.Button("Encoder")
+                examples_encoder = gr.Examples([["images/joconde.png"], ["images/braque.png"], ["images/Voronoy.jpg"]], inputs=[source_img])
         with gr.Column():
             encoded_img = gr.Image()
             with gr.Row():
                 bt = gr.Button("Decoder noisy image above")
+                
             
 
             
@@ -28,6 +34,7 @@ with demo:
             k = gr.Slider(label='k', minimum = 1, maximum = 10, step = 1, value = 1)
             with gr.Row():
                 b2 = gr.Button("Decoder")
+                examples_decoder = gr.Examples([["images/afghan_coded.png"], ["images/wave_coded.png"], ["images/spider_coded.png"]], inputs=[noise_img])
                 #bt = gr.Button("Decoder noisy image above")
         with gr.Column():
                 decoded_img = gr.Image()

utils.py

@@ -34,7 +34,7 @@ def encoder_cp(img,color_points):
         img2.putpixel((p[0],p[1]),(t[0],t[1],t[2]))
     return img2
 
-def encoder(img,p):
+def encoder(img,p=0.95):
     img = resize(VALUE,img)
     mask = get_mask(img,p)
     c_p, n_p = generate_points(mask)
@@ -77,12 +77,12 @@ def restore(img, k, color_points, noise_points):
             img.putpixel((p[0],p[1]),(r_m,v_m,b_m))
     return img
 
-def decoder(img,k):
+def decoder(img,k=1):
     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):
+def decoder_noise(img,k=1):
     img = Image.fromarray(img)
     #img = resize(VALUE,img)
     c_p, n_p = get_points(img)