Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,49 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
|
|
2 |
|
3 |
-
|
4 |
-
|
5 |
|
6 |
-
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
+
import cv2
|
3 |
+
import torch
|
4 |
+
from transformers import AutoModelForImageTranslation # Bu, örnek bir model. Gerçek model ismini buraya yazmanız gerekebilir.
|
5 |
|
6 |
+
# Göz teması düzeltme modeli - model adını uygun bir modelle değiştirin
|
7 |
+
model = AutoModelForImageTranslation.from_pretrained("username/gaze_correction_model") # model yolunu doğru şekilde ayarlayın
|
8 |
|
9 |
+
# Video işlemi fonksiyonu
|
10 |
+
def correct_gaze_in_video(video_path):
|
11 |
+
# Video dosyasını aç
|
12 |
+
cap = cv2.VideoCapture(video_path)
|
13 |
+
fps = cap.get(cv2.CAP_PROP_FPS)
|
14 |
+
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
15 |
+
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
16 |
+
|
17 |
+
# Çıkış videosu için ayarlar
|
18 |
+
output_path = "corrected_gaze_output.mp4"
|
19 |
+
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
|
20 |
+
out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
|
21 |
+
|
22 |
+
while cap.isOpened():
|
23 |
+
ret, frame = cap.read()
|
24 |
+
if not ret:
|
25 |
+
break
|
26 |
+
|
27 |
+
# Göz temasını düzeltmek için her kareyi işleme (örnek bir işleme işlemi)
|
28 |
+
inputs = preprocess_image(frame) # Görüntüyü modele uygun hale getir (normalize, resize vb.)
|
29 |
+
corrected_frame = model(inputs)
|
30 |
+
|
31 |
+
# Çıkışı video dosyasına ekleyin
|
32 |
+
out.write(corrected_frame)
|
33 |
+
|
34 |
+
# Kaynakları serbest bırak
|
35 |
+
cap.release()
|
36 |
+
out.release()
|
37 |
+
|
38 |
+
return output_path
|
39 |
+
|
40 |
+
# Gradio arayüzü
|
41 |
+
iface = gr.Interface(
|
42 |
+
fn=correct_gaze_in_video, # İşlem fonksiyonumuz
|
43 |
+
inputs="file", # Kullanıcıdan video dosyası girişi
|
44 |
+
outputs="file", # İşlenmiş video dosyası çıktısı
|
45 |
+
title="Gaze Correction in Video",
|
46 |
+
description="Upload a video to correct gaze direction."
|
47 |
+
)
|
48 |
+
|
49 |
+
iface.launch()
|