Update visualization/plot_3d_global.py

visualization/plot_3d_global.py

@@ -94,7 +94,6 @@ def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
         ax.set_yticklabels([])
         ax.set_zticklabels([])
 
-        # Adjusting frame output for GIF compatibility
         io_buf = io.BytesIO()
         plt.savefig(io_buf, format='png', dpi=96)
         io_buf.seek(0)
@@ -104,7 +103,7 @@ def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
         
         plt.close(fig)
         return frame
-
+    
     out = []
     for i in range(frame_number):
         frame = update(i)
@@ -118,8 +117,7 @@ def draw_to_batch(smpl_joints_batch, title_batch=None, outname=None):
     batch_size = len(smpl_joints_batch)
     frames = []
     for i in range(batch_size):
-        frame = plot_3d_motion([smpl_joints_batch[i], None, title_batch[i] if title_batch is not None else None])
-        frames.append(frame)
+        frames.extend(plot_3d_motion([smpl_joints_batch[i], None, title_batch[i] if title_batch is not None else None]))
     
     if frames:
         gif_bytes = io.BytesIO()