Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update similarity.py
Browse files- similarity.py +2 -2
similarity.py
CHANGED
@@ -102,8 +102,8 @@ def check_similarity(images: List[RequestModel]):
|
|
102 |
image = cv2.resize(image, original_image_shape[::-1])
|
103 |
|
104 |
similarity_score = ssim_sim(original_image, image)
|
105 |
-
similarity_orb_score = orb_sim(original_image, image)
|
106 |
-
similarity_mobilenet_score = mobilenet_sim(original_image, image)
|
107 |
except Exception as e:
|
108 |
logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
|
109 |
similarity_score = 0
|
|
|
102 |
image = cv2.resize(image, original_image_shape[::-1])
|
103 |
|
104 |
similarity_score = ssim_sim(original_image, image)
|
105 |
+
similarity_orb_score = orb_sim(original_image, image, images[0].assetCode, images[i].assetCode)
|
106 |
+
similarity_mobilenet_score = mobilenet_sim(original_image, image, images[0].assetCode, images[i].assetCode)
|
107 |
except Exception as e:
|
108 |
logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
|
109 |
similarity_score = 0
|