Update visualization/plot_3d_global.py

visualization/plot_3d_global.py

@@ -93,35 +93,59 @@ def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
         ax.set_xticklabels([])
         ax.set_yticklabels([])
         ax.set_zticklabels([])
-    
-        if out_name is not None: 
-            plt.savefig(out_name, dpi=96)
-            plt.close()
-            
-        else: 
-            io_buf = io.BytesIO()
-            fig.savefig(io_buf, format='png', dpi=96)
-            io_buf.seek(0)
-            img = Image.open(io_buf)
-            img_array = np.array(img)
-            io_buf.close()
-            plt.close()
-            return img_array
-    
+
+        io_buf = io.BytesIO()
+        fig.savefig(io_buf, format='png', dpi=96)
+        io_buf.seek(0)
+        img = Image.open(io_buf)
+        img_array = np.array(img)
+        io_buf.close()
+        plt.close()
+
+        return img_array
+
     out = [update(i) for i in range(frame_number)]
     out = np.stack(out, axis=0)
-    
+
     gif_bytes = io.BytesIO()
     imageio.mimsave(gif_bytes, out, format='GIF', fps=fps)
     gif_bytes.seek(0)
     return gif_bytes.getvalue()
 
-
 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):
-        gif_data = plot_3d_motion([smpl_joints_batch[i], None, title_batch[i] if title_batch is not None else None])
-        out.append(gif_data)
-        print(f"Generated GIF for batch {i+1}/{batch_size}, size: {len(gif_data)} bytes")
-    return out[0]
+    gif_data = plot_3d_motion([smpl_joints_batch[0], None, title_batch[0] if title_batch is not None else None])
+    return gif_data
+    
+
+        
+#         if out_name is not None: 
+#             plt.savefig(out_name, dpi=96)
+#             plt.close()
+            
+#         else: 
+#             io_buf = io.BytesIO()
+#             fig.savefig(io_buf, format='png', dpi=96)
+#             io_buf.seek(0)
+#             img = Image.open(io_buf)
+#             img_array = np.array(img)
+#             io_buf.close()
+#             plt.close()
+#             return img_array
+    
+#     out = [update(i) for i in range(frame_number)]
+#     out = np.stack(out, axis=0)
+    
+#     gif_bytes = io.BytesIO()
+#     imageio.mimsave(gif_bytes, out, format='GIF', fps=fps)
+#     gif_bytes.seek(0)
+#     return gif_bytes.getvalue()
+
+
+# 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):
+#         gif_data = plot_3d_motion([smpl_joints_batch[i], None, title_batch[i] if title_batch is not None else None])
+#         out.append(gif_data)
+#         print(f"Generated GIF for batch {i+1}/{batch_size}, size: {len(gif_data)} bytes")
+#     return out[0]