temp

app.py

@@ -74,6 +74,7 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
         att_heat = np.array(attns1[0,i,:,:].numpy(), dtype=np.float32)
         att_heat = np.uint8(att_heat / np.max(att_heat[:]) * 255.0)
         att_heat_bin  = np.where(att_heat>threshold, 255, 0)
+        print(att_heat_bin)
         all_att_bin1.append(att_heat_bin)
 
         att_heat = np.array(attns2[0,i,:,:].numpy(), dtype=np.float32)
@@ -84,7 +85,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
     
     fin_img = []
     img1rsz = np.copy(im1_cv)
-    print(img1rsz.size)
+    print(im1.size)
+    print(img1rsz.shape)
     for j, att in enumerate(all_att_bin1):
         att = cv2.resize(att, im1.size, interpolation=cv2.INTER_NEAREST)
         # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_CUBIC)
@@ -99,7 +101,7 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
             img1rsz[m,n, :] = col_[::-1]   
     fin_img.append(img1rsz)
             
-    img2rsz = np.copy(im1_cv)
+    img2rsz = np.copy(im2_cv)
     for j, att in enumerate(all_att_bin2):
         att = cv2.resize(att, im2.size, interpolation=cv2.INTER_NEAREST)
         # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_CUBIC)
@@ -124,10 +126,10 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
     plt.tight_layout()
     # fig.suptitle("Matching SFs", fontsize=16)
 
-    fig.canvas.draw()
-    # Now we can save it to a numpy array.
-    data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
-    data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+    # fig.canvas.draw()
+    # # Now we can save it to a numpy array.
+    # data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+    # data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
     return fig
 
 
@@ -137,6 +139,13 @@ description = "TBD"
 article = "<p style='text-align: center'><a href='https://github.com/naver/fire' target='_blank'>Original Github Repo</a></p>"
 
 
+# css = ".output-image, .input-image {height: 40rem !important; width: 100% !important;}"
+# css = "@media screen and (max-width: 600px) { .output_image, .input_image {height:20rem !important; width: 100% !important;} }"
+# css = ".output_image, .input_image {height: 600px !important}"
+css = ".input_image {height: 600px !important} .output_image, {height: 1200px !important}"
+# css = ".output-image, .input-image {height: 40rem !important; width: 100% !important;}"
+
+
 iface = gr.Interface(
     fn=generate_matching_superfeatures,
     inputs=[
@@ -150,6 +159,7 @@ iface = gr.Interface(
     title=title,
     description=description,
     article=article,
+    css=css,
     examples=[["chateau_1.png", "chateau_2.png", 6, 50]],
 )
 iface.launch()