edits

app.py

@@ -246,16 +246,14 @@ 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_
-    legend = plt.legend(handles, sf_idx_, loc=3, framealpha=1, frameon=False)
-
-    def export_legend(legend, filename="legend.png"):
-        fig  = legend.figure
-        fig.canvas.draw()
-        # bbox  = legend.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
-        # fig.savefig(filename, dpi="figure", bbox_inches=bbox)
-        return fig
+    fig_leg = plt.figure(3)
+    legend = plt.legend(handles, sf_idx_, loc=3, framealpha=1, frameon=False, facecolor='w')
+    # fig_leg  = legend.figure
+    fig_leg.canvas.draw()
+    # bbox  = legend.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
+    
 
-    return fig1, fig2, export_legend(legend)
+    return fig1, fig2, fig_leg
     # ','.join(map(str, sf_idx_))