3d change ND 2D NEW MODEL

main_noweb.py

@@ -33,8 +33,9 @@ os.system("nvidia-smi")
 print("[INFO]: Imported modules!")
 human = MMPoseInferencer("simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192") # simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192 dekr_hrnet-w32_8xb10-140e_coco-512x512
 hand = MMPoseInferencer("hand")
-human3d = MMPoseInferencer("pose-lift_simplebaseline3d_8xb64-200e_h36m") # pose3d="human3d"
-
+human3d = MMPoseInferencer("https://github.com/open-mmlab/mmpose/blob/main/configs/body_3d_keypoint/pose_lift/h36m/pose-lift_simplebaseline3d_8xb64-200e_h36m.py", 
+                           "https://download.openmmlab.com/mmpose/body3d/simple_baseline/simple3Dbaseline_h36m-f0ad73a4_20210419.pth") # pose3d="human3d"
+#https://github.com/open-mmlab/mmpose/tree/main/configs/hand_2d_keypoint/topdown_regression
 print("[INFO]: Downloaded models!")
 
 def check_fps(video):
@@ -44,7 +45,7 @@ def check_fps(video):
     return nframes, fps
 
 
-def get_frames(video, fps=15, height=512):
+def get_frames(video, fps=15, height=512, width=512):
 # extract the file name and extension
     split_tup = os.path.splitext(video)
 
@@ -60,12 +61,12 @@ def get_frames(video, fps=15, height=512):
     #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)
+        #clip_resized = moviepy.clip.resize(height=height)
+        clip.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)
+        #clip_resized = clip.resize(height=height)
+        #clip_resized.write_videofile(video, fps=clip.fps)
     
     print("video resized to 512 height")
   
@@ -120,6 +121,7 @@ def pose3d(video, kpt_threshold):
 
 def pose2d(video, kpt_threshold):
     video = check_extension(video)
+    get_frames(video)
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())