Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update similarity.py
Browse files- similarity.py +1 -1
similarity.py
CHANGED
@@ -66,7 +66,7 @@ def orb_sim(img1, img2, img1AssetCode, img2AssetCode):
|
|
66 |
except Exception as e:
|
67 |
logging.error("Erro ao verificar similaridade ORB", exc_info=True)
|
68 |
|
69 |
-
return
|
70 |
|
71 |
def ssim_sim(img1, img2):
|
72 |
s, _ = ssim(img1, img2, full=True)
|
|
|
66 |
except Exception as e:
|
67 |
logging.error("Erro ao verificar similaridade ORB", exc_info=True)
|
68 |
|
69 |
+
return 1 if 0 < score < 1 else score
|
70 |
|
71 |
def ssim_sim(img1, img2):
|
72 |
s, _ = ssim(img1, img2, full=True)
|