add dynamic and ensemble in gui

TNTwise · Dec 14, 2024 · 86595d0 · 86595d0
1 parent 8583bce
commit 86595d0
Show file tree

Hide file tree

Showing 3 changed files with 521 additions and 197 deletions.
diff --git a/REAL-Video-Enhancer.py b/REAL-Video-Enhancer.py
@@ -283,7 +283,7 @@ def updateVideoGUIDetails(self):
         self.interpolationContainer.setVisible(
             isInterpolate
         )
-        self.sloMoModeContainer.setVisible(
+        self.interpolateContainer_2.setVisible(
             isInterpolate   
         )
           # set interpolation container visible if interpolate model is not none
@@ -337,6 +337,8 @@ def startRender(self):
             ),
             benchmarkMode=self.benchmarkModeCheckBox.isChecked(),
             sloMoMode=self.sloMoModeCheckBox.isChecked(),
+            dyanmicScaleOpticalFlow=self.dynamicScaledOpticalFlowCheckBox.isChecked(),
+            ensemble=self.ensembleCheckBox.isChecked(),
         )
 
     def disableProcessPage(self):

diff --git a/src/ui/ProcessTab.py b/src/ui/ProcessTab.py
@@ -119,7 +119,7 @@ def QConnect(self):
         # set tile size visible to false by default
         self.parent.tileSizeContainer.setVisible(False)
         #set slo mo container visable to false by default
-        self.parent.sloMoModeContainer.setVisible(False)
+        self.parent.interpolateContainer_2.setVisible(False)
         # connect up tilesize container visiable
         self.parent.tilingCheckBox.stateChanged.connect(self.onTilingSwitch)
 
@@ -199,6 +199,8 @@ def run(
         interpolateModel: str,
         benchmarkMode: bool,
         sloMoMode: bool,
+        dyanmicScaleOpticalFlow: bool,
+        ensemble: bool,
     ):
         interpolateModels, upscaleModels = self.getModels(backend)
         if interpolateModel == "None":
@@ -225,6 +227,8 @@ def run(
         """
         self.benchmarkMode = benchmarkMode
         self.sloMoMode = sloMoMode
+        self.dyanmicScaleOpticalFlow = dyanmicScaleOpticalFlow
+        self.ensemble = ensemble
         # get model attributes
 
         if interpolateModel:
@@ -362,6 +366,14 @@ def renderToPipeThread(
                 command += [
                     "--slomo_mode",
                 ]
+            if self.dyanmicScaleOpticalFlow:
+                command += [
+                    "--dynamic_scaled_optical_flow",
+                ]
+            if self.ensemble:
+                command += [
+                    "--ensemble",
+                ]
 
         if self.settings["preview_enabled"] == "True":
             command += [