Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,11 @@ def calculate_similarity(img1, img2):
|
|
15 |
def image_similarity(img1, img2):
|
16 |
img1 = img1.astype(np.uint8)
|
17 |
img2 = img2.astype(np.uint8)
|
|
|
|
|
|
|
|
|
|
|
18 |
similarity_score = calculate_similarity(img1, img2)
|
19 |
result = f"Similarity Score: {similarity_score:.4f}"
|
20 |
return result
|
|
|
15 |
def image_similarity(img1, img2):
|
16 |
img1 = img1.astype(np.uint8)
|
17 |
img2 = img2.astype(np.uint8)
|
18 |
+
if img1.shape != img2.shape:
|
19 |
+
max_height = max(img1.shape[0], img2.shape[0])
|
20 |
+
max_width = max(img1.shape[1], img2.shape[1])
|
21 |
+
img1 = cv2.resize(img1, (max_width, max_height))
|
22 |
+
img2 = cv2.resize(img2, (max_width, max_height))
|
23 |
similarity_score = calculate_similarity(img1, img2)
|
24 |
result = f"Similarity Score: {similarity_score:.4f}"
|
25 |
return result
|