Upload similarity.py

similarity.py

@@ -51,26 +51,56 @@ def mobilenet_sim(img1, img2, img1AssetCode, img2AssetCode):
         logging.error("Erro ao calcular similaridade com MobileNet", exc_info=True)
         return 0
 
+# def orb_sim(img1, img2, img1AssetCode, img2AssetCode):
+#     score = 0
+#
+#     try:
+#         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:
+#             score = (len(similar_regions) / len(matches)) * 100
+#             if (score > 0):
+#                 logging.info(f"Orb score from {img1AssetCode} and {img2AssetCode} is {score}")
+#     except Exception as e:
+#         logging.error("Erro ao verificar similaridade ORB", exc_info=True)
+#
+#     return 1 if 0 < score < 1 else score
+
+
 def orb_sim(img1, img2, img1AssetCode, img2AssetCode):
     score = 0
-
     try:
         orb = cv2.ORB_create()
         kp_a, desc_a = orb.detectAndCompute(img1, None)
         kp_b, desc_b = orb.detectAndCompute(img2, None)
-    
+
+        if desc_a is None or desc_b is None:
+            logging.warning(f"ORB descriptors are None for {img1AssetCode} or {img2AssetCode}, skipping.")
+            return 0
+
+        if desc_a.dtype != desc_b.dtype or desc_a.shape[1] != desc_b.shape[1]:
+            logging.warning(f"ORB descriptors incompatible for {img1AssetCode} and {img2AssetCode}, skipping.")
+            return 0
+
         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:
             score = (len(similar_regions) / len(matches)) * 100
-            if (score > 0):
+            if score > 0:
                 logging.info(f"Orb score from {img1AssetCode} and {img2AssetCode} is {score}")
     except Exception as e:
         logging.error("Erro ao verificar similaridade ORB", exc_info=True)
 
     return 1 if 0 < score < 1 else score
 
+
 def ssim_sim(img1, img2):
     s, _ = ssim(img1, img2, full=True)
     return (s + 1) * 50