MarioPrzBasto commited on
Commit
c7afcdf
·
verified ·
1 Parent(s): 274bce3

Update similarity.py

Browse files
Files changed (1) hide show
  1. 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