backup

app.py

@@ -91,9 +91,9 @@ def NormalizeData(data):
 
 
 # Visualization
-def visualize_similarities(image1, image2):
-    image1 = Image.fromarray(image1)
-    image2 = Image.fromarray(image2)
+def visualize_similarities(q, n):
+    image1 = Image.fromarray(q)
+    image2 = Image.fromarray(n)
 
     a = get_layer4(image1).squeeze()
     b = get_layer4(image2).squeeze()
@@ -121,10 +121,35 @@ def visualize_similarities(image1, image2):
         vmax=1,
     )
 
+    axes[0].set_axis_off()
+    axes[1].set_axis_off()
+
     fig.colorbar(im1, ax=axes[0])
     fig.colorbar(im2, ax=axes[1])
     plt.tight_layout()
-    return fig
+
+    q_image = display_transform(image1)
+    nearest_image = display_transform(image2)
+
+    # make a binarized veruin of the Q
+    fig2, ax = plt.subplots(1, figsize=(5, 5))
+    ax.imshow(display_transform(image1))
+
+    # create a binarized version of  sim1 , for value below 0.5 set to 0 and above 0.5 set to 1
+    sim1_bin = np.where(sim1 > 0.5, 1, 0)
+    print(sim1_bin)
+    # create a binarized version of  sim2 , for value below 0.5 set to 0 and above 0.5 set to 1
+    sim2_bin = np.where(sim2 > 0.5, 1, 0)
+
+    ax.imshow(
+        skimage.transform.resize(sim1_bin, (224, 224)),
+        alpha=1,
+        cmap="binary",
+        vmin=0,
+        vmax=1,
+    )
+
+    return fig, q_image, nearest_image, fig2
 
 
 # GRADIO APP
@@ -132,14 +157,15 @@ main = gr.Interface(
     fn=visualize_similarities,
     inputs=["image", "image"],
     allow_flagging="never",
-    outputs="plot",
+    outputs=["plot", "image", "image", "plot"],
     cache_examples=True,
     enable_queue=False,
     examples=[
         [
             "./examples/Red_Winged_Blackbird_0012_6015.jpg",
             "./examples/Red_Winged_Blackbird_0025_5342.jpg",
-        ]
+        ],
+        ["./examples/Q.jpg", "./examples/1.jpg"],
     ],
 )