by cro

app.py

@@ -1,10 +1,6 @@
 import cv2
 import os
 import gradio as gr
-from super_slo import SuperSlo
-
-# Инициализация модели SuperSlo для интерполяции
-model = SuperSlo()
 
 # Функция для увеличения FPS
 def increase_fps(input_video_path, target_fps=60):
@@ -30,17 +26,25 @@ def increase_fps(input_video_path, target_fps=60):
         output.release()
         raise ValueError("Не удалось прочитать кадры из видео.")
 
+    prev_frame_gray = cv2.cvtColor(prev_frame, cv2.COLOR_BGR2GRAY)
+
     while True:
         ret, next_frame = cap.read()
         if not ret:
             break
 
+        next_frame_gray = cv2.cvtColor(next_frame, cv2.COLOR_BGR2GRAY)
+
+        flow = cv2.calcOpticalFlowFarneback(prev_frame_gray, next_frame_gray, None, 0.5, 3, 15, 3, 5, 1.2, 0)
+
         for i in range(1, int(ratio)):
             alpha = i / ratio
-            intermediate_frame = model.interpolate(prev_frame, next_frame, alpha)
+            flow_intermediate = flow * alpha
+            intermediate_frame = cv2.remap(prev_frame, flow_intermediate, None, cv2.INTER_LINEAR)
             output.write(intermediate_frame)
 
         output.write(next_frame)
+        prev_frame_gray = next_frame_gray
         prev_frame = next_frame
 
     cap.release()