Spaces:
Build error
Build error
temp
Browse files
app.py
CHANGED
|
@@ -115,14 +115,19 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
|
|
| 115 |
fin_img.append(img2rsz)
|
| 116 |
|
| 117 |
|
| 118 |
-
fig = plt.figure(
|
| 119 |
grid = ImageGrid(fig, 111, nrows_ncols=(2, 1), axes_pad=0.1)
|
| 120 |
for ax, img in zip(grid, fin_img):
|
| 121 |
ax.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
|
| 122 |
ax.axis('scaled')
|
| 123 |
ax.axis('off')
|
| 124 |
plt.tight_layout()
|
| 125 |
-
fig.suptitle("Matching SFs", fontsize=16)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 126 |
return fig
|
| 127 |
|
| 128 |
|
|
|
|
| 115 |
fin_img.append(img2rsz)
|
| 116 |
|
| 117 |
|
| 118 |
+
fig = plt.figure()
|
| 119 |
grid = ImageGrid(fig, 111, nrows_ncols=(2, 1), axes_pad=0.1)
|
| 120 |
for ax, img in zip(grid, fin_img):
|
| 121 |
ax.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
|
| 122 |
ax.axis('scaled')
|
| 123 |
ax.axis('off')
|
| 124 |
plt.tight_layout()
|
| 125 |
+
# fig.suptitle("Matching SFs", fontsize=16)
|
| 126 |
+
|
| 127 |
+
fig.canvas.draw()
|
| 128 |
+
# Now we can save it to a numpy array.
|
| 129 |
+
data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
|
| 130 |
+
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
|
| 131 |
return fig
|
| 132 |
|
| 133 |
|