edits

app.py

@@ -204,8 +204,9 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
         # print('att:', att.shape)
         mask2d = zip(*np.where(att==255))
         for m,n in mask2d:
-            col_ = col.colors[j] if j < 7 else col.colors[j+1]
-            if j == 0: col_ = col.colors[9]
+            col_ = col.colors[j]
+            # col_ = col.colors[j] if j < 7 else col.colors[j+1]
+            # 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)
@@ -243,7 +244,7 @@ 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[i]) for i in range(n_sf_ids)]
+    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)