Update visualization/plot_3d_global.py

visualization/plot_3d_global.py

@@ -114,15 +114,21 @@ def draw_to_batch(smpl_joints_batch, title_batch=None, outname=None):
     batch_size = len(smpl_joints_batch)
     out = []
     for i in range(batch_size):
-        frames = [plot_3d_motion([smpl_joints_batch[i], None, title_batch[i] if title_batch is not None else None])]
+        # Generate frames for the current batch item
+        frames = [plot_3d_motion([smpl_joints_batch[i], None, title_batch[i] if title_batch is not None else None]) for _ in range(frame_number)]
+        # Stack frames to create a sequence
+        frames = np.stack(frames, axis=0)
+        # Create a BytesIO object to save GIF
         gif_bytes = io.BytesIO()
         imageio.mimsave(gif_bytes, frames, format='GIF', fps=120)
         gif_bytes.seek(0)
+        # Get the GIF data
         gif_data = gif_bytes.getvalue()
         out.append(gif_data)
         print(f"Generated GIF for batch {i+1}/{batch_size}, size: {len(gif_data)} bytes")
     return out
 
+
         
 #         if out_name is not None: 
 #             plt.savefig(out_name, dpi=96)