[Pythonclient] Fix hello_car.py print, reshape statements

microsoft · May 10, 2020 · d029cfc · d029cfc
1 parent 961456c
commit d029cfc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/PythonClient/car/hello_car.py b/PythonClient/car/hello_car.py
@@ -9,6 +9,7 @@
 client = airsim.CarClient()
 client.confirmConnection()
 client.enableApiControl(True)
+print("API Control enabled: %s" % (client.isApiControlEnabled()))
 car_controls = airsim.CarControls()
 
 for idx in range(3):
@@ -54,7 +55,7 @@
         airsim.ImageRequest("1", airsim.ImageType.DepthPerspective, True), #depth in perspective projection
         airsim.ImageRequest("1", airsim.ImageType.Scene), #scene vision image in png format
         airsim.ImageRequest("1", airsim.ImageType.Scene, False, False)])  #scene vision image in uncompressed RGB array
-    print('Retrieved images: %d', len(responses))
+    print('Retrieved images: %d' % (len(responses)) )
 
     for response in responses:
         filename = 'c:/temp/py' + str(idx)
@@ -69,7 +70,7 @@
         else: #uncompressed array
             print("Type %d, size %d" % (response.image_type, len(response.image_data_uint8)))
             img1d = np.fromstring(response.image_data_uint8, dtype=np.uint8) # get numpy array
-            img_rgb = img1d.reshape(response.height, response.width, 3) # reshape array to 3 channel image array H X W X 3
+            img_rgb = img1d.reshape(response.height, response.width, -1) # reshape array to 3 channel image array H X W X 3
             cv2.imwrite(os.path.normpath(filename + '.png'), img_rgb) # write to png 
 
 #restore to original state