hi superslo

app.py

@@ -1,6 +1,10 @@
 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):
@@ -12,12 +16,10 @@ def increase_fps(input_video_path, target_fps=60):
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     codec = cv2.VideoWriter_fourcc(*"mp4v")
 
-    # Проверка: если входное FPS >= 60, просто копируем файл
     if input_fps >= target_fps:
         cap.release()
-        return input_video_path  # Возвращаем оригинальное видео
+        return input_video_path
 
-    # Расчет коэффициента интерполяции
     ratio = target_fps / input_fps
     output = cv2.VideoWriter(output_path, codec, target_fps, (width, height))
 
@@ -28,26 +30,17 @@ 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
-            flow_intermediate = flow * alpha
-            intermediate_frame = cv2.remap(prev_frame, flow_intermediate, None, cv2.INTER_LINEAR)
+            intermediate_frame = model.interpolate(prev_frame, next_frame, alpha)
             output.write(intermediate_frame)
 
         output.write(next_frame)
-        prev_frame_gray = next_frame_gray
         prev_frame = next_frame
 
     cap.release()
@@ -55,7 +48,6 @@ def increase_fps(input_video_path, target_fps=60):
 
     return output_path
 
-# Gradio интерфейс
 def process_video(video_path):
     return increase_fps(video_path)
 
@@ -64,13 +56,9 @@ interface = gr.Interface(
     inputs=gr.Video(label="Загрузите видео для обработки"),
     outputs=gr.Video(label="Видео с увеличенным FPS"),
     title="Увеличение FPS до 60",
-    description=(
-        "Этот инструмент увеличивает частоту кадров видео до 60 FPS с использованием "
-        "оптического потока. Загрузите видео с низким FPS, чтобы получить плавный результат."
-    ),
+    description="Этот инструмент увеличивает частоту кадров видео до 60 FPS с использованием оптического потока. Загрузите видео с низким FPS, чтобы получить плавный результат.",
     examples=[]
 )
 
-# Запуск интерфейса
 if __name__ == "__main__":
     interface.launch()