3d change ND 2D NEW MODEL

main_noweb.py

@@ -9,6 +9,7 @@ import gradio as gr
 import moviepy.editor as moviepy
 
 
+
 # System and files
 import os
 import glob
@@ -30,9 +31,9 @@ else:
 os.system("nvidia-smi")
 
 print("[INFO]: Imported modules!")
-human = MMPoseInferencer("dekr_hrnet-w32_8xb10-140e_coco-512x512") # ipr_res50_debias-8xb64-210e_coco-256x256-055a7699_20220913
+human = MMPoseInferencer("ipr_res50_debias-8xb64-210e_coco-256x256-055a7699_20220913") # dekr_hrnet-w32_8xb10-140e_coco-512x512
 hand = MMPoseInferencer("hand")
-human3d = MMPoseInferencer("simplebaseline3d_h36m") # pose3d="human3d"
+human3d = MMPoseInferencer("pose-lift_simplebaseline3d_8xb64-200e_h36m") # pose3d="human3d"
 
 print("[INFO]: Downloaded models!")
 
@@ -42,6 +43,33 @@ def check_fps(video):
     fps = cap.get(cv2.CAP_PROP_FPS)   
     return nframes, fps
 
+
+def get_frames(video, fps=15, height=512):
+# extract the file name and extension
+    split_tup = os.path.splitext(video)
+
+    file_name = split_tup[0]
+    file_extension = split_tup[1]
+
+    if file_extension != ".mp4":
+        print("Converting to mp4")
+        video = file_name+".mp4"
+
+    clip = moviepy.VideoFileClip(video)
+    
+    #check fps
+    if clip.fps > 15:
+        print("vide rate is over 15, resetting to 15")
+        clip_resized = moviepy.clip.resize(height=height)
+        clip_resized.write_videofile(video, fps=fps)
+    else:
+        print("video rate is OK")
+        clip_resized = clip.resize(height=height)
+        clip_resized.write_videofile(video, fps=clip.fps)
+    
+    print("video resized to 512 height")
+  
+
 def check_extension(video):
     split_tup = os.path.splitext(video)