fix scenechange detection

NevermindNilas · Sep 28, 2024 · 5ce7314 · 5ce7314
1 parent be02fff
commit 5ce7314
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/src/argumentsChecker.py b/src/argumentsChecker.py
@@ -504,7 +504,13 @@ def adjustFeature(
                 f"New dedup sensitivity for {args.dedup_method} is: {args.dedup_sens}"
             )
 
-    sensMap = {"differential": 0.65, "shift_lpips": 0.50, "maxxvit": 0.9}
+    sensMap = {
+        "differential": 0.65,
+        "shift_lpips": 0.50,
+        "maxxvit": 0.9,
+        "maxxvit-tensorrt": 0.9,
+        "maxxvit-directml": 0.9,
+    }
     if args.scenechange_method in sensMap:
         args.scenechange_sens = sensMap[args.scenechange_method] - (
             args.scenechange_sens / 1000

diff --git a/src/scenechange.py b/src/scenechange.py
@@ -89,9 +89,7 @@ def __call__(self, frame):
 
 
 class SceneChangeTensorRT:
-    def __init__(
-        self, half, sceneChangeThreshold=0.85, sceneChangeMethod="scenechange-tensorrt"
-    ):
+    def __init__(self, half, sceneChangeThreshold=0.85, sceneChangeMethod=""):
         self.half = half
         self.sceneChangeThreshold = sceneChangeThreshold
         self.sceneChangeMethod = sceneChangeMethod

diff --git a/src/unifiedInterpolate.py b/src/unifiedInterpolate.py
@@ -612,7 +612,7 @@ def processFrame(self, frame, name=None):
 
     @torch.inference_mode()
     def cacheFrameReset(self, frame):
-        self.I0.copy_(self.processFrame(frame, "I0"), non_blocking=True)
+        self.processFrame(frame, "I0")
         if self.norm is not None:
             self.f0.copy_(self.norm(self.processFrame(frame)), non_blocking=True)
         if self.interpolateSkip is not None: