From c83fa30daa049e46f92f71913834c65740d9cf3d Mon Sep 17 00:00:00 2001 From: Erwin Coumans Date: Wed, 12 Sep 2018 08:19:30 -0700 Subject: [PATCH] use numpy to reshape image data to matplotlib compatible format fixes issue #1871 --- examples/pybullet/examples/testrender.py | 9 ++++++++- examples/pybullet/examples/testrender_np.py | 8 ++++---- 2 files changed, 12 insertions(+), 5 deletions(-) diff --git a/examples/pybullet/examples/testrender.py b/examples/pybullet/examples/testrender.py index 48fe8356c..8be4be7ff 100644 --- a/examples/pybullet/examples/testrender.py +++ b/examples/pybullet/examples/testrender.py @@ -5,6 +5,7 @@ import matplotlib.pyplot as plt import pybullet import time +import numpy as np #to reshape for matplotlib plt.ion() @@ -59,7 +60,13 @@ while(1): #note that sending the data using imshow to matplotlib is really slow, so we use set_data #plt.imshow(rgb,interpolation='none') - image.set_data(rgb) + + + #reshape is needed + np_img_arr = np.reshape(rgb, (h, w, 4)) + np_img_arr = np_img_arr*(1./255.) + + image.set_data(np_img_arr) ax.plot([0]) #plt.draw() #plt.show() diff --git a/examples/pybullet/examples/testrender_np.py b/examples/pybullet/examples/testrender_np.py index fc8acd310..814e7e0d0 100644 --- a/examples/pybullet/examples/testrender_np.py +++ b/examples/pybullet/examples/testrender_np.py @@ -60,15 +60,15 @@ while (1): #note that sending the data to matplotlib is really slow - #reshape is not needed - #np_img_arr = np.reshape(rgb, (h, w, 4)) - #np_img_arr = np_img_arr*(1./255.) + #reshape is needed + np_img_arr = np.reshape(rgb, (h, w, 4)) + np_img_arr = np_img_arr*(1./255.) #show #plt.imshow(np_img_arr,interpolation='none',extent=(0,1600,0,1200)) #image = plt.imshow(np_img_arr,interpolation='none',animated=True,label="blah") - image.set_data(rgb)#np_img_arr) + image.set_data(np_img_arr) ax.plot([0]) #plt.draw() #plt.show()