Update visualization/plot_3d_global.py

visualization/plot_3d_global.py

@@ -11,7 +11,10 @@ import imageio
 
 def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
     matplotlib.use('Agg')
-    
+
+    dir_path = "/workspace/animation"
+    if not os.path.exists(dir_path):
+        os.makedirs(dir_path)    
     
     joints, out_name, title = args
     
@@ -117,7 +120,7 @@ def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
         plt.close()
 
         # Save individual frames for verification
-        frame_path = f"/workspace/animation/frame_{index}.png"
+        frame_path = os.path.join(dir_path, f"frame_{index}.png")
         plt.imsave(frame_path, arr)
         if not os.path.exists(frame_path):
             print(f"Error: Frame {index} not saved to {frame_path}")