MarioPrzBasto commited on
Commit
6ce3ea3
·
verified ·
1 Parent(s): 4381896

Update similarity.py

Browse files
Files changed (1) hide show
  1. similarity.py +2 -2
similarity.py CHANGED
@@ -161,14 +161,14 @@ def load_image_url(source, assetCode, contentType=None, ffmpeg_path='ffmpeg', fr
161
  def check_similarity(images: List[RequestModel]):
162
  logging.info(f"Checking similarity for main source with resource id {images[0].originId}")
163
 
164
- original_image = load_image_url(images[0].source)
165
  original_image_shape = original_image.shape
166
 
167
  results = []
168
 
169
  for i in range(1, len(images)):
170
  try:
171
- image = load_image_url(images[i].source)
172
  image = cv2.resize(image, original_image_shape[::-1])
173
 
174
  similarity_score = ssim_sim(original_image, image)
 
161
  def check_similarity(images: List[RequestModel]):
162
  logging.info(f"Checking similarity for main source with resource id {images[0].originId}")
163
 
164
+ original_image = load_image_url(images[0].source, images[0].assetCode)
165
  original_image_shape = original_image.shape
166
 
167
  results = []
168
 
169
  for i in range(1, len(images)):
170
  try:
171
+ image = load_image_url(images[i].source, images[i].source)
172
  image = cv2.resize(image, original_image_shape[::-1])
173
 
174
  similarity_score = ssim_sim(original_image, image)