edits

app.py

@@ -196,7 +196,6 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
             if j == 0: col_ = col.colors[9]
             col_ = 255*np.array(colors.to_rgba(col_))[:3]
             img1rsz[m,n, :] = col_[::-1]   
-    fin_img.append(img1rsz)
             
     img2rsz = np.copy(im2_cv)
     print('im2:', im2.size)
@@ -214,7 +213,6 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
             # if j == 0: col_ = col.colors[9]
             col_ = 255*np.array(colors.to_rgba(col_))[:3]
             img2rsz[m,n, :] = col_[::-1]   
-    fin_img.append(img2rsz)
 
     fig1 = plt.figure(1)
     plt.imshow(cv2.cvtColor(img1rsz, cv2.COLOR_BGR2RGB))
@@ -231,29 +229,13 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
     ax2.axis('off')
     plt.tight_layout()    
     # fig2.canvas.draw()
-       
-    # fig = plt.figure()
-    # grid = ImageGrid(fig, 111, nrows_ncols=(2, 1),  axes_pad=0.1)
-    # for ax, img in zip(grid, fin_img):
-    #     ax.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
-    #     ax.axis('scaled')
-    #     ax.axis('off')
-    # 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,))
-    
-
     
     f = lambda m,c: plt.plot([],[],marker=m, color=c, ls="none")[0]
     handles = [f("s", col.colors[i]) for i in range(n_sf_ids)]
     fig_leg = plt.figure(3)
     legend = plt.legend(handles, sf_idx_, framealpha=1, frameon=False, facecolor='w',fontsize=15, loc="center")
     # fig_leg  = legend.figure
-    fig_leg.canvas.draw()
+    # fig_leg.canvas.draw()
     ax3 = plt.gca()
     # ax2.axis('scaled')
     ax3.axis('off')