Update main.py

main.py

@@ -27,6 +27,21 @@ BASE_DIR = "/tmp/data"
 app = FastAPI()
 mobilenet = MobileNetV2(weights="imagenet", include_top=False, pooling='avg')
 
+def orb_sim(img1, img2):
+    score = 0
+
+    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
+
+    return 1 if 0 < score < 1 else score
+    
 def orb_sim(img1, img2):
     # ORB
     orb = cv2.ORB_create()