feat(#38): set ssim + psnr as default

mse is still not enough

stagesepx/cutter/cut_range.py

@@ -146,25 +146,18 @@ def get_length(self):
 
     def is_stable(self,
                   threshold: float = None,
-                  mse_threshold: float = None,
                   psnr_threshold: float = None,
                   **_) -> bool:
         # IMPORTANT function!
         # it decided whether a range is stable => everything is based on it!
         if not threshold:
             threshold = 0.95
-        if not mse_threshold:
-            mse_threshold = 0.7
 
         # ssim
         res = np.mean(self.ssim) > threshold
-        # mse
-        res = res and np.mean(self.mse) < mse_threshold
-
-        # default: disable if no threshold
-        # usually, ssim + mse is enough
-        if psnr_threshold:
-            res = res and np.mean(self.psnr) < psnr_threshold
+        # psnr (double check if stable)
+        if res and psnr_threshold:
+            res = np.mean(self.psnr) > psnr_threshold
 
         return res
 

stagesepx/toolbox.py

@@ -168,7 +168,7 @@ def calc_mse(pic1: np.ndarray, pic2: np.ndarray) -> float:
 
 def calc_psnr(pic1: np.ndarray, pic2: np.ndarray) -> float:
     # PSNR: https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio
-    return compare_psnr(pic1, pic2)
+    return compare_psnr(pic1, pic2) / 100
 
 
 def compress_frame(old: np.ndarray,