edits

app.py

@@ -88,6 +88,11 @@ def match(query_feat, pos_feat, LoweRatioTh=0.9):
     
 
 # sf_idx_ = [55, 14, 5, 4, 52, 57, 40, 9]
+def clear_figures():
+    plt.figure().clear()
+    plt.close()
+    plt.cla()
+    plt.clf()
 
 col = plt.get_cmap('tab10')
 
@@ -95,6 +100,8 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
     print('im1:', im1.size)
     print('im2:', im2.size)
 
+    clear_figures()
+
     net = net_sfm
     if Imagenet_model:
         net = net_imagenet
@@ -156,8 +163,6 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
     else:
         sf_idx_ = [i for i in sf_idx_ if i in list(ind_match)]
 
-
-
     # Store all binary SF att maps to show them all at once in the end
     all_att_bin1 = []
     all_att_bin2 = []
@@ -245,7 +250,6 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
     
     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)]
-    labels = sf_idx_
     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