Update similarity.py (#4)

- Update similarity.py (5c8906bf3f45e631db581d0607eb1c28129f87b8)

similarity.py

@@ -11,6 +11,18 @@ import logging
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
+def orb_sim(img1, img2):
+    orb = cv2.ORB_create()
+    kp_a, desc_a = orb.detectAndCompute(img1, None)
+    kp_b, desc_b = orb.detectAndCompute(img2, None)
+
+    bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
+    matches = bf.match(desc_a, desc_b)
+    similar_regions = [i for i in matches if i.distance < 20]
+    if len(matches) == 0:
+        return 0
+    return len(similar_regions) / len(matches)
+
 def load_image_url(source):
     Image.MAX_IMAGE_PIXELS = None
 
@@ -40,9 +52,11 @@ def check_similarity(images: List[RequestModel]):
             image = cv2.resize(image, original_image_shape[::-1])
             s, _ = ssim(original_image, image, full=True)
             similarity_score = (s + 1) * 50
+            similarity_orb_score = orb_sim(original_image, image) * 100
         except Exception as e:
             logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
             similarity_score = 0
+            similarity_orb_score = 0
 
         response = ResponseModel(originId=images[i].originId, source=images[i].source, sequence=images[i].sequence,
                                  assetCode=images[i].assetCode, similarity=similarity_score)