make directory

main.py

@@ -10,6 +10,8 @@ import torch
 
 # Gradio
 import gradio as gr
+import moviepy.editor as moviepy
+
 
 # System and files
 import os
@@ -106,6 +108,12 @@ def pose3d(video):
 
 
 def pose2d(video):
+    print("Transforming input")
+    clip = moviepy.VideoFileClip(video)
+
+    new_video = video[:-5]+".mp4"
+    clip.write_videofile(new_video)
+
     vidname = video.split("/")[-1]
     print("VIDNAME", vidname)
     add_dir = str(uuid.uuid4())
@@ -115,23 +123,22 @@ def pose2d(video):
     print("[INFO]: LIST CURRENT OUT DIR: ", os.listdir("/".join(video.split("/")[:-1])))
 
 
-    result_generator = human(video, 
-                                 vis_out_dir = add_dir,
-                                 return_vis=True,
-                                 thickness=2,
-                                 rebase_keypoint_height=True,
-                                 #kpt_thr=kpt_thr,
-                                 device="cuda"
-                                 )    
+    result_generator = human(new_video, 
+                            vis_out_dir = vis_out_dir,
+                            return_vis=True,
+                            thickness=2,
+                            rebase_keypoint_height=True,
+                            #kpt_thr=kpt_thr,
+                            device="cuda"
+                            )    
     
     result = [result for result in result_generator] #next(result_generator)        
-    print(result)
-
+    #print(result)
 
-    out_file = glob.glob(os.path.join(add_dir, "*.mp4")) + glob.glob(os.path.join(add_dir, "*.webm")) 
+    out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm")) 
     #out_file = os.path.join(vis_out_dir, "sample_flip.webm")
     print("OTFILE: ", out_file)
-    print("ADD DIR ", os.listdir(add_dir))
+    print("ADD DIR ", os.listdir(vis_out_dir))
 
     return out_file
 
@@ -171,7 +178,7 @@ if __name__ == "__main__":
                     with gr.Column():
                         video_input = gr.Video(source="upload", type="filepath", height=512)
                         # Insert slider with kpt_thr
-                        file_kpthr = gr.Slider(0, 1, value=1, label="Keypoint threshold",step=5, info='1')
+                        file_kpthr = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Keypoint threshold')
 
                         submit_pose_file = gr.Button("Make 2d pose estimation")
                         submit_pose3d_file = gr.Button("Make 3d pose estimation")
@@ -188,8 +195,7 @@ if __name__ == "__main__":
                     with gr.Column():
                         webcam_input = gr.Video(source="webcam", height=512)
                         
-                        web_kpthr = gr.Slider(0, 1, value=1, label="Keypoint threshold",step=5, info='1')
-
+                        web_kpthr = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Keypoint threshold')
 
                         submit_pose_web = gr.Button("Make 2d pose estimation")
                         submit_pose3d_web = gr.Button("Make 3d pose estimation")

requirements.txt

@@ -3,4 +3,5 @@ numpy
 opencv-python
 ultralytics
 lap
-torchvision
+torchvision
+MoviePy