Yaroslav
commited on
fixing buuugs
Browse files
app.py
CHANGED
@@ -1,18 +1,11 @@
|
|
1 |
import cv2
|
2 |
-
import numpy as np
|
3 |
import os
|
4 |
import gradio as gr
|
5 |
|
6 |
# Функция для увеличения FPS
|
7 |
-
def increase_fps(
|
8 |
-
input_path = "input.mp4"
|
9 |
output_path = "output.mp4"
|
10 |
-
|
11 |
-
# Сохранение входного видео
|
12 |
-
with open(input_path, "wb") as f:
|
13 |
-
f.write(input_video.read())
|
14 |
-
|
15 |
-
cap = cv2.VideoCapture(input_path)
|
16 |
input_fps = cap.get(cv2.CAP_PROP_FPS)
|
17 |
frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
18 |
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
@@ -21,9 +14,8 @@ def increase_fps(input_video, target_fps=60):
|
|
21 |
|
22 |
# Проверка: если входное FPS >= 60, просто копируем файл
|
23 |
if input_fps >= target_fps:
|
24 |
-
|
25 |
-
|
26 |
-
return f.read()
|
27 |
|
28 |
# Расчет коэффициента интерполяции
|
29 |
ratio = target_fps / input_fps
|
@@ -55,19 +47,11 @@ def increase_fps(input_video, target_fps=60):
|
|
55 |
cap.release()
|
56 |
output.release()
|
57 |
|
58 |
-
|
59 |
-
with open(output_path, "rb") as f:
|
60 |
-
output_video = f.read()
|
61 |
-
|
62 |
-
# Очистка временных файлов
|
63 |
-
os.remove(input_path)
|
64 |
-
os.remove(output_path)
|
65 |
-
|
66 |
-
return output_video
|
67 |
|
68 |
# Gradio интерфейс
|
69 |
-
def process_video(
|
70 |
-
return increase_fps(
|
71 |
|
72 |
interface = gr.Interface(
|
73 |
fn=process_video,
|
|
|
1 |
import cv2
|
|
|
2 |
import os
|
3 |
import gradio as gr
|
4 |
|
5 |
# Функция для увеличения FPS
|
6 |
+
def increase_fps(input_video_path, target_fps=60):
|
|
|
7 |
output_path = "output.mp4"
|
8 |
+
cap = cv2.VideoCapture(input_video_path)
|
|
|
|
|
|
|
|
|
|
|
9 |
input_fps = cap.get(cv2.CAP_PROP_FPS)
|
10 |
frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
11 |
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
|
|
14 |
|
15 |
# Проверка: если входное FPS >= 60, просто копируем файл
|
16 |
if input_fps >= target_fps:
|
17 |
+
cap.release()
|
18 |
+
return input_video_path # Возвращаем оригинальное видео
|
|
|
19 |
|
20 |
# Расчет коэффициента интерполяции
|
21 |
ratio = target_fps / input_fps
|
|
|
47 |
cap.release()
|
48 |
output.release()
|
49 |
|
50 |
+
return output_path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
|
52 |
# Gradio интерфейс
|
53 |
+
def process_video(video_path):
|
54 |
+
return increase_fps(video_path)
|
55 |
|
56 |
interface = gr.Interface(
|
57 |
fn=process_video,
|