warn and return false on transition mismatch

TNTwise · Nov 15, 2024 · f4ab018 · f4ab018
1 parent 5e0b1db
commit f4ab018
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/backend/src/pytorch/InterpolateArchs/__pycache__/DetectInterpolateArch.cpython-311.pyc b/backend/src/pytorch/InterpolateArchs/__pycache__/DetectInterpolateArch.cpython-311.pyc
diff --git a/backend/src/utils/SceneDetect.py b/backend/src/utils/SceneDetect.py
@@ -1,6 +1,7 @@
 import numpy as np
 import cv2
 from collections import deque
+import sys
 from .Util import bytesToImg
 from .PySceneDetectUtils import ContentDetector
 
@@ -196,7 +197,8 @@ def sceneDetect(self, frame: np.ndarray):
         self.frameNum += 1
         if len(frameList) > 0:
             if self.frameNum != frameList[0] + 1:
-                print("-------" + f"{self.frameNum}, {frameList[0]}")
+                print(f"Transition Mismatch {self.frameNum} is not equal to {frameList[0] + 1}, skipping", file=sys.stderr)
+                return False
 
         return len(frameList) > 0