AuroraMiddleware/bullet3
1
0
Fork 0
mirror of https://github.com/bulletphysics/bullet3 synced 2025-01-18 21:10:05 +00:00
Code Issues Releases Wiki Activity

use numpy to reshape image data to matplotlib compatible format

Browse Source 
fixes issue #1871
This commit is contained in:
Erwin Coumans 2018-09-12 08:19:30 -07:00
parent 9fa3e44539
commit c83fa30daa
2 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
examples/pybullet/examples/testrender.py
View File

@ -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()

8
examples/pybullet/examples/testrender_np.py
View File

@ -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()