Update app.py
Browse files
app.py
CHANGED
@@ -14,11 +14,14 @@ def calculate_similarity(img1, img2):
|
|
14 |
# Function to compute similarity scores for all images
|
15 |
def compute_similarity(target_image, image_list):
|
16 |
scores = []
|
|
|
17 |
for image in image_list:
|
18 |
-
|
|
|
19 |
scores.append(similarity_score)
|
20 |
return scores
|
21 |
|
|
|
22 |
# Function to handle the Gradio interface
|
23 |
def image_similarity(target_image, image_list):
|
24 |
target_image = target_image.astype(np.uint8)
|
|
|
14 |
# Function to compute similarity scores for all images
|
15 |
def compute_similarity(target_image, image_list):
|
16 |
scores = []
|
17 |
+
target_image_resized = cv2.resize(target_image, (target_image.shape[1], target_image.shape[0]))
|
18 |
for image in image_list:
|
19 |
+
image_resized = cv2.resize(image, (target_image.shape[1], target_image.shape[0]))
|
20 |
+
similarity_score = calculate_similarity(target_image_resized, image_resized)
|
21 |
scores.append(similarity_score)
|
22 |
return scores
|
23 |
|
24 |
+
|
25 |
# Function to handle the Gradio interface
|
26 |
def image_similarity(target_image, image_list):
|
27 |
target_image = target_image.astype(np.uint8)
|