Update visualization/plot_3d_global.py

visualization/plot_3d_global.py

@@ -94,16 +94,13 @@ def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
         ax.set_yticklabels([])
         ax.set_zticklabels([])
 
-        if out_name:
-            plt.savefig(out_name, dpi=96)
-            # plt.close()            
-        else:
-            io_buf = io.BytesIO()
-            plt.savefig(io_buf, format='png', dpi=96)
-            io_buf.seek(0)
-            img = Image.open(io_buf)
-            frame = np.array(img)
-            io_buf.close()
+        # Adjusting frame output for GIF compatibility
+        io_buf = io.BytesIO()
+        plt.savefig(io_buf, format='png', dpi=96)
+        io_buf.seek(0)
+        img = Image.open(io_buf)
+        frame = np.array(img.convert('RGB'), dtype=np.uint8)  # Ensure correct data type and remove alpha channel if present
+        io_buf.close()
         
         plt.close(fig)
         return frame