added return in check extension

main_noweb.py

@@ -29,8 +29,8 @@ else:
 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")
+human = MMPoseInferencer("human", device=device) # "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", device=device)
 #model3d = gr.State()
 # human3d = MMPoseInferencer(device=device, 
 #                                pose3d="human3d", 
@@ -75,6 +75,7 @@ def get_frames(video, fps=50, height=512, width=512):
         print("video rate is OK")
         #clip_resized = clip.resize(height=height)
         #clip.write_videofile(video, fps=clip.fps)
+    return video
     
 
 def check_extension(video):
@@ -104,7 +105,6 @@ def pose3d(video, kpt_threshold):
     print("HUMAN 3d downloaded!!")    
     # Define new unique folder
     add_dir = str(uuid.uuid4())
-    vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
 
     os.makedirs(add_dir)
     print(check_fps(video))
@@ -139,12 +139,10 @@ def pose2d(video, kpt_threshold):
     print(check_fps(video))
     result_generator = human(video, 
                             vis_out_dir = add_dir,
-                            #return_vis=True,
                             radius = 5,
                             thickness=4,
                             rebase_keypoint_height=True,
                             kpt_thr=kpt_threshold,
-                            device=device,
                             pred_out_dir = add_dir
                             )    
     
@@ -171,9 +169,7 @@ def pose2dhand(video, kpt_threshold):
                                  radius = 5,
                                  rebase_keypoint_height=True,
                                  kpt_thr=kpt_threshold,
-                                 pred_out_dir = add_dir, 
-
-                                 device=device)    
+                                 pred_out_dir = add_dir)    
     
     result = [result for result in result_generator] #next(result_generator)