YannisK commited on
Commit
9180d7e
1 Parent(s): 8ff5253
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -247,9 +247,13 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
247
  handles = [f("s", col.colors[i]) for i in range(n_sf_ids)]
248
  labels = sf_idx_
249
  fig_leg = plt.figure(3)
250
- legend = plt.legend(handles, sf_idx_, loc=3, framealpha=1, frameon=False, facecolor='w')
251
  # fig_leg = legend.figure
252
  fig_leg.canvas.draw()
 
 
 
 
253
  # bbox = legend.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
254
 
255
 
 
247
  handles = [f("s", col.colors[i]) for i in range(n_sf_ids)]
248
  labels = sf_idx_
249
  fig_leg = plt.figure(3)
250
+ legend = plt.legend(handles, sf_idx_, loc=3, framealpha=1, frameon=False, facecolor='w',fontsize=15, loc="center")
251
  # fig_leg = legend.figure
252
  fig_leg.canvas.draw()
253
+ ax3 = plt.gca()
254
+ # ax2.axis('scaled')
255
+ ax3.axis('off')
256
+ plt.tight_layout()
257
  # bbox = legend.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
258
 
259