fix bug: mkdir output_*

videopose_PSTMO.py

@@ -153,7 +153,7 @@ def main(args):
     output_dir_dict['npy'] = npy_filename
     np.save(npy_filename, prediction, allow_pickle=True)
 
-    anim_output = {'Ours': prediction}
+    anim_output = {'Skeleton': prediction}
     input_keypoints = image_coordinates(input_keypoints[..., :2], w=1000, h=1002)
 
     ckpt, time3 = ckpt_time(time2)
@@ -201,6 +201,16 @@ def inference_video(video_path, detector_2d):
 
 
 def gr_video2mc(video_path):
+
+    if not os.path.exists('output_3Dpose_npy'):
+        os.makedirs('output_3Dpose_npy')
+    if not os.path.exists('output_alphapose'):
+        os.makedirs('output_alphapose')
+    if not os.path.exists('output_miframes'):
+        os.makedirs('output_miframes')
+    if not os.path.exists('output_videos'):
+        os.makedirs('output_videos')
+
     FPS_mine_imator = 30
     output_dir_dict = inference_video(video_path, 'alpha_pose')
     Hk.hpe2keyframes(output_dir_dict['npy'], FPS_mine_imator, f"output_miframes/{output_dir_dict['video_name']}.miframes")