Spaces:

spuun
/

ilynaga

Sleeping

spuun commited on Jan 15

Commit

90b7b6f

verified ·

1 Parent(s): dbc055b

fix: ssim

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ def compare_face_ssim(img1, img2, face_area):
         face2 = face2.astype(np.float32) / 255.0
     # Calculate multichannel SSIM
-    ssim_value = ssim(face1, face2, data_range=1.0, multichannel=True)
     return ssim_value
 def predict_and_compare(image):

         face2 = face2.astype(np.float32) / 255.0
     # Calculate multichannel SSIM
+    ssim_value = ssim(face1, face2, data_range=1.0, channel_axis=2, win_size=3)
     return ssim_value
 def predict_and_compare(image):