Update visualization/plot_3d_global.py

visualization/plot_3d_global.py

@@ -98,7 +98,7 @@ def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
         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
+        frame = np.array(img.convert('RGB'), dtype=np.uint8)
         io_buf.close()
         
         plt.close(fig)
@@ -107,7 +107,7 @@ def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4):
     out = []
     for i in range(frame_number):
         frame = update(i)
-        if frame.ndim == 3 and frame.shape[2] == 3:  # Check that frame is H x W x 3
+        if frame.ndim == 3 and frame.shape[2] == 3:
             out.append(frame)
         else:
             print(f"Frame {i} has incorrect shape or channels: {frame.shape}")