fix naming convention

NevermindNilas · Dec 26, 2024 · 249969f · 249969f
1 parent db44dcb
commit 249969f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/dedup/dedup.py b/src/dedup/dedup.py
@@ -189,7 +189,7 @@ def processFrame(self, frame):
         )
 
 
-class FlownetSDedup:
+class DedupFlownetS:
     def __init__(
         self,
         half: bool = True,
@@ -293,8 +293,7 @@ def prepareFrame(self, frame):
     def __call__(self, frame):
         if self.prevFrame is None:
             with torch.cuda.stream(self.normStream):
-                self.prevFrame = frame
-                self.prevFrame = self.prepareFrame(self.prevFrame)
+                self.prevFrame = self.prepareFrame(frame)
             self.normStream.synchronize()
             return False
 

diff --git a/src/utils/initializeModels.py b/src/utils/initializeModels.py
@@ -475,9 +475,9 @@ def initializeModels(self):
                 )
 
             case "flownets":
-                from src.dedup.dedup import FlownetSDedup
+                from src.dedup.dedup import DedupFlownetS
 
-                dedup_process = FlownetSDedup(
+                dedup_process = DedupFlownetS(
                     half=self.half,
                     dedupSens=self.dedup_sens,
                     height=self.height,