new way of creating mp4

WiggleResults/split.py

@@ -4,6 +4,7 @@ import argparse
 import imageio 
 import numpy as np
 import cv2
+import mediapy as media
 
 parser = argparse.ArgumentParser(description='change to useful name')
 parser.add_argument('--dim', default=256, type=int, help='dimention image')
@@ -17,21 +18,14 @@ dim = args.dim
 
 
 def createVideo(array_pil, space_time = 5, name = "out_"):
-
-    size = array_pil[0].size[0]
-
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    video = cv2.VideoWriter(name +".mp4", # Filename
-                        fourcc,
-                        30, # 10 frames per second is chosen as a demo, 30FPS and 60FPS is more typical for a YouTube video
-                        (size,size) # The width and height come from the stats of image1
-                        )
-
-    for image_pil in array_pil:
-        for i in range (0,space_time):
-            video.write(cv2.cvtColor(np.array(image_pil), cv2.COLOR_RGB2BGR))
     
-    video.release()
+    ar_im = []
+    for j in range(3):
+        for image in array_pil:
+            for i in range (0,space_time):
+                ar_im.append(np.array(image))
+
+    media.write_video(name +".mp4" , ar_im, fps=30, qp=18)
     
 
 def gif_order (data, center=True):
@@ -100,7 +94,7 @@ for file in dirs:
                 pointleft+= dim+4
             # Ya tengo todas las imagenes podria hacer el gif aca
             rowImages = gif_order(rowImages,center=False)
-            createVideo(rowImages, 5, name = "./" + ROOT_DIR + '/' + file[:-4] + "_" + str(i))
+            createVideo(rowImages, 3, name = "./" + ROOT_DIR + '/' + file[:-4] + "_" + str(i))
 
             name = "./" + ROOT_DIR + '/' + file[:-4] + "_" + str(i) + '.gif'
             rowImages[0].save(name, save_all=True,format='GIF', append_images=rowImages[1:], optimize=True, duration=100, loop=0)

app.py

@@ -10,6 +10,7 @@ import subprocess
 
 def calculate_depth(model_type, gan_type, dim, slider, img):
 
+    
     if not os.path.exists('temp'):
       os.system('mkdir temp')
 
@@ -50,6 +51,7 @@ def calculate_depth(model_type, gan_type, dim, slider, img):
     formatted = (output * 255.0 / np.max(output)).astype('uint8')
     out_im = Image.fromarray(formatted)
     out_im.save("Images/Input-Test/1_d.png", "PNG")
+    
 
     c_images = '1'
     name_output = 'out'
@@ -61,7 +63,10 @@ def calculate_depth(model_type, gan_type, dim, slider, img):
                     ])
     subprocess.run(["python",  "WiggleResults/split.py", "--dim", dim])
 
-    return [out_im,f'WiggleResults/' + name_output + '_0.gif', f'WiggleResults/' + name_output + '_0.mp4', f'WiggleResults/'+ name_output + '.jpg']
+    path_video = os.path.join(os.path.dirname(__file__), 'WiggleResults' , name_output + '_0.mp4')
+    print(path_video)
+
+    return [img,f'WiggleResults/' + name_output + '_0.gif', path_video, f'WiggleResults/'+ name_output + '.jpg']
 
 
 with gr.Blocks() as demo:
@@ -82,7 +87,7 @@ with gr.Blocks() as demo:
         out = [gr.Image(type="pil", label="depth_estimation")]
     with gr.Row():      
         out.append(gr.Image(type="file", label="Output_wiggle_gif"))
-        out.append(gr.Video(label="Output_wiggle_video", format='mp4'))
+        out.append(gr.Video(label="Output_wiggle_video"))
     out.append(gr.Image(type="file", label="Output_images"))
     btn = gr.Button("Calculate depth + Wiggle")
     btn.click(fn=calculate_depth, inputs=inp, outputs=out)