From 72dc1bb4dd14e9fa9ee18f95e7b8c9ebd42bd424 Mon Sep 17 00:00:00 2001 From: James Braza Date: Sun, 2 Jan 2022 15:11:17 -0500 Subject: Added numpy reshapings to raw image --- examples/pybullet/examples/pointCloudFromCameraImage.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/pybullet/examples/pointCloudFromCameraImage.py b/examples/pybullet/examples/pointCloudFromCameraImage.py index a156090f8..c56a3eb86 100644 --- a/examples/pybullet/examples/pointCloudFromCameraImage.py +++ b/examples/pybullet/examples/pointCloudFromCameraImage.py @@ -72,8 +72,8 @@ imgW = int(width / 10) imgH = int(height / 10) img = p.getCameraImage(imgW, imgH, renderer=p.ER_BULLET_HARDWARE_OPENGL) -rgbBuffer = img[2] -depthBuffer = img[3] +rgbBuffer = np.reshape(img[2], (imgH, imgW, 4)) +depthBuffer = np.reshape(img[3], [imgW, imgH]) print("rgbBuffer.shape=", rgbBuffer.shape) print("depthBuffer.shape=", depthBuffer.shape) -- cgit v1.2.1