a

.ipynb_checkpoints/app-checkpoint.py

@@ -4,13 +4,24 @@ import cv2
 from encoded_video import EncodedVideo, write_video
 import torch
 import numpy as np
+import gc
 
-def video_identity(filepath):
-    vid = EncodedVideo.from_path(filepath)
-    clip = vid.get_clip(start_sec, start_sec + duration)
-    video_arr = torch.from_numpy(clip['video']).permute(3, 0, 1, 2)
-    write_video('out.mp4', video_arr, fps=60)
-    capture = cv2.VideoCapture('out.mp4')
+def video_identity(video):
+    # vid = EncodedVideo.from_path(filepath)
+    # clip = vid.get_clip(start_sec, start_sec + duration)
+    # video_arr = torch.from_numpy(clip['video']).permute(3, 0, 1, 2)
+    # write_video('out.mp4', video_arr, fps=60)
+    
+    gc.collect()
+    output_video_path = "./output.mp4"
+    cap = cv2.VideoCapture(video)
+    fourcc = cv2.VideoWriter_fourcc(*"I420")  # encoder
+
+    out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (int(cap.get(3)), int(cap.get(4))))
+    
+    
+    
+    capture = cv2.VideoCapture('output.mp4')
     while (True):
     
         success, frame = capture.read()
@@ -25,7 +36,7 @@ def video_identity(filepath):
     img=cv2.imread('train/frame_0')
     return img
 demo = gr.Interface(video_identity, 
-                    gr.inputs.Video(source='upload'), 
-                    gr.inputs.Image(), 
+                    gr.Video(source='upload'), 
+                    gr.Image(), 
                     cache_examples=True)
 demo.launch(debug=True)

.ipynb_checkpoints/requirements-checkpoint.txt

@@ -1,4 +1,5 @@
 opencv-python
 encoded-video
 torch
-numpy
+numpy
+gc-python-utils

app.py

@@ -4,13 +4,24 @@ import cv2
 from encoded_video import EncodedVideo, write_video
 import torch
 import numpy as np
+import gc
 
-def video_identity(filepath):
-    vid = EncodedVideo.from_path(filepath)
-    clip = vid.get_clip(start_sec, start_sec + duration)
-    video_arr = torch.from_numpy(clip['video']).permute(3, 0, 1, 2)
-    write_video('out.mp4', video_arr, fps=60)
-    capture = cv2.VideoCapture('out.mp4')
+def video_identity(video):
+    # vid = EncodedVideo.from_path(filepath)
+    # clip = vid.get_clip(start_sec, start_sec + duration)
+    # video_arr = torch.from_numpy(clip['video']).permute(3, 0, 1, 2)
+    # write_video('out.mp4', video_arr, fps=60)
+    
+    gc.collect()
+    output_video_path = "./output.mp4"
+    cap = cv2.VideoCapture(video)
+    fourcc = cv2.VideoWriter_fourcc(*"I420")  # encoder
+
+    out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (int(cap.get(3)), int(cap.get(4))))
+    
+    
+    
+    capture = cv2.VideoCapture('output.mp4')
     while (True):
     
         success, frame = capture.read()
@@ -25,7 +36,7 @@ def video_identity(filepath):
     img=cv2.imread('train/frame_0')
     return img
 demo = gr.Interface(video_identity, 
-                    gr.inputs.Video(source='upload'), 
-                    gr.inputs.Image(), 
+                    gr.Video(source='upload'), 
+                    gr.Image(), 
                     cache_examples=True)
 demo.launch(debug=True)

requirements.txt

@@ -1,4 +1,5 @@
 opencv-python
 encoded-video
 torch
-numpy
+numpy
+gc-python-utils